From 4e27161b8e736d5038eaadda69a548ba8e33514b Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Tue, 24 Sep 2024 17:58:22 +0800
Subject: [PATCH] feat: add download submission.csv button for kaggle scenario
 (#317)

* add download submission.csv button for kaggle scenario

* fix CI
---
 rdagent/log/ui/app.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/rdagent/log/ui/app.py b/rdagent/log/ui/app.py
index 1408fa50..66e5bd38 100644
--- a/rdagent/log/ui/app.py
+++ b/rdagent/log/ui/app.py
@@ -533,6 +533,16 @@ def feedback_window():
 - **Reason**: {h.reason}"""
                 )
 
+            if isinstance(state.scenario, KGScenario):
+                if fbe := state.msgs[round]["ef.runner result"]:
+                    submission_path = fbe[0].content.experiment_workspace.workspace_path / "submission.csv"
+                    st.download_button(
+                        label="**Download** submission.csv",
+                        data=submission_path.read_bytes(),
+                        file_name="submission.csv",
+                        mime="text/csv",
+                    )
+
 
 def evolving_window():
     title = "Development🛠️" if isinstance(state.scenario, SIMILAR_SCENARIOS) else "Development🛠️ (evolving coder)"