Add HF Spaces environment detection for proper launch

app.py

@@ -9,6 +9,7 @@ from PIL import Image
 import logging
 from pathlib import Path
 import sys
+import os
 
 # Add backend to path
 sys.path.insert(0, str(Path(__file__).parent))
@@ -205,4 +206,17 @@ with gr.Blocks(title="Architecture AI Enhancer", theme=gr.themes.Soft()) as demo
 # Launch configuration
 if __name__ == "__main__":
     demo.queue(max_size=10)  # Enable queue for multiple users
-    demo.launch()  # HF Spaces auto-configures host and port
+    
+    # Detect if running on HF Spaces
+    is_hf_space = os.getenv("SPACE_ID") is not None
+    
+    if is_hf_space:
+        # HF Spaces configuration
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            show_api=False
+        )
+    else:
+        # Local development
+        demo.launch(share=True)