Update app.py

app.py

@@ -7,26 +7,49 @@ from huggingface_hub import HfApi, upload_file, list_repo_files, hf_hub_download
 # Configuration for Hugging Face Repository
 REPO_ID = "MarcosRodrigo/Breakfast-Poll"  # Use the correct format: namespace/repo_name
 HISTORY_DIR = "history"  # Directory within the repository to store history
+TEMP_FILE = "current_selections.csv"  # Temporary file to store current selections
 
 # Hugging Face API (requires a token with write access)
 hf_token = st.secrets["HF_TOKEN"]  # Use the token stored in the secrets manager
 api = HfApi()
 
-# Initialize session state for temporary and historical storage
-if "users" not in st.session_state:
-    st.session_state.users = []
-if "current_selections" not in st.session_state:
-    st.session_state.current_selections = []
+# Load temporary selections from the shared file
+def load_current_selections():
+    if os.path.exists(TEMP_FILE):
+        return pd.read_csv(TEMP_FILE)
+    else:
+        return pd.DataFrame(columns=["Name", "Drinks", "Food"])
+
+# Save current user selections to the shared CSV file
+def save_current_selection_to_file(current_selections):
+    current_selections.to_csv(TEMP_FILE, index=False)
+
+# Upload the shared file to Hugging Face repository for persistence
+def upload_temp_file_to_repo():
+    if os.path.exists(TEMP_FILE):
+        upload_file(
+            path_or_fileobj=TEMP_FILE,
+            path_in_repo=TEMP_FILE,
+            repo_id=REPO_ID,
+            token=hf_token,
+            repo_type="space"
+        )
+
+# Download the shared file from the repository to ensure persistence
+def download_temp_file_from_repo():
+    try:
+        hf_hub_download(repo_id=REPO_ID, filename=TEMP_FILE, repo_type="space", token=hf_token)
+    except Exception:
+        # If the file does not exist in the repo, create an empty file
+        pd.DataFrame(columns=["Name", "Drinks", "Food"]).to_csv(TEMP_FILE, index=False)
 
 # Load history from the repository
 def load_history():
     history = []
-    # Specify `repo_type="space"` to indicate this is a Space repository
     files_in_repo = list_repo_files(REPO_ID, token=hf_token, repo_type="space")
     history_files = [f for f in files_in_repo if f.startswith(f"{HISTORY_DIR}/") and f.endswith(".txt")]
 
     for file in history_files:
-        # Download the file from the Space repository with `repo_type="space"`
         local_filepath = hf_hub_download(repo_id=REPO_ID, filename=file, token=hf_token, repo_type="space")
         summary_df = pd.read_csv(local_filepath)
         date = file.split("/")[-1].split(".txt")[0]
@@ -45,9 +68,11 @@ def save_summary_to_file(summary_df):
     # Upload the file to the Space repository with `repo_type="space"`
     upload_file(path_or_fileobj=local_filepath, path_in_repo=filename, repo_id=REPO_ID, token=hf_token, repo_type="space")
 
-# Load history into session state
+# Load persistent history and temporary selections on app start
 if "history" not in st.session_state:
+    download_temp_file_from_repo()  # Ensure the latest temp file is present
     st.session_state.history = load_history()
+    st.session_state.current_selections = load_current_selections().to_dict(orient="records")
 
 # Sidebar for navigating through different views
 menu = st.sidebar.selectbox("Select View", ["Poll", "History"])
@@ -68,7 +93,7 @@ if menu == "Poll":
     if st.session_state.step == 1:
         st.header("Step 1: Enter your name")
         name = st.text_input("Name:")
-        if st.button("Next", key="step1_next") and name:  # Unique key for the button
+        if st.button("Next", key="step1_next") and name:
             st.session_state.users.append(name)
             st.session_state.step = 2
 
@@ -81,7 +106,7 @@ if menu == "Poll":
             "Italiano", "Café con soja", "Té", "Manzanilla", "Nada"
         ]
         selected_drinks = st.multiselect("Choose your drinks:", drinks_options)
-        if st.button("Next", key="step2_next") and selected_drinks:  # Unique key for the button
+        if st.button("Next", key="step2_next") and selected_drinks:
             st.session_state.current_selections.append({"Name": st.session_state.users[-1], "Drinks": selected_drinks})
             st.session_state.step = 3
 
@@ -93,7 +118,7 @@ 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:  # Unique key for the button
+        if st.button("Next", key="step3_next") and selected_food:
             st.session_state.current_selections[-1]["Food"] = selected_food
             st.session_state.step = 4
 
@@ -104,17 +129,22 @@ if menu == "Poll":
             df = pd.DataFrame(st.session_state.current_selections)
             st.table(df)
 
-            if st.button("Submit Summary", key="submit_summary"):  # Unique key for the button
+            if st.button("Submit Summary", key="submit_summary"):
                 # Save the current summary to a text file in the repository
                 save_summary_to_file(df)
-                
+
+                # Save and upload current selections for real-time updates
+                save_current_selection_to_file(df)
+                upload_temp_file_to_repo()
+
                 # Add to session state history
                 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}")
                 reset_selections()
                 st.session_state.step = 1
+                st.experimental_rerun()
 
 # History view to check past summaries
 elif menu == "History":