Upload 11 files

app.py

@@ -1,8 +1,11 @@
 import streamlit as st
 import json
 import os
+import yaml
 from datetime import datetime, date, timedelta
 import pandas as pd
+from yaml.loader import SafeLoader
+import streamlit_authenticator as stauth
 from config_manager import ConfigManager
 from dashboard import Dashboard
 from tasks import TasksManager
@@ -190,22 +193,120 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
+def load_auth_config():
+    """Load authentication configuration from Google Drive"""
+    try:
+        config_manager = st.session_state.config_manager
+        
+        # Try to load config.yaml from Google Drive root
+        if config_manager.google_drive_enabled:
+            config_content = config_manager.drive_manager.download_file('config.yaml')
+            if config_content:
+                # Parse YAML content
+                config = yaml.load(config_content, Loader=SafeLoader)
+                return config
+        
+        # Fallback to local config.yaml if Google Drive fails
+        if os.path.exists('config.yaml'):
+            with open('config.yaml') as file:
+                config = yaml.load(file, Loader=SafeLoader)
+                return config
+        
+        st.error("❌ No auth config found")
+        return None
+    except Exception as e:
+        st.error(f"Error loading authentication config: {e}")
+        return None
+
+def get_user_folder_from_username(username):
+    """Get the user folder based on username using wedding mappings"""
+    try:
+        # Load auth config to get wedding mappings
+        auth_config = st.session_state.get('auth_config')
+        if not auth_config:
+            # Fallback to loading config directly
+            auth_config = load_auth_config()
+        
+        if auth_config and 'wedding_mappings' in auth_config:
+            wedding_mappings = auth_config['wedding_mappings']
+            
+            # Search through all wedding mappings to find the user
+            for wedding_name, wedding_info in wedding_mappings.items():
+                if 'users' in wedding_info and username in wedding_info['users']:
+                    return wedding_info['folder']
+        
+        # Fallback to old hardcoded logic for backward compatibility
+        if username == 'demo':
+            return 'demo_data'
+        elif username == 'laraandumang':
+            return 'laraandumang'
+        else:
+            return 'demo_data'  # Default fallback
+            
+    except Exception as e:
+        st.error(f"Error getting user folder for {username}: {e}")
+        # Fallback to demo_data on error
+        return 'demo_data'
+
+def get_wedding_info_for_user(username):
+    """Get wedding information for a specific user"""
+    try:
+        auth_config = st.session_state.get('auth_config')
+        if not auth_config:
+            auth_config = load_auth_config()
+        
+        if auth_config and 'wedding_mappings' in auth_config:
+            wedding_mappings = auth_config['wedding_mappings']
+            
+            for wedding_name, wedding_info in wedding_mappings.items():
+                if 'users' in wedding_info and username in wedding_info['users']:
+                    return {
+                        'wedding_name': wedding_name,
+                        'folder': wedding_info['folder'],
+                        'users': wedding_info['users'],
+                        'display_name': wedding_info.get('wedding_name', wedding_name)
+                    }
+        
+        return None
+    except Exception as e:
+        st.error(f"Error getting wedding info for {username}: {e}")
+        return None
+
 def main():
     # Initialize session state
     if 'config_manager' not in st.session_state:
         st.session_state.config_manager = ConfigManager()
     
-    # Check if app is initialized
-    if 'app_initialized' not in st.session_state:
-        st.session_state.app_initialized = False
-    
-    if not st.session_state.app_initialized:
-        show_landing_page()
+    # Load authentication configuration
+    if 'auth_config' not in st.session_state:
+        st.session_state.auth_config = load_auth_config()
+    
+    if st.session_state.auth_config is None:
+        st.error("Failed to load authentication configuration. Please check your Google Drive setup.")
+        st.stop()
+    
+    # Create authenticator (auto_hash=True by default, so passwords will be hashed automatically)
+    authenticator = stauth.Authenticate(
+        st.session_state.auth_config['credentials'],
+        st.session_state.auth_config['cookie']['name'],
+        st.session_state.auth_config['cookie']['key'],
+        st.session_state.auth_config['cookie']['expiry_days']
+    )
+    
+    # Check authentication status
+    if 'authentication_status' not in st.session_state:
+        st.session_state.authentication_status = None
+    
+    # Check if user is already authenticated
+    if st.session_state.get('authentication_status'):
+        # User is authenticated, show main app
+        show_main_app(authenticator)
     else:
