Update app.py

app.py

@@ -19,9 +19,6 @@ 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"
@@ -814,27 +811,8 @@ def create_gradio_interface():
 # Create Gradio app
 gradio_app = create_gradio_interface()
 
-@app.get("/")
-async def root():
-    return {
-        "message": "Storybook Generator API + UI is running!",
-        "features": [
-            "API: Complete storybook generation to OCI bucket",
-            "UI: Test image generation with local file management",
-            "File Management: View, download, delete local images"
-        ],
-        "endpoints": {
-            "generate_storybook": "POST /api/generate-storybook",
-            "check_status": "GET /api/job-status/{job_id}",
-            "local_images": "GET /api/local-images"
-        }
-    }
-
-# For Hugging Face Spaces deployment
-def get_app():
-    return app
-
 # Mount Gradio under a specific path to avoid conflicts
+from fastapi.middleware.wsgi import WSGIMiddleware
 app.mount("/ui", WSGIMiddleware(gradio_app.server))
 
 # Enhanced root endpoint that explains the API structure