Update app.py

app.py

@@ -4,14 +4,9 @@ import uvicorn
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-from app.main import app as moderation_api, moderate_text, TextRequest
-# If main.py sits next to app.py (no app/ folder), use:
-# from main import app as moderation_api, moderate_text, TextRequest
-
-
-@spaces.GPU
-def _satisfy_zerogpu_startup_check():
-    return "GPU available"
+# main.py next to app.py:
+from main import app as moderation_api, moderate_text, TextRequest, generate_suggestions
+# If main.py is inside an app/ folder, use: from app.main import ...
 
 
 async def moderate_ui(text):
@@ -35,28 +30,30 @@ async def moderate_ui(text):
         return f"Error: {str(e)}"
 
 
-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",
-)
+with gr.Blocks(title="Multilingual Moderation API") as demo:
+    gr.Markdown("# Multilingual Moderation API")
+    inp = gr.Textbox(lines=3, placeholder="Type a toxic sentence here...", label="Text")
+    btn = gr.Button("Moderate")
+    out = gr.Textbox(lines=12, label="Moderation Result")
+    btn.click(moderate_ui, inp, out, api_name="moderate_ui")
 
-# Root FastAPI app
-root = FastAPI()
+    # Hidden event: registers the @spaces.GPU function so ZeroGPU detects it
+    with gr.Row(visible=False):
+        h_in = gr.Textbox()
+        h_out = gr.JSON()
+        h_btn = gr.Button()
+        h_btn.click(generate_suggestions, h_in, h_out, api_name="rewrite")
 
-# CORS so your Angular app can call it from the browser
+
+root = FastAPI()
 root.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # replace with your Angular domain(s) in production
+    allow_origins=["*"],  # restrict to your Angular domain in production
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# API first -> /api/moderate and /api/health
-root.mount("/api", moderation_api)
-
-# Gradio UI at "/"
+root.mount("/api", moderation_api)          # -> /api/moderate, /api/health
 app = gr.mount_gradio_app(root, demo, path="/")
 
 if __name__ == "__main__":