Update app.py

app.py

@@ -19,6 +19,9 @@ import uuid
 import hashlib
 from enum import Enum
 import random
+# Add these imports at the top (after existing imports)
+import threading
+from fastapi.middleware.wsgi import WSGIMiddleware
 
 # External OCI API URL - YOUR BUCKET SAVING API
 OCI_API_BASE_URL = "https://yukee1992-oci-story-book.hf.space"
@@ -831,14 +834,98 @@ async def root():
 def get_app():
     return app
 
+# Mount Gradio under a specific path to avoid conflicts
+app.mount("/ui", WSGIMiddleware(gradio_app.server))
+
+# Enhanced root endpoint that explains the API structure
+@app.get("/")
+async def root():
+    return {
+        "message": "Storybook Generator API is running!",
+        "api_endpoints": {
+            "health_check": "GET /api/health",
+            "generate_storybook": "POST /api/generate-storybook",
+            "check_job_status": "GET /api/job-status/{job_id}",
+            "local_images": "GET /api/local-images"
+        },
+        "web_interface": "GET /ui",
+        "note": "Use API endpoints for programmatic access, /ui for testing interface"
+    }
+
+# Add a simple test endpoint
+@app.get("/api/test")
+async def test_endpoint():
+    return {
+        "status": "success", 
+        "message": "API is working correctly",
+        "timestamp": datetime.now().isoformat()
+    }
+
+# For Hugging Face Spaces deployment
+def get_app():
+    return app
+
 if __name__ == "__main__":
     import uvicorn
-    print("🚀 Starting Complete Storybook Generator...")
-    print("📚 Features: API (n8n) + UI (Gradio) + Local File Management")
-    print("🔗 OCI API:", OCI_API_BASE_URL)
-    print("📁 Local storage:", PERSISTENT_IMAGE_DIR)
-    print("🌐 Web Interface: http://localhost:7860")
-    print("🔌 API Endpoints: http://localhost:7860/api/")
+    import os
+    
+    # Check if we're running on Hugging Face Spaces
+    HF_SPACE = os.environ.get('SPACE_ID') is not None
     
-    # Launch Gradio interface
-    gradio_app.launch(server_name="0.0.0.0", server_port=7860)
+    if HF_SPACE:
+        print("🚀 Running on Hugging Face Spaces - Integrated Mode")
+        print("📚 API endpoints available at: /api/*")
+        print("🎨 Web interface available at: /ui")
+        print("🔌 Both API and UI running on same port")
+        
+        # On Hugging Face, we need to run only one server
+        # Gradio will be mounted at /ui, FastAPI at root
+        gradio_app.launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=False,
+            show_error=True,
+            quiet=False
+        )
+    else:
+        # Local development - run separate servers
+        print("🚀 Running locally - Separate API and UI servers")
+        print("📚 API endpoints: http://localhost:8000/api/*")
+        print("🎨 Web interface: http://localhost:7860/ui")
+        
+        def run_fastapi():
+            """Run FastAPI on port 8000 for API calls"""
+            uvicorn.run(
+                app, 
+                host="0.0.0.0", 
+                port=8000, 
+                log_level="info",
+                access_log=False
+            )
+        
+        def run_gradio():
+            """Run Gradio on port 7860 for web interface"""
+            gradio_app.launch(
+                server_name="0.0.0.0",
+                server_port=7860,
+                share=False,
+                show_error=True,
+                quiet=True
+            )
+        
+        # Start both servers in separate threads
+        api_thread = threading.Thread(target=run_fastapi, daemon=True)
+        ui_thread = threading.Thread(target=run_gradio, daemon=True)
+        
+        api_thread.start()
+        print("✅ FastAPI server started on port 8000")
+        
+        ui_thread.start()
+        print("✅ Gradio server started on port 7860")
+        
+        # Keep the main thread alive
+        try:
+            while True:
+                time.sleep(1)
+        except KeyboardInterrupt:
+            print("🛑 Shutting down servers...")