Update app.py

app.py

@@ -1,42 +1,45 @@
 import gradio as gr
-
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 from gradio_client import Client
+import uvicorn
+
+app = FastAPI()
 
+# Enable CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],      # or restrict to specific domains
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
-def chat(message):
-    # return f"You said: {message}"
-    msg = f"Translate Zomi to English: '{message}'"
-    client = Client("Chatboong/Gemini_Translator") # working as of 20251125,
+def translate_zomi(text: str):
+    msg = f"Translate Zomi to English: '{text}'"
+    client = Client("Chatboong/Gemini_Translator")
     result = client.predict(
-    		message=msg,
-    		lang="English",
-    		is_streaming=True,
-    		api_name="/chat"
+        message=msg,
+        lang="English",
+        is_streaming=True,
+        api_name="/chat"
     )
     return result
-    # try: 
-    #     client = Client("Chatboong/Gemini_Translator") 
-    #     result = client.predict(
-    # 		message=msg,
-    # 		lang="English",
-    # 		is_streaming=True,
-    # 		api_name="chat"
-    #     )
-    #     return result 
-    # except Exception as e: 
-    #     return f"Upstream error: {e}"
-    
-
-with gr.Blocks() as demo: 
-    gr.Markdown("### Zomi Translator") 
-    inp = gr.Textbox(label="Input") 
-    out = gr.Textbox(label="Output") 
-    inp.submit(chat, inp, out) 
-    demo.load( 
-        fn=chat, 
-        inputs=gr.Textbox(), 
-        outputs=gr.Textbox(), 
-        api_name="/chat" 
-    ) 
-
-demo.launch()
+
+@app.post("/chat")
+async def chat_api(message: str):
+    return {"translation": translate_zomi(message)}
+
+def chat_ui(message):
+    return translate_zomi(message)
+
+with gr.Blocks() as demo:
+    gr.Markdown("### Zomi Translator")
+    inp = gr.Textbox(label="Input")
+    out = gr.Textbox(label="Output")
+    inp.submit(chat_ui, inp, out)
+
+app = gr.mount_gradio_app(app, demo, path="/")
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)