Update app.py

app.py

@@ -5,23 +5,23 @@ import os
 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
+REPO_ID = "MarcosRodrigo/Breakfast-Poll"
+HISTORY_DIR = "history"
+TEMP_FILE = "current_selections.csv"
 
 # Hugging Face API (requires a token with write access)
-hf_token = st.secrets["HF_TOKEN"]  # Use the token stored in the secrets manager
+hf_token = st.secrets["HF_TOKEN"]
 api = HfApi()
 
 # Initialize all required session state variables
 if "users" not in st.session_state:
-    st.session_state.users = []  # List to hold users
+    st.session_state.users = []
 if "current_selections" not in st.session_state:
-    st.session_state.current_selections = []  # List to hold current selections
+    st.session_state.current_selections = []
 if "step" not in st.session_state:
-    st.session_state.step = 1  # Step for navigation
+    st.session_state.step = 1
 if "history" not in st.session_state:
-    st.session_state.history = []  # List to hold history data
+    st.session_state.history = []
 
 # Load temporary selections from the shared file
 def load_current_selections():
@@ -45,12 +45,11 @@ def upload_temp_file_to_repo():
             repo_type="space"
         )
 
-# Download the shared file from the repository to ensure persistence
+# Download the shared file from the repository to ensure persistence and real-time updates
 def download_temp_file_from_repo():
     try:
-        hf_hub_download(repo_id=REPO_ID, filename=TEMP_FILE, repo_type="space", token=hf_token)
+        hf_hub_download(repo_id=REPO_ID, filename=TEMP_FILE, repo_type="space", token=hf_token, local_dir=".")
     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
@@ -70,22 +69,18 @@ def load_history():
 def save_summary_to_file(summary_df):
     timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     filename = f"{HISTORY_DIR}/{timestamp}.txt"
-    
-    # Save the DataFrame to a local CSV file
     local_filepath = f"{timestamp}.txt"
     summary_df.to_csv(local_filepath, index=False)
-
-    # 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 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
+    download_temp_file_from_repo()
     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"])
+menu = st.sidebar.selectbox("Select View", ["Poll", "Current", "History"])
 
 # Function to reset the current selections after submission
 def reset_selections():
@@ -96,10 +91,6 @@ def reset_selections():
 if menu == "Poll":
     st.title("Breakfast Poll Application")
 
-    # Step 1: User's Name
-    if "step" not in st.session_state:
-        st.session_state.step = 1
-
     if st.session_state.step == 1:
         st.header("Step 1: Enter your name")
         name = st.text_input("Name:")
@@ -140,22 +131,22 @@ if menu == "Poll":
             st.table(df)
 
             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)
+                save_current_selection_to_file(pd.DataFrame(st.session_state.current_selections))
                 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()
 
+# "Current" view to display the current summary of all users' selections
+elif menu == "Current":
+    st.title("Current Selections of All Users")
+    download_temp_file_from_repo()  # Ensure the latest file is downloaded
+    current_df = load_current_selections()
+    st.table(current_df)
+
 # History view to check past summaries
 elif menu == "History":
     st.title("Breakfast Poll History")