Update app.py

app.py

@@ -1,44 +1,30 @@
 import gradio as gr
 import spaces
+from app.main import app as fastapi_app
 
-from app.main import moderate_text, TextRequest
-
+# Dummy function to pass ZeroGPU check
 @spaces.GPU
 def _satisfy_zerogpu_startup_check():
     return "GPU available"
 
-async def moderate_ui(text):
-    if not text.strip():
-        return "Please enter some text."
-    
-    try:
-        request = TextRequest(text=text)
-        response = await moderate_text(request)
-        
-        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"
-            # Loop through the 5 suggestions and format them
-            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)}"
+# Create the Gradio Interface
+def moderate_ui(text):
+    return f"Processing text: {text}"
 
 demo = gr.Interface(
     fn=moderate_ui,
-    inputs=gr.Textbox(lines=3, placeholder="Type a toxic sentence here..."),
-    outputs=gr.Textbox(lines=12, label="Moderation Result"), # Increased lines for output
-    title="Multilingual Moderation API",
-    description="Enter text to detect abuse/threats and get suggested proper sentences."
+    inputs="text",
+    outputs="text",
+    title="Multilingual Moderation API"
 )
 
+# Mount FastAPI onto Gradio's app
+# We also explicitly add the docs URL route so it's accessible
+demo.app.mount("/api", fastapi_app)
+
+# This line is the key: it exposes the /api/docs route on the main server
+# by re-mounting the FastAPI app's docs endpoint.
+demo.app.add_route("/api/docs", fastapi_app.docs_ui_router)
+
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)