-        show_main_app()
+        # Show login page and handle authentication
+        show_login_page(authenticator)
 
-def show_landing_page():
-    """Show the new landing page with website-like design and authentication"""
+def show_login_page(authenticator):
+    """Show the login page"""
     
     # Hero Section
     st.markdown("""
@@ -307,12 +408,12 @@ def show_landing_page():
             margin-bottom: 2rem;
             box-shadow: 0 4px 6px rgba(0,0,0,0.1);
         ">
-            <h3 style="color: #2d5016; margin-bottom: 1rem;">📅 Event Planning</h3>
+            <h3 style="color: #2d5016; margin-bottom: 1rem;">👰 Wedding Party</h3>
             <ul style="color: #666; line-height: 1.8;">
-                <li>Plan multiple events</li>
-                <li>Create detailed timelines</li>
-                <li>Manage locations</li>
-                <li>Track meal choices</li>
+                <li>Manage bridal party</li>
+                <li>Track responsibilities</li>
+                <li>Coordinate schedules</li>
+                <li>Store contact info</li>
             </ul>
         </div>
         """, unsafe_allow_html=True)
@@ -327,12 +428,12 @@ def show_landing_page():
             margin-bottom: 2rem;
             box-shadow: 0 4px 6px rgba(0,0,0,0.1);
         ">
-            <h3 style="color: #2d5016; margin-bottom: 1rem;">👰 Wedding Party</h3>
+            <h3 style="color: #2d5016; margin-bottom: 1rem;">📊 Dashboard</h3>
             <ul style="color: #666; line-height: 1.8;">
-                <li>Manage wedding party roles</li>
-                <li>Track responsibilities</li>
-                <li>Store contact information</li>
-                <li>Assign special tasks</li>
+                <li>Visual progress tracking</li>
+                <li>Key metrics overview</li>
+                <li>Timeline management</li>
+                <li>Quick insights</li>
             </ul>
         </div>
         """, unsafe_allow_html=True)
@@ -350,67 +451,32 @@ def show_landing_page():
             <h3 style="color: #2d5016; margin-bottom: 1rem;">☁️ Cloud Sync</h3>
             <ul style="color: #666; line-height: 1.8;">
                 <li>Google Drive integration</li>
-                <li>Access from anywhere</li>
                 <li>Automatic backups</li>
-                <li>Multi-device sync</li>
+                <li>Multi-device access</li>
+                <li>Real-time updates</li>
             </ul>
         </div>
         """, unsafe_allow_html=True)
     
     st.markdown("---")
     
-    # Get Started Section
-    st.markdown("## 🚀 Get Started")
-    
-    st.markdown("""
-    Ready to start planning your wedding? Choose one of the options below to get started.
-    """)
-    
-    # Create two columns for the buttons
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        st.markdown("### 🎭 Try Demo Mode")
-        st.markdown("""
-        Experience the app with sample data including:
-        • Sample wedding (Emma & James)
-        • Demo guests with RSVPs
-        • Vendors with complex payment schedules
-        • Tasks in various stages
-        • Wedding party information
-        """)
-        
-        if st.button("🎭 Start with Demo Data", type="primary", use_container_width=True):
-            with st.spinner("Loading demo data from Google Drive..."):
-                config_manager = st.session_state.config_manager
-                if config_manager.load_demo_data_from_drive():
-                    st.session_state.app_initialized = True
-                    st.success("✅ Demo data loaded successfully!")
-                    st.rerun()
-                else:
-                    st.error("❌ Failed to load demo data from Google Drive. Please check your connection and try again.")
+    # Login Section
+    st.markdown("## 🔐 Login to Your Wedding Planner")
     
-    with col2:
-        st.markdown("### 📝 Load Your Wedding")
-        st.markdown("""
-        Load your existing wedding data from Google Drive:
-        • Your guest list and RSVPs
-        • Your vendors and payments
-        • Your tasks and timeline
-        • Your wedding party details
-        """)
-        
-        st.markdown("**Note:** This will load data from the `laraandumang` folder in your Google Drive.")
-        
-        if st.button("🎉 Get Started", type="secondary", use_container_width=True):
-            with st.spinner("Loading your wedding data from Google Drive..."):
-                config_manager = st.session_state.config_manager
-                if config_manager.load_existing_data_from_drive():
-                    st.session_state.app_initialized = True
-                    st.success("✅ Wedding data loaded successfully!")
-                    st.rerun()
-                else:
-                    st.error("❌ Failed to load data from Google Drive. Please check your connection and try again.")
+    # Login form
+    try:
+        authenticator.login(location='main')
+    except Exception as e:
+        st.error(f"Login error: {e}")
+    
+    # Check authentication status and show appropriate message
+    if st.session_state.get('authentication_status') is False:
+        st.error("❌ Invalid username or password")
+    elif st.session_state.get('authentication_status') is None:
+        st.info("🔐 Please enter your username and password")
+    elif st.session_state.get('authentication_status'):
+        st.success(f"✅ Welcome, {st.session_state.get('name', 'User')}!")
+        st.rerun()  # Refresh to show main app
     
 
 
@@ -611,244 +677,48 @@ def show_wedding_setup_form():
         else:
             st.error("Please fill in at least the partner names and wedding date range in the form above.")
 
-def show_setup_page():
-    st.markdown('<div class="main-header"><h1>💒 Wedding Planner Setup</h1></div>', unsafe_allow_html=True)
-    
-    st.markdown("### Welcome! Let's set up your wedding planning app.")
-    
-    # Show Google Drive status on setup page
-    show_google_drive_status_setup()
-    
-    # Demo mode option
-    st.markdown("#### Choose Your Experience")
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        st.markdown("**🎭 Try Demo Mode**")
-        st.markdown("Experience the app with sample data including:")
-        st.markdown("• Sample wedding (Emma & James)")
-        st.markdown("• Demo guests with RSVPs")
-        st.markdown("• Vendors with complex payment schedules")
-        st.markdown("• Tasks in various stages")
-        st.markdown("• Wedding party information")
-        
-        if st.button("Start with Demo Data", type="primary"):
-            if st.session_state.config_manager.set_demo_mode(True):
-                st.success("Demo mode enabled! Loading sample data...")
-                st.rerun()
-            else:
-                st.error("Failed to enable demo mode.")
-    
-    with col2:
-        st.markdown("**📝 Start Fresh**")
-        st.markdown("Create your own wedding from scratch:")
-        st.markdown("• Enter your wedding details")
-        st.markdown("• Add your own events")
-        st.markdown("• Build your guest list")
-        st.markdown("• Manage your vendors")
-        st.markdown("• Track your tasks")
-        
-        if st.button("Create My Wedding", type="secondary"):
-            if st.session_state.config_manager.set_demo_mode(False):
-                st.success("Demo mode disabled. Ready to create your wedding!")
-                st.rerun()
-            else:
-                st.error("Failed to disable demo mode.")
-    
-    st.markdown("---")
+
+def show_main_app(authenticator):
+    # Get current user from session state (set by authenticator.login)
+    username = st.session_state.get('username')
+    name = st.session_state.get('name')
     
-    # Initialize session state for events and form data if not exists
-    if 'setup_events' not in st.session_state:
-        st.session_state.setup_events = []
-    if 'setup_form_data' not in st.session_state:
-        st.session_state.setup_form_data = {
-            'partner1_name': '',
-            'partner2_name': '',
-            'venue_city': '',
-            'wedding_start_date': date.today(),
-            'wedding_end_date': date.today(),
-            'custom_tags': '',
-            'task_assignees': ''
-        }
+    if not username or not name:
+        st.error("Authentication error: Missing user information")
+        return
     
