Update app.py

app.py

@@ -32,18 +32,15 @@ def load_current_selections():
 
 # Save current user selections to the shared CSV file without overwriting previous data
 def save_current_selection_to_file(current_selections):
-    # Convert list columns to strings to avoid unhashable type errors
     current_selections["Drinks"] = current_selections["Drinks"].apply(lambda x: ", ".join(x) if isinstance(x, list) else x)
     current_selections["Food"] = current_selections["Food"].apply(lambda x: ", ".join(x) if isinstance(x, list) else x)
 
-    # Read the existing file to avoid overwriting
     if os.path.exists(TEMP_FILE):
         existing_selections = pd.read_csv(TEMP_FILE)
         combined_selections = pd.concat([existing_selections, current_selections]).drop_duplicates()
     else:
         combined_selections = current_selections
 
-    # Save the updated DataFrame to the CSV file
     combined_selections.to_csv(TEMP_FILE, index=False)
 
 # Upload the shared file to Hugging Face repository for persistence
@@ -57,6 +54,15 @@ def upload_temp_file_to_repo():
             repo_type="space"
         )
 
+# Delete a file from the repository (e.g., `current_selections.csv`)
+def delete_file_from_repo(filename):
+    api.delete_file(
+        path_in_repo=filename,
+        repo_id=REPO_ID,
+        token=hf_token,
+        repo_type="space"
+    )
+
 # Download the shared file from the repository to ensure persistence and real-time updates
 def download_temp_file_from_repo():
     try:
@@ -82,16 +88,13 @@ def save_summary_to_history():
     timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     history_filename = f"{HISTORY_DIR}/{timestamp}.txt"
 
-    # Ensure the history directory exists
     if not os.path.exists(HISTORY_DIR):
         os.makedirs(HISTORY_DIR)
 
-    # Read the current selections file
     if os.path.exists(TEMP_FILE):
         summary_df = pd.read_csv(TEMP_FILE)
         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
 
@@ -120,7 +123,6 @@ if menu == "Poll":
             st.session_state.users.append(name)
             st.session_state.step = 2
 
-    # Step 2: Select Drinks
     if st.session_state.step == 2:
         st.header("Step 2: Select your drink(s)")
         drinks_options = [
@@ -133,7 +135,6 @@ 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 and Save Selections
     if st.session_state.step == 3:
         st.header("Step 3: Select your food(s)")
         food_options = [
@@ -158,33 +159,26 @@ elif menu == "Current":
     st.table(current_df)
 
     if st.button("Submit Summary to History"):
-        # Save the current selections to history
         timestamp = save_summary_to_history()
         st.success(f"Summary saved to history at {timestamp}")
-
-        # Reload history after saving to reflect the new entry
         st.session_state.history = load_history()
-
-        # Delete the current selections file and clear session state
+        
+        # Clear local and remote current selections
         if os.path.exists(TEMP_FILE):
-            os.remove(TEMP_FILE)  # Remove the local current_selections.csv file
-            # Reset the session state to reflect the empty state
-            st.session_state.current_selections = []
+            os.remove(TEMP_FILE)
+            delete_file_from_repo(TEMP_FILE)  # Delete the file from the remote repo
 
             # Create an empty CSV to replace the deleted one
             pd.DataFrame(columns=["Name", "Drinks", "Food"]).to_csv(TEMP_FILE, index=False)
-            # Upload the empty file to the repository to ensure shared state is reset
             upload_temp_file_to_repo()
-            
-        # Refresh the view to show the empty table
-        st.experimental_set_query_params(step="reset")
 
+        st.experimental_set_query_params(step="reset")
 
 # History view to check past summaries
 elif menu == "History":
     st.title("Breakfast Poll History")
     if not st.session_state.history:
-        st.session_state.history = load_history()  # Reload history if it's empty
+        st.session_state.history = load_history()
     if st.session_state.history:
         for record in st.session_state.history:
             st.subheader(f"Date: {record['Date']}")