Update app.py

app.py

@@ -2,24 +2,15 @@ import streamlit as st
 import pandas as pd
 from datetime import datetime
 import os
-from huggingface_hub import HfApi, Repository
+from huggingface_hub import HfApi, upload_file, list_repo_files, hf_hub_download
 
 # Configuration for Hugging Face Repository
-REPO_ID = "your-huggingface-username/your-repo-name"  # Replace with your repository ID
-LOCAL_DIR = "repo"  # Local directory to sync with the repo
-history_dir = os.path.join(LOCAL_DIR, "history")
+REPO_ID = "https://huggingface.co/spaces/MarcosRodrigo/Breakfast-Poll"  # Replace with your repository ID
+HISTORY_DIR = "history"  # Directory within the repository to store history
 
-# Hugging Face Authentication (requires a token with write access)
+# Hugging Face API (requires a token with write access)
 hf_token = st.secrets["HF_TOKEN"]  # Store your token in Streamlit secrets
-
-# Initialize the Hugging Face repository
-repo = Repository(local_dir=LOCAL_DIR, clone_from=REPO_ID, use_auth_token=hf_token)
-
-# Function to commit changes to the Hugging Face repository
-def commit_changes(message="Updating history"):
-    repo.git_add(pattern="history/*")
-    repo.git_commit(message=message)
-    repo.git_push()
+api = HfApi()
 
 # Initialize session state for temporary and historical storage
 if "users" not in st.session_state:
@@ -27,28 +18,30 @@ if "users" not in st.session_state:
 if "current_selections" not in st.session_state:
     st.session_state.current_selections = []
 
-# Load history from existing text files in the repository
+# Load history from the repository
 def load_history():
     history = []
-    repo.git_pull()  # Pull the latest changes from the repo
-    if not os.path.exists(history_dir):
-        os.makedirs(history_dir)
-    for filename in os.listdir(history_dir):
-        if filename.endswith(".txt"):
-            date = filename.split(".txt")[0]
-            filepath = os.path.join(history_dir, filename)
-            summary_df = pd.read_csv(filepath)
-            history.append({"Date": date, "Summary": summary_df})
+    files_in_repo = list_repo_files(REPO_ID, token=hf_token)
+    history_files = [f for f in files_in_repo if f.startswith(f"{HISTORY_DIR}/") and f.endswith(".txt")]
+
+    for file in history_files:
+        local_filepath = hf_hub_download(repo_id=REPO_ID, filename=file, token=hf_token)
+        summary_df = pd.read_csv(local_filepath)
+        date = file.split("/")[-1].split(".txt")[0]
+        history.append({"Date": date, "Summary": summary_df})
     return history
 
-# Save current summary to a text file in the repository
+# Save the current summary to a text file and upload to the repository
 def save_summary_to_file(summary_df):
     timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-    if not os.path.exists(history_dir):
-        os.makedirs(history_dir)
-    filepath = os.path.join(history_dir, f"{timestamp}.txt")
-    summary_df.to_csv(filepath, index=False)
-    commit_changes(message=f"Added summary for {timestamp}")
+    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 repository
+    upload_file(path_or_fileobj=local_filepath, path_in_repo=filename, repo_id=REPO_ID, token=hf_token)
 
 # Load history into session state
 if "history" not in st.session_state: