Update app.py

app.py

@@ -1,25 +1,51 @@
 import gradio as gr
 import spaces
-from app.main import app as fastapi_app
+from app.main import app as fastapi_app, moderate_text, TextRequest
 
 # Dummy function to pass ZeroGPU check
 @spaces.GPU
 def _satisfy_zerogpu_startup_check():
     return "GPU available"
 
-# Create the Gradio Interface
-def moderate_ui(text):
-    return f"Processing text: {text}"
+# The real UI function that calls your FastAPI logic
+async def moderate_ui(text):
+    if not text.strip():
+        return "Please enter some text."
+    
+    try:
+        # Create the Pydantic request object
+        request = TextRequest(text=text)
+        
+        # Call your FastAPI function directly
+        response = await moderate_text(request)
+        
+        # Format the output for the Gradio 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 += "✅ Suggested Proper Sentences:\n"
+            for i, suggestion in enumerate(response.rewritten_text, 1):
+                result += f"{i}. {suggestion}\n"
+        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)}"
 
+# Build the Gradio 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..."),
+    outputs=gr.Textbox(lines=12, label="Moderation Result"),
+    title="Multilingual Moderation API",
+    description="Enter text to detect abuse/threats and get suggested proper sentences."
 )
 
-# Mount FastAPI onto Gradio's app
-# This exposes your API at /api/moderate
+# Mount FastAPI onto Gradio for your external apps to use
 demo.app.mount("/api", fastapi_app)
 
 if __name__ == "__main__":