Update app.py

app.py

@@ -1,32 +1,28 @@
 import gradio as gr
 import spaces
 
-# 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():
     return "GPU available"
 
-# 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}\""
+            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:
@@ -36,15 +32,13 @@ async def moderate_ui(text):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# 4. Build the Interface
 demo = gr.Interface(
     fn=moderate_ui,
-    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"),
+    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 a suggested proper sentence."
+    description="Enter text to detect abuse/threats and get suggested proper sentences."
 )
 
-# 5. Launch
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)