Update app.py

app.py

@@ -1,31 +1,50 @@
 import gradio as gr
 import spaces
-import uvicorn
 
-# Dummy GPU function at the module level to pass ZeroGPU startup check
+# 1. Import the logic and models from your FastAPI app
+from app.main import moderate_text, TextRequest
+
+# 2. ZeroGPU Startup Check
 @spaces.GPU
 def _satisfy_zerogpu_startup_check():
-    """Dummy function to bypass ZeroGPU startup requirement."""
     return "GPU available"
 
-from app.main import app as fastapi_app
-
-# 1. Create the Gradio Interface
-def moderate_ui(text):
-    return f"Processing text: {text}"
+# 3. The Real UI Function
+async def moderate_ui(text):
+    if not text.strip():
+        return "Please enter some text."
+    
+    try:
+        # Create the request object expected by your FastAPI route
+        request = TextRequest(text=text)
+        
+        # Call your actual FastAPI function directly
+        response = await moderate_text(request)
+        
+        # Format the result nicely for the UI
+        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 += f"✅ Suggested Proper Sentence:\n\"{response.rewritten_text}\""
+        elif response.is_flagged:
+            result += "✅ Suggested Proper Sentence: (Model could not generate a rewrite)"
+        else:
+            result += "✅ This text is clean. No proper sentence needed."
+            
+        return result
+    except Exception as e:
+        return f"Error: {str(e)}"
 
+# 4. Build the Interface
 demo = gr.Interface(
     fn=moderate_ui,
-    inputs="text",
-    outputs="text",
-    title="Multilingual Moderation API"
+    inputs=gr.Textbox(lines=3, placeholder="Type a toxic sentence here (e.g., 'I will kill you')..."),
+    outputs=gr.Textbox(lines=6, label="Moderation Result"),
+    title="Multilingual Moderation API",
+    description="Enter text to detect abuse/threats and get a suggested proper sentence."
 )
 
-# 2. Mount your FastAPI app onto Gradio's underlying FastAPI instance
-# This keeps your /moderate and /health endpoints accessible.
-# We mount it under the "/api" path to avoid conflicts.
-demo.app.mount("/api", fastapi_app)
-
-# 3. Launch Gradio (required for ZeroGPU)
+# 5. Launch
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)