Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import sys
 import traceback
 import warnings
@@ -20,6 +21,27 @@ def launch_final_blend_sceneweaver(share: bool = True, debug: bool = False):
             ui = UIManager()
             print("✅ UIManager instance created successfully")
 
+            # Pre-load inpainting pipeline on Hugging Face Spaces
+            # This ensures first request doesn't timeout due to model loading
+            if os.getenv('SPACE_ID'):
+                print("\n🔧 Detected Hugging Face Spaces environment")
+                print("⏳ Pre-loading inpainting pipeline to avoid first-request timeout...")
+                try:
+                    success = ui.sceneweaver.switch_to_inpainting_mode(
+                        conditioning_type='canny',
+                        progress_callback=lambda msg, pct: print(f"   [{pct:3d}%] {msg}")
+                    )
+                    if success:
+                        print("✅ Inpainting pipeline pre-loaded successfully!")
+                        print("   All requests will now complete within 300s limit")
+                    else:
+                        print("⚠️  Pre-load failed, will load on first request")
+                        print("   (First inpainting request may timeout)")
+                except Exception as e:
+                    print(f"⚠️  Pre-load warning: {e}")
+                    print("   Continuing anyway - will load on demand")
+                print()
+
             # Launch UI
             print("🚀 Launching interface...")
             interface = ui.launch(share=share, debug=debug)
@@ -78,4 +100,4 @@ def main():
         raise
 
 if __name__ == "__main__":
-    main()
+    main()