Update app.py

app.py

@@ -1,26 +1,28 @@
 import os
 import uvicorn
 from fastapi import FastAPI
+from fastapi.middleware.wsgi import WSGIMiddleware
 import gradio as gr
 
-api = FastAPI()
+# import your separate logic
+from api import api_app   # FastAPI app for API endpoints
+from main import build_ui # function that returns a Gradio Blocks/Interface
 
-@api.get("/ping")
-async def ping():
-    return {"message": "pong"}
+# Create main FastAPI app
+app = FastAPI()
 
-with gr.Blocks() as demo:
-    gr.Markdown("## Gradio + FastAPI running in HF Spaces")
-    msg = gr.Textbox()
-    out = gr.Textbox()
-    msg.change(lambda x: f"You typed: {x}", msg, out)
+# Mount API under /api
+app.mount("/api", api_app)
 
-# Mount gradio at root
-api = gr.mount_gradio_app(api, demo, path="/")
+# Build and mount Gradio UI under /
+gradio_app = gr.routes.App(build_ui())  # build_ui() returns gr.Blocks/Interface
+app.mount("/", gradio_app)
 
 if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860))  # 👈 use HF Spaces port
-    uvicorn.run(api, host="0.0.0.0", port=port)
+    # Use Hugging Face's assigned port, fallback to 7860 locally
+    port = int(os.environ.get("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)
+