Update main.py

main.py

@@ -1,8 +1,12 @@
 import gradio as gr
 import httpx
+import os
+
+# Use the container's base URL (relative path works inside the same app)
+BASE_URL = os.environ.get("BASE_URL", "")  # "" means same host:port as Gradio
 
 def query_agent(agent: str, message: str):
-    url = f"http://localhost:7860/api/{agent}"  # works locally
+    url = f"{BASE_URL}/api/{agent}"
     try:
         r = httpx.post(url, json={"text": message})
         return r.json().get("response", "No response")
@@ -10,7 +14,7 @@ def query_agent(agent: str, message: str):
         return str(e)
 
 def query_all(message: str):
-    url = "http://localhost:7860/api/run_all"
+    url = f"{BASE_URL}/api/run_all"
     try:
         r = httpx.post(url, json={"text": message})
         return [resp["response"] for resp in r.json()["responses"]]
@@ -23,10 +27,10 @@ def build_ui():
 
         with gr.Tab("Single Agent"):
             msg = gr.Textbox(label="Message")
-            agent = gr.Dropdown(["agent1","agent2","agent3","agent4"], label="Choose Agent")
+            agent = gr.Dropdown(["agent1", "agent2", "agent3", "agent4"], label="Choose Agent")
             output = gr.Textbox(label="Agent Response")
             btn = gr.Button("Send")
-            btn.click(fn=lambda m,a: query_agent(a,m), inputs=[msg,agent], outputs=output)
+            btn.click(fn=lambda m, a: query_agent(a, m), inputs=[msg, agent], outputs=output)
 
         with gr.Tab("All Agents"):
             msg_all = gr.Textbox(label="Message for All Agents")
@@ -37,3 +41,4 @@ def build_ui():
     return demo
 
 
+