app.py

import streamlit as st
import pandas as pd
from datetime import datetime
import os
from huggingface_hub import HfApi, Repository

# 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")

# Hugging Face Authentication (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()

# 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 history from existing text files in 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})
    return history

# Save current summary to a text file in 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}")

# Load history into session state
if "history" not in st.session_state:
    st.session_state.history = load_history()

# Sidebar for navigating through different views
menu = st.sidebar.selectbox("Select View", ["Poll", "History"])

# Function to reset the current selections after submission
def reset_selections():
    st.session_state.users = []
    st.session_state.current_selections = []

# Poll view with four consecutive steps
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:")
        if st.button("Next", key="step1_next") and name:  # Unique key for the button
            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 = [
            "Café con leche", "Colacao", "Descafeinado con leche", "Cortado", 
            "Aguasusia", "Aguasusia susia", "Café descafeinado con leche desnatada", 
            "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
            st.session_state.current_selections.append({"Name": st.session_state.users[-1], "Drinks": selected_drinks})
            st.session_state.step = 3

    # Step 3: Select Food
    if st.session_state.step == 3:
        st.header("Step 3: Select your food(s)")
        food_options = [
            "Barrita con aceite", "Barrita con tomate", "Palmera de chocolate", 
            "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
            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"):  # Unique key for the button
                # Save the current summary to a text file in the repository
                save_summary_to_file(df)
                
                # 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

# History view to check past summaries
elif menu == "History":
    st.title("Breakfast Poll History")
    if st.session_state.history:
        for record in st.session_state.history:
            st.subheader(f"Date: {record['Date']}")
            st.table(record["Summary"])
    else:
        st.write("No history records found.")