Update app.py

app.py

@@ -509,33 +509,47 @@ async def startup_event():
 
 async def initialize_after_startup():
     """Initialize app components after server has started"""
+    global state
+    
     try:
+        print(f"[{FLOW_ID}] Starting initialization...")
+        
         # Load state with timeout
         try:
-            async with asyncio.timeout(30):
-                await load_state_from_hf()
+            print(f"[{FLOW_ID}] Loading state from HF...")
+            await asyncio.wait_for(load_state_from_hf(), timeout=30.0)
+            print(f"[{FLOW_ID}] State loaded successfully")
         except asyncio.TimeoutError:
             print("WARNING: Timeout loading state from HF, using default state")
+        except Exception as e:
+            print(f"WARNING: Error loading state from HF: {e}, using default state")
         
         # Update file list with timeout  
         try:
-            async with asyncio.timeout(60):
-                await update_file_list()
+            print(f"[{FLOW_ID}] Updating file list...")
+            await asyncio.wait_for(update_file_list(), timeout=60.0)
+            print(f"[{FLOW_ID}] File list updated successfully")
         except asyncio.TimeoutError:
             print("WARNING: Timeout updating file list, using existing state")
+        except Exception as e:
+            print(f"WARNING: Error updating file list: {e}, using existing state")
         
         # Start processing if needed
-        if state.current_index < state.total_files:
-            state.is_running = True
-            asyncio.create_task(process_next_file_task())
-            print(f"[{FLOW_ID}] Started processing from index {state.current_index}")
-        else:
-            state.is_running = False
-            print(f"[{FLOW_ID}] All files processed. Starting in Idle mode.")
+        async with state_lock:
+            if state.current_index < state.total_files:
+                state.is_running = True
+                asyncio.create_task(process_next_file_task())
+                print(f"[{FLOW_ID}] Started processing from index {state.current_index}/{state.total_files}")
+            else:
+                state.is_running = False
+                print(f"[{FLOW_ID}] All files processed. Starting in Idle mode.")
             
     except Exception as e:
-        print(f"[{FLOW_ID}] Error during initialization: {e}")
-        state.is_running = False
+        print(f"[{FLOW_ID}] Critical error during initialization: {e}")
+        import traceback
+        traceback.print_exc()
+        async with state_lock:
+            state.is_running = False
 
 
 @app.get("/", response_class=HTMLResponse)