Upload 2 files

app.py

@@ -857,7 +857,11 @@ def create_new_user_and_wedding():
             'wedding_events': events
         }
         
-        # Create initial data files
+        # Store wedding config in session state for immediate use
+        # The files will be created on-demand when the user first accesses the app
+        st.session_state[f'wedding_config_{account_info["username"]}'] = wedding_config
+        
+        # Create initial data files structure (will be created on first use)
         initial_data_files = {
             'guest_list_data.json': {},
             'rsvp_data.json': {},
@@ -866,14 +870,18 @@ def create_new_user_and_wedding():
             'wedding_party.json': []
         }
         
-        # Upload wedding data to Google Drive
+        # Store initial data in session state
+        for filename, data in initial_data_files.items():
+            st.session_state[f'{filename}_{account_info["username"]}'] = data
+        
+        # Try to upload wedding config to Google Drive (this is the most important file)
         if config_manager.google_drive_enabled:
-            # Upload wedding config
-            config_manager.drive_manager.upload_file(f'{wedding_folder}/wedding_config.json', wedding_config)
-            
-            # Upload initial data files
-            for filename, data in initial_data_files.items():
-                config_manager.drive_manager.upload_file(f'{wedding_folder}/{filename}', data)
+            if config_manager.drive_manager.upload_file(f'{wedding_folder}/wedding_config.json', wedding_config):
+                st.success("✅ Wedding configuration saved to Google Drive!")
+            else:
+                st.warning("⚠️ Could not upload wedding config to Google Drive due to storage quota limitations. Your data is saved locally and will sync when you first use the app.")
+        
+        st.info("💡 **Note**: Your wedding data files will be created automatically when you first add guests, tasks, or vendors. This approach works better with Google Drive storage limitations.")
         
         # Update session state auth config
         st.session_state.auth_config = auth_config
@@ -1120,8 +1128,14 @@ def show_main_app(authenticator):
                     st.session_state.app_initialized = True
                     st.session_state.last_user_folder = user_folder
                 else:
-                    st.error("Failed to load wedding data.")
-                    return
+                    # Check if this is a new user who just signed up
+                    if 'wedding_config_' + username in st.session_state:
+                        st.info("🎉 Welcome! Your wedding planner is ready. Your data will be saved to Google Drive when you first add content or manually sync.")
+                        st.session_state.app_initialized = True
+                        st.session_state.last_user_folder = user_folder
+                    else:
+                        st.error("Failed to load wedding data. Please try logging in again or contact support.")
+                        return
     
     # Load config
     config = st.session_state.config_manager.load_config()

config_manager.py

@@ -573,13 +573,85 @@ class ConfigManager:
                 
                 self.data_loaded_from_drive = True
                 return True
-            return False
+            
+            # If no config found in Google Drive, check if this is a new user with data in session state
+            return self.load_new_user_data_from_session()
+            
         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_new_user_data_from_session(self):
+        """Load new user data from session state (for users who just signed up)"""
+        try:
+            # Get current username from session state
+            username = st.session_state.get('username')
+            if not username:
+                return False
+            
+            # Check if we have wedding config for this user in session state
+            config_key = f'wedding_config_{username}'
+            if config_key in st.session_state:
+                config_content = st.session_state[config_key]
+                
+                # Save config locally (in memory or file)
+                if self.use_memory_storage:
+                    self.memory_data['wedding_config.json'] = config_content
+                    # Update cache in session state
+                    st.session_state['cached_wedding_config'] = config_content
+                    print("Stored new user wedding_config.json in memory")
+                else:
+                    config_path = self.get_config_file_path()
+                    config_dir = os.path.dirname(config_path)
+                    if config_dir:
+                        os.makedirs(config_dir, exist_ok=True)
+                    with open(config_path, 'w') as f:
+                        json.dump(config_content, f, indent=2)
+                    # Update cache in session state
+                    st.session_state['cached_wedding_config'] = config_content
+                    print(f"Stored new user wedding_config.json in file: {config_path}")
+                
+                # Load data files from session state
+                data_files = [
+                    'guest_list_data.json',
+                    'rsvp_data.json', 
+                    'tasks.json',
+                    'vendors.json',
+                    'wedding_party.json'
+                ]
+                
+                for file_name in data_files:
+                    data_key = f'{file_name}_{username}'
+                    if data_key in st.session_state:
+                        content = st.session_state[data_key]
+                        if self.use_memory_storage:
+                            self.memory_data[file_name] = content
+                            # Update cache in session state
+                            cache_key = f"cached_{file_name}"
+                            st.session_state[cache_key] = content
+                            print(f"Stored new user {file_name} in memory")
+                        else:
+                            file_path = self.get_data_file_path(file_name)
+                            file_dir = os.path.dirname(file_path)
+                            if file_dir:
+                                os.makedirs(file_dir, exist_ok=True)
+                            with open(file_path, 'w') as f:
+                                json.dump(content, f, indent=2)
+                            # Update cache in session state
+                            cache_key = f"cached_{file_name}"
+                            st.session_state[cache_key] = content
+                            print(f"Stored new user {file_name} in file: {file_path}")
+                
+                self.data_loaded_from_drive = True
+                return True
+            
+            return False
+        except Exception as e:
+            print(f"Error loading new user data from session: {e}")
+            return False
+    
     def load_demo_data_from_drive(self):
         """Load demo wedding data from Google Drive and enable demo mode"""
         if not self.google_drive_enabled: