Update app.py

app.py

@@ -73,13 +73,18 @@ def load_history():
         history.append({"Date": date, "Summary": summary_df})
     return history
 
-# Save the current summary to a text file and upload to the repository
-def save_summary_to_file(summary_df):
+# Save the current summary to a text file in the history directory
+def save_summary_to_history():
     timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-    filename = f"{HISTORY_DIR}/{timestamp}.txt"
-    local_filepath = f"{timestamp}.txt"
-    summary_df.to_csv(local_filepath, index=False)
-    upload_file(path_or_fileobj=local_filepath, path_in_repo=filename, repo_id=REPO_ID, token=hf_token, repo_type="space")
+    history_filename = f"{HISTORY_DIR}/{timestamp}.txt"
+    # Read the current selections file
+    if os.path.exists(TEMP_FILE):
+        summary_df = pd.read_csv(TEMP_FILE)
+        # Save the DataFrame to a text file in the history directory
+        summary_df.to_csv(history_filename, index=False)
+        # Upload the file to the repository
+        upload_file(path_or_fileobj=history_filename, path_in_repo=history_filename, repo_id=REPO_ID, token=hf_token, repo_type="space")
+    return timestamp
 
 # Load persistent history and temporary selections on app start
 if "history" not in st.session_state:
@@ -119,7 +124,7 @@ if menu == "Poll":
             st.session_state.current_selections.append({"Name": st.session_state.users[-1], "Drinks": selected_drinks})
             st.session_state.step = 3
 
-    # Step 3: Select Food
+    # Step 3: Select Food and Save Selections
     if st.session_state.step == 3:
         st.header("Step 3: Select your food(s)")
         food_options = [
@@ -127,34 +132,28 @@ if menu == "Poll":
             "Palmera de chocolate blanco", "Yogurt", "Pincho de tortilla", "Nada"
         ]
         selected_food = st.multiselect("Choose your food:", food_options)
-        if st.button("Next", key="step3_next") and selected_food:
+        if st.button("Save Selections", key="save_selections") and selected_food:
             st.session_state.current_selections[-1]["Food"] = selected_food
-            st.session_state.step = 4
-
-    # Step 4: Display Summary
-    if st.session_state.step == 4:
-        st.header("Step 4: Summary of Selections")
-        if st.session_state.current_selections:
             df = pd.DataFrame(st.session_state.current_selections)
-            st.table(df)
-
-            if st.button("Submit Summary", key="submit_summary"):
-                save_summary_to_file(df)
-                save_current_selection_to_file(pd.DataFrame(st.session_state.current_selections))
-                upload_temp_file_to_repo()
-                timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-                st.session_state.history.append({"Date": timestamp, "Summary": df})
-                st.success(f"Summary submitted at {timestamp}")
-                st.session_state.step = 1
-                st.experimental_set_query_params(step="reset")
-
-# "Current" view to display the current summary of all users' selections
+            save_current_selection_to_file(df)
+            upload_temp_file_to_repo()
+            st.success(f"Selections saved for {st.session_state.users[-1]}!")
+            st.session_state.step = 1
+
+# "Current" view to display the current summary of all users' selections and submit to history
 elif menu == "Current":
     st.title("Current Selections of All Users")
     if st.button("Reload Selections"):
         download_temp_file_from_repo()  # Download the latest version of the shared file
     current_df = load_current_selections()
     st.table(current_df)
+    if st.button("Submit Summary to History"):
+        timestamp = save_summary_to_history()
+        st.success(f"Summary saved to history at {timestamp}")
+        # Clear the current selections
+        os.remove(TEMP_FILE)
+        upload_temp_file_to_repo()  # Upload the cleared file
+        st.experimental_set_query_params(step="reset")
 
 # History view to check past summaries
 elif menu == "History":