-    # Basic wedding information form
-    with st.form("wedding_setup"):
-        st.markdown("#### Basic Wedding Information")
-        
-        col1, col2 = st.columns(2)
-        with col1:
-            partner1_name = st.text_input("Partner 1 Name", value=st.session_state.setup_form_data['partner1_name'], placeholder="Enter first partner's name")
-            partner2_name = st.text_input("Partner 2 Name", value=st.session_state.setup_form_data['partner2_name'], placeholder="Enter second partner's name")
-            venue_city = st.text_input("City", value=st.session_state.setup_form_data['venue_city'], placeholder="Enter city")
-        
-        with col2:
-            st.markdown("**Wedding Date Range**")
-            wedding_start_date = st.date_input("Start Date", value=st.session_state.setup_form_data['wedding_start_date'])
-            wedding_end_date = st.date_input("End Date", value=st.session_state.setup_form_data['wedding_end_date'])
-            
-            if wedding_end_date < wedding_start_date:
-                st.error("End date must be after start date")
-                wedding_end_date = wedding_start_date
-        
-        st.markdown("#### Task Organization")
-        st.info("Tasks will be automatically grouped by your wedding events.")
-        
-        st.markdown("#### Custom Tags")
-        st.markdown("Enter custom tags (one per line):")
-        custom_tags = st.text_area("Custom Tags", value=st.session_state.setup_form_data['custom_tags'], placeholder="e.g.,\nUrgent\nHigh Priority\nDeposit Required\nResearch Needed")
-        
-        st.markdown("#### Task Assignees")
-        st.markdown("Enter people who will regularly be assigned tasks (one per line):")
-        task_assignees = st.text_area("Task Assignees", value=st.session_state.setup_form_data['task_assignees'], placeholder="e.g.,\nMom\nDad\nWedding Planner\nBest Friend\nCoordinator")
-        
-        form_submitted = st.form_submit_button("Update Wedding Information")
-        
-        if form_submitted:
-            # Update session state with form data
-            st.session_state.setup_form_data = {
-                'partner1_name': partner1_name,
-                'partner2_name': partner2_name,
-                'venue_city': venue_city,
-                'wedding_start_date': wedding_start_date,
-                'wedding_end_date': wedding_end_date,
-                'custom_tags': custom_tags,
-                'task_assignees': task_assignees
-            }
-            st.success("Wedding information updated!")
-            st.rerun()
+    # Set user folder based on username
+    user_folder = get_user_folder_from_username(username)
     
-    # Event management section (outside form)
-    st.markdown("#### Wedding Events")
-    st.markdown("Define all your wedding events with their details:")
+    # Get wedding info for the user (for future use)
+    wedding_info = get_wedding_info_for_user(username)
     
-    # Add/Remove event buttons
-    col1, col2 = st.columns(2)
-    with col1:
-        if st.button("➕ Add Event"):
-            # Set default date to wedding start date
-            wedding_start = st.session_state.setup_form_data['wedding_start_date']
-            st.session_state.setup_events.append({
-                "name": "New Event",
-                "description": "",
-                "date_offset": 0,
-                "requires_meal_choice": False,
-                "meal_options": [],
-                "location": "",
-                "address": ""
-            })
-            st.rerun()
+    # Update config manager to use the correct user folder
+    config_manager = st.session_state.config_manager
+    config_manager.set_user_folder(user_folder)
     
-    with col2:
-        if len(st.session_state.setup_events) > 0 and st.button("➖ Remove Last Event"):
-            st.session_state.setup_events.pop()
-            st.rerun()
+    # Check if we need to load data (either not initialized or user changed)
+    current_user_folder = config_manager.get_current_user_folder()
+    user_changed = st.session_state.get('last_user_folder') != user_folder
     
