Update app.py

app.py

@@ -2,27 +2,18 @@ import gradio as gr
 import spaces
 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"
 
-# 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):
@@ -31,12 +22,10 @@ async def moderate_ui(text):
             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=gr.Textbox(lines=3, placeholder="Type a toxic sentence here..."),
@@ -45,7 +34,6 @@ demo = gr.Interface(
     description="Enter text to detect abuse/threats and get suggested proper sentences."
 )
 
-# Mount FastAPI onto Gradio for your external apps to use
 demo.app.mount("/api", fastapi_app)
 
 if __name__ == "__main__":