Update src/streamlit_app.py

src/streamlit_app.py

@@ -863,23 +863,19 @@ if st.session_state.ready_to_continue:
                 if idx < len(st.session_state.all_dataframes):
                     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"
-                            )
+                    
+                    # 添加下载按钮
+                    tmp_path = f"/tmp/subtask_{idx+1}_display.csv"
+                    df.to_csv(tmp_path, index=False)
+                    with open(tmp_path, "rb") as f:
+                        st.download_button(
+                            label="📥 Download full dataset (CSV)",
+                            data=f,
+                            file_name=os.path.basename(tmp_path),
+                            mime="text/csv"
+                        )
 
     # 检索完成后打标记
     if not st.session_state.retrieval_done:                     # ★