Update app.py

app.py

@@ -1,29 +1,31 @@
+import spaces  # must be first
 import gradio as gr
-import spaces
+import uvicorn
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from app.main import app as moderation_app   # your existing FastAPI app with /moderate
 
-# Dummy function to pass ZeroGPU check
+from app.main import app as moderation_api, moderate_text, TextRequest
+# If main.py sits next to app.py (no app/ folder), use:
+# from main import app as moderation_api, moderate_text, TextRequest
+
+
 @spaces.GPU
 def _satisfy_zerogpu_startup_check():
     return "GPU available"
 
-# The UI function
+
 async def moderate_ui(text):
     if not text.strip():
         return "Please enter some text."
     try:
-        from app.main import moderate_text, TextRequest
-        request = TextRequest(text=text)
-        response = await moderate_text(request)
+        response = await moderate_text(TextRequest(text=text))
 
         result = f"🚨 Flagged: {response.is_flagged}\n\n"
         result += f"📝 Original: {response.original_text}\n\n"
         if response.is_flagged and response.rewritten_text:
             result += "✅ Suggested Proper Sentences:\n"
-            for i, suggestion in enumerate(response.rewritten_text, 1):
-                result += f"{i}. {suggestion}\n"
+            for i, s in enumerate(response.rewritten_text, 1):
+                result += f"{i}. {s}\n"
         elif response.is_flagged:
             result += "✅ (Model could not generate a rewrite)"
         else:
@@ -32,7 +34,7 @@ async def moderate_ui(text):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Create Gradio UI
+
 demo = gr.Interface(
     fn=moderate_ui,
     inputs=gr.Textbox(lines=3, placeholder="Type a toxic sentence here..."),
@@ -40,25 +42,22 @@ demo = gr.Interface(
     title="Multilingual Moderation API",
 )
 
-# Create the main FastAPI app
-app = FastAPI(title="Multilingual Moderation API")
+# Root FastAPI app
+root = FastAPI()
 
-# Add CORS middleware – allow all origins (or restrict as needed)
-app.add_middleware(
+# CORS so your Angular app can call it from the browser
+root.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],   # replace "*" with your Angular app's origin for production
-    allow_credentials=True,
+    allow_origins=["*"],  # replace with your Angular domain(s) in production
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# Mount the Gradio UI at the root path (or "/ui")
-app = gr.mount_gradio_app(app, demo, path="/")
+# API first -> /api/moderate and /api/health
+root.mount("/api", moderation_api)
 
-# Mount the moderation API at "/api" – keeps your /api/moderate endpoint
-app.mount("/api", moderation_app)
+# Gradio UI at "/"
+app = gr.mount_gradio_app(root, demo, path="/")
 
-# If you run this file directly:
 if __name__ == "__main__":
-    import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)