Spaces:
Running on Zero
Running on Zero
| import uvicorn | |
| import gradio as gr | |
| import spaces | |
| # Import your FastAPI app and logic | |
| from app.main import app as fastapi_app, moderate_text, TextRequest | |
| def _satisfy_zerogpu_startup_check(): | |
| return "GPU available" | |
| # The UI function | |
| 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" | |
| for i, suggestion in enumerate(response.rewritten_text, 1): | |
| result += f"{i}. {suggestion}\n" | |
| elif response.is_flagged: | |
| result += "✅ (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 Gradio UI | |
| 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"), | |
| title="Multilingual Moderation API", | |
| ) | |
| # 🔥 THE MAGIC FIX: Mount Gradio INSIDE FastAPI, at the "/ui" path | |
| app = gr.mount_gradio_app(fastapi_app, demo, path="/ui") | |
| if __name__ == "__main__": | |
| # Run FastAPI (which now contains Gradio) | |
| uvicorn.run(app, host="0.0.0.0", port=7860) |