Update src/streamlit_app.py

src/streamlit_app.py

@@ -821,24 +821,65 @@ if st.session_state.ready_to_continue:
                             "summary": summary
                             })
                         st.dataframe(df.head(50))
+                        # save_path = save_dataset(df, f"subtask_{idx+1}")
+                        # if save_path:
+                        #     # summary = evaluate_dataset_with_gpt(subtask, df)
+                        #     st.markdown("**📝 Dataset Introduction:**")
+                        #     st.write(summary)
+                        st.dataframe(df.head(50))
                         save_path = save_dataset(df, f"subtask_{idx+1}")
                         if save_path:
-                            # summary = evaluate_dataset_with_gpt(subtask, df)
                             st.markdown("**📝 Dataset Introduction:**")
                             st.write(summary)
+                            # 添加下载按钮
+                            with open(save_path, "rb") as f:
+                                st.download_button(
+                                    label="📥 Download full dataset (CSV)",
+                                    data=f,
+                                    file_name=os.path.basename(save_path),
+                                    mime="text/csv"
+                                )
+                            with open(save_path, "rb") as f:
+                                st.download_button(
+                                    label="📥 Download full dataset (CSV)",
+                                    data=f,
+                                    file_name=os.path.basename(save_path),
+                                    mime="text/csv"
+                                )
                 if 'progress_bar' in locals():
                     progress_bar.progress((idx + 1) / total)
 
             # ---------- 之后 rerun：只展示 ----------
-            else:                                               # ★
+            # else:                                               # ★
+            #     if idx < len(st.session_state.all_dataframes):
+            #         # _hash, df = st.session_state.all_dataframes[idx]
+            #         # df = standardize_latlon(df)  
+            #         # st.dataframe(df.head(50))
+            #         entry = st.session_state.all_dataframes[idx]          # ➕ 新行
+            #         df = standardize_latlon(entry["df"])
+            #         st.dataframe(df.head(50))
+            #         st.write(entry.get("summary", ""))
+            else:  # ★
                 if idx < len(st.session_state.all_dataframes):
-                    # _hash, df = st.session_state.all_dataframes[idx]
-                    # df = standardize_latlon(df)  
-                    # st.dataframe(df.head(50))
-                    entry = st.session_state.all_dataframes[idx]          # ➕ 新行
+                    entry = st.session_state.all_dataframes[idx]
                     df = standardize_latlon(entry["df"])
+                    
+                    # 显示前 50 行数据
                     st.dataframe(df.head(50))
+                    
+                    # 显示数据简介
                     st.write(entry.get("summary", ""))
+            
+                    # 保存并添加下载按钮
+                    save_path = save_dataset(df, f"subtask_{idx+1}")
+                    if save_path:
+                        with open(save_path, "rb") as f:
+                            st.download_button(
+                                label="📥 Download full dataset (CSV)",
+                                data=f,
+                                file_name=os.path.basename(save_path),
+                                mime="text/csv"
+                            )
 
     # 检索完成后打标记
     if not st.session_state.retrieval_done:                     # ★