Update app.py

app.py

@@ -318,10 +318,16 @@ with gr.Blocks(title="LinkBERT by DEJAN AI") as demo:
     )
 
 
-# === Custom FastAPI with CORS ===
+# === Create FastAPI app with custom routes, then mount Gradio ===
+class ClassifyRequest(BaseModel):
+    text: str
+    threshold: float = 70.0
+
+
+# Create FastAPI app FIRST
 app = FastAPI()
 
-# Allow all origins for now (you can restrict to dejan.ai later)
+# Add CORS middleware
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -330,15 +336,10 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
-class ClassifyRequest(BaseModel):
-    text: str
-    threshold: float = 70.0
-
-
+# Add custom API routes BEFORE mounting Gradio
 @app.post("/api/classify")
 async def api_classify(request: ClassifyRequest):
-    """Direct REST endpoint - bypasses Gradio entirely"""
+    """Direct REST endpoint"""
     result = classify_text_json(request.text, request.threshold)
     return result
 
@@ -348,10 +349,18 @@ async def health():
     return {"status": "ok", "model": config.model_name}
 
 
-# Mount Gradio app
-app = gr.mount_gradio_app(app, demo, path="/")
+@app.get("/")
+async def root():
+    """Redirect root to Gradio UI"""
+    from fastapi.responses import RedirectResponse
+    return RedirectResponse(url="/gradio")
+
+
+# Mount Gradio AFTER defining routes
+app = gr.mount_gradio_app(app, demo, path="/gradio")
+
 
-# Launch
+# Launch with uvicorn
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)