Update src/streamlit_app.py

src/streamlit_app.py

@@ -853,24 +853,48 @@ if st.session_state.ready_to_continue:
             #         df = standardize_latlon(entry["df"])
             #         st.dataframe(df.head(50))
             #         st.write(entry.get("summary", ""))
-            else:  # ★
+            # else:  # ★
+            #     if idx < len(st.session_state.all_dataframes):
+            #         entry = st.session_state.all_dataframes[idx]
+            #         df = standardize_latlon(entry["df"])
+            #         st.dataframe(df.head(50))
+            #         st.write(entry.get("summary", ""))
+                    
+            #         # 添加下载按钮
+            #         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 dataset (CSV)",
+            #                 data=f,
+            #                 file_name=os.path.basename(tmp_path),
+            #                 mime="text/csv",
+            #                 key=f"download_rerun_{idx}" 
+            #             )
+
+            else:  
                 if idx < len(st.session_state.all_dataframes):
                     entry = st.session_state.all_dataframes[idx]
                     df = standardize_latlon(entry["df"])
                     st.dataframe(df.head(50))
-                    st.write(entry.get("summary", ""))
-                    
-                    # 添加下载按钮
-                    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 dataset (CSV)",
-                            data=f,
-                            file_name=os.path.basename(tmp_path),
-                            mime="text/csv",
-                            key=f"download_rerun_{idx}" 
-                        )
+            
+                    if not st.session_state.do_visualize:   # ✅ 防止重复展示 summary 和按钮
+                        st.markdown("**📝 Dataset Introduction:**")
+                        st.write(entry.get("summary", ""))
+                        
+                        # 添加下载按钮
+                        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 dataset (CSV)",
+                                data=f,
+                                file_name=os.path.basename(tmp_path),
+                                mime="text/csv",
+                                key=f"download_rerun_{idx}" 
+                            )
+
+
 
     # 检索完成后打标记
     if not st.session_state.retrieval_done:                     # ★