-    # Display events
-    if st.session_state.setup_events:
-        for i, event in enumerate(st.session_state.setup_events):
-            with st.expander(f"Event {i+1}: {event['name']}", expanded=True):
-                col1, col2 = st.columns(2)
-                
-                with col1:
-                    event_name = st.text_input("Event Name", value=event['name'], key=f"event_name_{i}")
-                    event_description = st.text_input("Time", value=event['description'], placeholder="e.g., 2:00 PM, 6:30 PM", key=f"event_desc_{i}")
-                    event_location = st.text_input("Location Name", value=event.get('location', ''), placeholder="e.g., Central Park, Grand Ballroom", key=f"event_location_{i}")
-                
-                with col2:
-                    # Get wedding date range
-                    wedding_start = st.session_state.setup_form_data['wedding_start_date']
-                    wedding_end = st.session_state.setup_form_data['wedding_end_date']
-                    
-                    # Calculate current event date from date_offset
-                    current_event_date = wedding_start + timedelta(days=event['date_offset'])
-                    
-                    # Use date input without constraints - allow any date
-                    event_date = st.date_input(
-                        "Event Date", 
-                        value=current_event_date, 
-                        key=f"event_date_{i}",
-                        help="Select any date for this event"
-                    )
-                    
-                    # Show warning if date is outside wedding range
-                    if event_date < wedding_start or event_date > wedding_end:
-                        st.warning(f"⚠️ Selected date is outside your wedding date range ({wedding_start.strftime('%B %d, %Y')} - {wedding_end.strftime('%B %d, %Y')})")
-                    
-                    requires_meal_choice = st.checkbox("Requires Meal Choice", value=event['requires_meal_choice'], key=f"event_meal_{i}")
-                    
-                    # Meal options section (only show if meal choice is required)
-                    if requires_meal_choice:
-                        st.markdown("**Meal Options**")
-                        st.markdown("Enter meal options (one per line):")
-                        current_meal_options = event.get('meal_options', [])
-                        meal_options_text = '\n'.join(current_meal_options) if current_meal_options else ''
-                        meal_options = st.text_area("Meal Options", value=meal_options_text, placeholder="e.g.,\nDuck\nSurf & Turf\nRisotto (vegetarian)\nStuffed Squash (vegetarian)", key=f"event_meal_options_{i}", height=100)
-                    else:
-                        meal_options = ""
-                    
-                    event_address = st.text_area("Address", value=event.get('address', ''), placeholder="Enter full address (street, city, state, zip code)", key=f"event_address_{i}", height=80)
-                
-                # Calculate date_offset from the selected date
-                date_offset = (event_date - wedding_start).days
-                
-                # Parse meal options
-                meal_options_list = []
-                if requires_meal_choice and meal_options:
-                    meal_options_list = [option.strip() for option in meal_options.split('\n') if option.strip()]
-                
-                # Update session state
-                st.session_state.setup_events[i] = {
-                    "name": event_name,
-                    "description": event_description,
-                    "date_offset": date_offset,
-                    "requires_meal_choice": requires_meal_choice,
-                    "meal_options": meal_options_list,
-                    "location": event_location,
-                    "address": event_address
-                }
-    else:
-        st.info("No events added yet. Click 'Add Event' to get started!")
+    if not st.session_state.get('app_initialized', False) or user_changed:
+        with st.spinner(f"Loading {name}'s wedding data..."):
+            if user_folder == 'demo_data':
+                if config_manager.load_demo_data_from_drive():
+                    st.session_state.app_initialized = True
+                    st.session_state.last_user_folder = user_folder
+                    st.success("✅ Demo data loaded successfully!")
+                else:
+                    st.error("Failed to load demo data.")
+                    return
+            else:
+                if config_manager.load_existing_data_from_drive():
+                    st.session_state.app_initialized = True
+                    st.session_state.last_user_folder = user_folder
+                else:
+                    st.error("Failed to load wedding data.")
+                    return
     
