Update app.py

app.py

@@ -1,45 +1,43 @@
-# 1. Import spaces FIRST, before any other CUDA-related packages
 import spaces
 import gradio as gr
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-import uvicorn
 
 # Import your moderation router
 from app.api.v1.endpoints.moderation import router as moderation_router
 
-# 2. Dummy GPU function that MUST be wired to a Gradio event for ZeroGPU
+# 1. Create a dummy GPU function. 
+# CRITICAL: This EXACT function must be passed to the button click below.
 @spaces.GPU
 def dummy_gpu_check():
     return "ZeroGPU Hardware Connected!"
 
-# 3. Build a minimal Gradio UI and bind the GPU function
+# 2. Build the Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# 🛡️ Moderation API")
     gr.Markdown("The API is live at `/api/v1/moderate`.")
+    
     with gr.Row():
         btn = gr.Button("Test ZeroGPU Connection")
         output = gr.Textbox()
+        # Bind the decorated function directly to the event
         btn.click(fn=dummy_gpu_check, inputs=[], outputs=output)
 
-# 4. Create the main FastAPI application
-app = FastAPI(title="Moderation API")
+# 3. Access Gradio's internal FastAPI app and add your routes
+app = demo.app
 
-# 5. Add CORS middleware to allow your Angular app to connect
+# 4. Add CORS middleware for Angular
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"], # In production, replace with your Angular app URL
+    allow_origins=["*"], # Your Angular app URL can go here in production
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# 6. Include your moderation routes FIRST
+# 5. Attach your moderation API to Gradio's app
 app.include_router(moderation_router, prefix="/api/v1")
 
-# 7. Mount the Gradio UI as a sub-application at the end
-app = gr.mount_gradio_app(app, demo, path="/ui")
-
-# 8. Run the FastAPI app (not demo.launch())
+# 6. Launch Gradio (This passes the ZeroGPU startup scanner)
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    demo.launch()