Update app.py

app.py

@@ -106,6 +106,10 @@ def chat_with_infinite_agent(message, lang, history):
 #Gradio UI
 css_path = os.path.join(os.path.dirname(__file__), "style.css")
 
+def api_predict(message, lang="English"):
+    history, _ = chat_with_infinite_agent(message, lang, [])
+    return {"response": history[-1]["content"]}
+
 with gr.Blocks(title="Infinite Agent", css=open(css_path).read()) as demo:
     #Avatar image
     gr.Image("avatar.png", elem_id="agent-avatar")
@@ -131,11 +135,7 @@ with gr.Blocks(title="Infinite Agent", css=open(css_path).read()) as demo:
     msg.submit(chat_with_infinite_agent, [msg, language, chatbot], [chatbot, chatbot])
     clear.click(lambda: [], None, chatbot, queue=False)
 
-    #API endpoint
-    def api_predict(message, lang="English"):
-        history, _ = chat_with_infinite_agent(message, lang, [])
-        return history[-1]["content"]
-
-    demo.add_api_route("/api/predict", api_predict, methods=["POST"])
+    #Expose the custom API endpoint
+    gr.api(api_predict, api_name="/api/predict")
 
 demo.launch()