Update app.py

app.py

@@ -6,8 +6,6 @@ import random
 import os
 
 # --- CONFIGURATION ---
-# We will use your username and a standard name for the storage dataset
-# The app will find your username automatically from the token
 DATASET_NAME = "my-french-flashcards-data" 
 CSV_FILE = "cards.csv"
 
@@ -21,7 +19,6 @@ def get_api():
     return HfApi(token=token), token
 
 def get_user_repo_id(api, token):
-    # Determine the current user's username to build the repo_id
     user_info = api.whoami(token=token)
     return f"{user_info['name']}/{DATASET_NAME}"
 
@@ -30,7 +27,6 @@ def load_data():
     repo_id = get_user_repo_id(api, token)
     
     try:
-        # Try to download the CSV
         local_path = hf_hub_download(
             repo_id=repo_id,
             filename=CSV_FILE,
@@ -39,26 +35,22 @@ def load_data():
         )
         return pd.read_csv(local_path)
     except Exception:
-        # If download fails, the dataset or file might not exist yet.
-        # Check if repo exists, if not create it.
+        # If repo or file doesn't exist, create repo
         try:
             api.create_repo(repo_id=repo_id, repo_type="dataset", private=True, exist_ok=True)
         except Exception as e:
             st.error(f"Could not create storage repository: {e}")
         
-        # Return an empty dataframe structure
         return pd.DataFrame(columns=["french", "context", "english", "active"])
 
 def save_data(df):
     api, token = get_api()
     repo_id = get_user_repo_id(api, token)
     
-    # Save dataframe to buffer
     csv_buffer = StringIO()
     df.to_csv(csv_buffer, index=False)
     csv_buffer.seek(0)
     
-    # Upload to Hub
     try:
         api.upload_file(
             path_or_fileobj=csv_buffer.read().encode(),
@@ -74,26 +66,23 @@ def save_data(df):
 
 # --- UI LOGIC ---
 
-# Initialize Session State
 if 'page' not in st.session_state:
     st.session_state['page'] = 'home'
 if 'df' not in st.session_state:
     st.session_state['df'] = load_data()
 if 'deck' not in st.session_state:
-    st.session_state['deck'] = [] # List of indices from df
+    st.session_state['deck'] = [] 
 if 'current_card_idx' not in st.session_state:
     st.session_state['current_card_idx'] = 0
 if 'side_cycle_count' not in st.session_state:
-    st.session_state['side_cycle_count'] = 0 # 0=Fr, 1=Ctx, 2=Eng, 3=Ctx -> Loop
+    st.session_state['side_cycle_count'] = 0 
 
 def set_page(page_name):
+    # UPDATED: Just set state, let Streamlit handle the rerun automatically
     st.session_state['page'] = page_name
-    st.rerun()
 
 def start_session():
-    # Filter only active cards
     df = st.session_state['df']
-    # Ensure 'active' column exists and is boolean (handle legacy data)
     if 'active' not in df.columns:
         df['active'] = True
     
@@ -110,32 +99,20 @@ def start_session():
     set_page('play')
 
 def next_card():
-    # Move to next card in the deck
     st.session_state['current_card_idx'] += 1
-    # Reset side to 0 (French)
     st.session_state['side_cycle_count'] = 0 
     
-    # If we reached end of deck, go home
     if st.session_state['current_card_idx'] >= len(st.session_state['deck']):
         st.balloons()
         st.success("Session Complete!")
         set_page('home')
-    else:
-        st.rerun()
 
 def archive_current_card():
-    # Get actual index in the dataframe
     current_deck_idx = st.session_state['current_card_idx']
     if current_deck_idx < len(st.session_state['deck']):
         df_idx = st.session_state['deck'][current_deck_idx]
-        
-        # Update DataFrame
         st.session_state['df'].at[df_idx, 'active'] = False
-        
-        # Save to Cloud
         save_data(st.session_state['df'])
-        
-        # Move to next card
         next_card()
 
 def add_card(french, context, english):
@@ -152,13 +129,11 @@ def add_card(french, context, english):
 def unarchive(df_idx):
     st.session_state['df'].at[df_idx, 'active'] = True
     save_data(st.session_state['df'])
-    st.rerun()
 
 # --- PAGE RENDERING ---
 
 st.set_page_config(page_title="French Flashcards", page_icon="🇫🇷", layout="centered")
 
-# Custom CSS for "Beautiful & Minimal" look
 st.markdown("""
     <style>
     .stButton>button {
@@ -213,13 +188,9 @@ elif st.session_state['page'] == 'archive':
     st.title("Archive")
     st.button("Back", on_click=lambda: set_page('home'))
     
-    # Show active/inactive status clearly
     df = st.session_state['df']
     if not df.empty:
-        # Display as a table with checkboxes? 
-        # Streamlit data editor is easiest
         st.write("Below are your inactive cards. Click 'Unarchive' to restore.")
-        
         archived = df[df['active'] == False]
         if archived.empty:
             st.info("Archive is empty.")
@@ -234,16 +205,12 @@ elif st.session_state['page'] == 'archive':
         st.write("No cards yet.")
 
 elif st.session_state['page'] == 'play':
-    # Get current card data
     deck_idx = st.session_state['current_card_idx']
     
     if deck_idx < len(st.session_state['deck']):
         df_idx = st.session_state['deck'][deck_idx]
         card = st.session_state['df'].iloc[df_idx]
         
-        # Determine which side to show based on cycle
-        # Cycle logic: 0->1->2->3(which is 2 again)->0...
-        # Map: 0:Fr, 1:Ctx, 2:En, 3:Ctx
         cycle_map = {0: 'French', 1: 'Context', 2: 'English', 3: 'Context'}
         current_side_idx = st.session_state['side_cycle_count'] % 4
         side_name = cycle_map[current_side_idx]
@@ -253,12 +220,10 @@ elif st.session_state['page'] == 'play':
         elif side_name == 'Context': content = card['context']
         elif side_name == 'English': content = card['english']
         
-        # Progress Bar
         progress = (deck_idx + 1) / len(st.session_state['deck'])
         st.progress(progress)
         st.caption(f"Card {deck_idx + 1} of {len(st.session_state['deck'])}")
 
-        # Card Display
         st.markdown(f"""
         <div class="card-container">
             <div class="card-label">{side_name}</div>
@@ -266,7 +231,6 @@ elif st.session_state['page'] == 'play':
         </div>
         """, unsafe_allow_html=True)
         
-        # Buttons
         col1, col2 = st.columns([2, 1])
         with col1:
             def cycle_side():
@@ -281,5 +245,4 @@ elif st.session_state['page'] == 'play':
         st.button("Quit Session", on_click=lambda: set_page('home'))
         
     else:
-        # Fallback if index error
         set_page('home')