Update app.py

Api mounted with gradio server

app.py

@@ -13,6 +13,10 @@ import uvicorn
 import subprocess
 import shutil
 from pathlib import Path
+from gradio.routes import mount_gradio_app
+from fastapi import FastAPI, UploadFile, File, Form, HTTPException
+from fastapi.responses import JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
 
 
 
@@ -493,38 +497,13 @@ def create_interface():
     
     return interface
 
-
-def run_fastapi():
-    uvicorn.run(fastapi_app, host="0.0.0.0", port=8000)
-
-
-def main():
-    try:
-        threading.Thread(target=run_fastapi, daemon=True).start()
-
-        interface = create_interface()
-        print("🚀 Starting Gradio UI on http://localhost:7860")
-        print("🧠 FastAPI JSON endpoint available at http://localhost:8000/align")
-
-        interface.launch(
-            server_name="0.0.0.0",
-            server_port=7860,
-            share=False,
-            debug=False
-        )
-
-    except ImportError as e:
-        print("❌ Missing dependency:", e)
-    except Exception as e:
-        print("❌ Error launching application:", e)
-
-
-from fastapi import FastAPI, UploadFile, File, Form, HTTPException
-from fastapi.responses import JSONResponse
-from fastapi.middleware.cors import CORSMiddleware
-
+# Wrap everything in FastAPI
 fastapi_app = FastAPI()
 
+# Mount Gradio interface
+interface = create_interface()
+app = mount_gradio_app(fastapi_app, interface, path="/")
+
 fastapi_app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -606,4 +585,4 @@ async def align_api(
 
 
 if __name__ == "__main__":
-    main()
+    uvicorn.run(app, host="0.0.0.0", port=7860)