-    # Save configuration button (after event management)
-    st.markdown("---")
-    if st.button("Save Configuration", type="primary"):
-        # Get form values from session state
-        form_data = st.session_state.setup_form_data
-        
-        if form_data['partner1_name'] and form_data['partner2_name'] and form_data['wedding_start_date'] and form_data['wedding_end_date']:
-            # Parse tags (task groups will be auto-generated from events)
-            custom_tags_list = [tag.strip() for tag in form_data['custom_tags'].split('\n') if tag.strip()]
-            task_assignees_list = [assignee.strip() for assignee in form_data['task_assignees'].split('\n') if assignee.strip()]
-            
-            # Create configuration
-            config = {
-                'wedding_info': {
-                    'partner1_name': form_data['partner1_name'],
-                    'partner2_name': form_data['partner2_name'],
-                    'wedding_start_date': form_data['wedding_start_date'].isoformat(),
-                    'wedding_end_date': form_data['wedding_end_date'].isoformat(),
-                    'venue_city': form_data['venue_city']
-                },
-                'custom_settings': {
-                    'custom_tags': custom_tags_list,
-                    'task_assignees': task_assignees_list
-                },
-                'wedding_events': st.session_state.setup_events
-            }
-            
-            # Save configuration
-            st.session_state.config_manager.save_config(config)
-            # Clear setup session state
-            if 'setup_events' in st.session_state:
-                del st.session_state.setup_events
-            if 'setup_form_data' in st.session_state:
-                del st.session_state.setup_form_data
-            st.success("Configuration saved successfully!")
-            st.rerun()
-        else:
-            st.error("Please fill in at least the partner names and wedding date range in the form above.")
-
-def show_main_app():
     # Load config
     config = st.session_state.config_manager.load_config()
     wedding_info = config.get('wedding_info', {})
@@ -869,6 +739,11 @@ def show_main_app():
     else:
         header_text = f"{partner1} & {partner2}'s Wedding Planner \n"
     
+    # Add wedding mapping info if available
+    user_wedding_info = get_wedding_info_for_user(username)
+    if user_wedding_info:
+        header_text += f"👥 Wedding: {user_wedding_info['display_name']} \n"
+    
     # Add demo mode indicator
     if is_demo_mode:
         header_text += "🎭 DEMO MODE - Sample Data"
@@ -897,11 +772,25 @@ def show_main_app():
     
     # Sidebar navigation
     with st.sidebar:
-        # Reset app button
-        if st.button("🔄 Reset App", type="secondary"):
+        # User info and logout
+        # Extract first name for a more friendly greeting
+        first_name = name.split()[0] if name else "User"
+        st.markdown(f"**Welcome, {first_name}!**")
+        if authenticator.logout(location='sidebar', key='logout_button'):
+            # Clear session state on logout
+            for key in list(st.session_state.keys()):
+                if key not in ['config_manager', 'auth_config']:
+                    del st.session_state[key]
+            
+            # Reset app initialization state
             st.session_state.app_initialized = False
-            # Clear all cached data and reset config manager state
-            st.session_state.config_manager.reset_app_state()
+            st.session_state.last_user_folder = None
+            
+            # Reset config manager state
+            if 'config_manager' in st.session_state:
+                st.session_state.config_manager.reset_app_state()
+                st.session_state.config_manager.user_folder = None
+            
             st.rerun()
         
         st.markdown("---")

config_manager.py

@@ -48,6 +48,14 @@ class ConfigManager:
             return "demo_data"
         return "laraandumang"
     
+    def set_user_folder(self, folder_name):
+        """Set the user-specific folder name"""
+        self.user_folder = folder_name
+    
+    def get_current_user_folder(self):
+        """Get the currently set user folder"""
+        return getattr(self, 'user_folder', self.get_user_folder())
+    
     def _initialize_google_drive(self):
         """Initialize Google Drive connection"""
         try:
@@ -68,7 +76,7 @@ class ConfigManager:
         
         try:
             # Get user-specific folder
