Update app.py

app.py

@@ -1,14 +1,27 @@
+import os
 import gradio as gr
-from agent import agent  # Import your refactored agent
-from Gradio_UI import GradioUI
+from agent import agent  # your CodeAgent
 
-# Optional: Uncomment to test agent endpoint separately
-# def run_agent(question):
-#     try:
-#         result = agent(question)
-#         return [str(result)]
-#     except Exception as e:
-#         return [f"Error: {e}"]
+def run_agent(question: str) -> str:
+    """
+    A simple shim that takes one input string and returns
+    the agent's final text answer as a plain string.
+    """
+    # If you want stateless calls, you can reset agent memory here.
+    response = agent(question)
+    return str(response).strip()
+
+iface = gr.Interface(
+    fn=run_agent,
+    inputs=gr.Textbox(lines=2, label="Question"),
+    outputs=gr.Textbox(label="Answer"),
+    title="My Agent — Single-Question API",
+    description=(
+        "POST to /run/predict with JSON {\"data\": [\"<your question>\"]},\n"
+        "and you’ll get back {\"data\": [\"<answer>\"]}."
+    )
+)
 
 if __name__ == "__main__":
-    GradioUI(agent).launch(debug=True, share=True)
+    # Gradio will automatically spin up the /run/predict endpoint for this Interface
+    iface.launch(server_name="0.0.0.0", server_port=7860, debug=True, share=False)