-            user_folder = self.get_user_folder()
+            user_folder = self.get_current_user_folder()
             
             # List of data files to sync
             data_files = [
@@ -128,6 +136,7 @@ class ConfigManager:
         except Exception as e:
             print(f"Error syncing from Google Drive: {e}")
             # Don't fail the entire initialization if sync fails
+            # This is expected behavior - user can manually sync later
     
     def _sync_to_google_drive(self):
         """Sync local data files to Google Drive"""
@@ -162,6 +171,17 @@ class ConfigManager:
     
     def load_app_config(self):
         """Load app configuration from file"""
+        # For Hugging Face Spaces, check /tmp directory first
+        if self.is_huggingface:
+            tmp_config_path = f"/tmp/{self.app_config_file}"
+            if os.path.exists(tmp_config_path):
+                try:
+                    with open(tmp_config_path, 'r') as f:
+                        return json.load(f)
+                except (json.JSONDecodeError, FileNotFoundError):
+                    pass
+        
+        # Check original location
         if os.path.exists(self.app_config_file):
             try:
                 with open(self.app_config_file, 'r') as f:
@@ -372,7 +392,13 @@ class ConfigManager:
         """Toggle demo mode on/off"""
         self.app_config["demo_mode"] = not self.app_config.get("demo_mode", False)
         try:
-            with open(self.app_config_file, 'w') as f:
+            # For Hugging Face Spaces, use /tmp directory to avoid permission issues
+            if self.is_huggingface:
+                config_path = f"/tmp/{self.app_config_file}"
+            else:
+                config_path = self.app_config_file
+            
+            with open(config_path, 'w') as f:
                 json.dump(self.app_config, f, indent=2)
             return True
         except Exception as e:
@@ -383,7 +409,13 @@ class ConfigManager:
         """Set demo mode to specific value"""
         self.app_config["demo_mode"] = enabled
         try:
-            with open(self.app_config_file, 'w') as f:
+            # For Hugging Face Spaces, use /tmp directory to avoid permission issues
+            if self.is_huggingface:
+                config_path = f"/tmp/{self.app_config_file}"
+            else:
+                config_path = self.app_config_file
+            
+            with open(config_path, 'w') as f:
                 json.dump(self.app_config, f, indent=2)
             return True
         except Exception as e:
@@ -487,7 +519,7 @@ class ConfigManager:
             self.set_demo_mode(False)
             
             # Get user-specific folder
-            user_folder = self.get_user_folder()
+            user_folder = self.get_current_user_folder()
             
             # Check if wedding_config.json exists in Google Drive user folder
             config_content = self.drive_manager.download_file(f'{user_folder}/wedding_config.json')
@@ -544,6 +576,8 @@ class ConfigManager:
             return False
         except Exception as e:
             print(f"Error loading existing data from Google Drive: {e}")
+            # This is a common issue on Hugging Face Spaces due to network/SSL issues
+            # The error is expected and the user can retry manually
             return False
     
     def load_demo_data_from_drive(self):
@@ -556,7 +590,7 @@ class ConfigManager:
             self.set_demo_mode(True)
             
             # Load demo data from demo folder in Google Drive
-            demo_folder = self.get_user_folder()  # Will return "demo_data" since demo mode is enabled
+            demo_folder = self.get_current_user_folder()  # Will return "demo_data" since demo mode is enabled
             
             # Check if wedding_config.json exists in demo folder
             config_content = self.drive_manager.download_file(f'{demo_folder}/wedding_config.json')
@@ -613,6 +647,8 @@ class ConfigManager:
             return False
         except Exception as e:
             print(f"Error loading demo data from Google Drive: {e}")
+            # This is a common issue on Hugging Face Spaces due to network/SSL issues
+            # The error is expected and the user can retry manually
             return False
     
     def get_google_drive_status(self):
@@ -691,7 +727,7 @@ class ConfigManager:
         
         try:
             # Get user-specific folder
-            user_folder = self.get_user_folder()
+            user_folder = self.get_current_user_folder()
             
             modified_files = self.get_modified_files()
             
@@ -750,4 +786,4 @@ class ConfigManager:
             return True
         except Exception as e:
             print(f"Error resetting app state: {e}")
-            return False
+            return False

requirements.txt

@@ -1,4 +1,4 @@
-streamlit==1.44.0
+streamlit==1.28.1
 pandas==2.1.3
 plotly==5.17.0
 datetime
@@ -8,5 +8,5 @@ google-api-python-client>=2.0.0
 google-auth-httplib2>=0.1.0
 google-auth-oauthlib>=0.5.0
 google-auth>=2.0.0
-streamlit-authenticator>=0.4.2
+streamlit-authenticator>=0.2.3
 PyYAML>=6.0