changes

ui.py

@@ -1,48 +1,51 @@
+import os
+
 import gradio as gr
 import requests
 
-BASE_URL = "http://localhost:8000"
+BASE_URL = os.getenv("SPACE_URL", "http://localhost:8000")
 
 current_logs = []
 current_actions = []
 
 
 def reset_env(task):
-    global current_logs, current_actions
+    try:
+        res = requests.post(f"{BASE_URL}/reset", json={"task": task})
+        res.raise_for_status()
 
-    res = requests.post(f"{BASE_URL}/reset", json={"task": task})
-    data = res.json()
+        data = res.json()
+        obs = data["observation"]
 
-    current_logs = data["observation"].get("logs", [])
-    current_actions = []
+        return (
+            format_logs(obs.get("logs", [])),
+            "Environment reset",
+            "Running...",
+        )
 
-    return (
-        format_logs(current_logs),
-        "Environment reset",
-        "Running..."
-    )
+    except Exception as e:
+        return ("Error occurred", str(e), "Failed", None)
 
 
 def step_env(action):
-    global current_logs, current_actions
+    try:
+        res = requests.post(f"{BASE_URL}/step", json={"action": action})
+        res.raise_for_status()
 
-    res = requests.post(f"{BASE_URL}/step", json={"action": action})
-    data = res.json()
+        data = res.json()
 
-    obs = data["observation"]
-    reward = data["reward"]
-    done = data["done"]
+        obs = data["observation"]
+        reward = data["reward"]
+        done = data["done"]
 
-    current_logs = obs.get("logs", [])
-    current_actions.append(f"{action} → reward={reward}")
-
-    status = f"{'✅ Resolved' if done else 'Running'} | Reward: {reward}"
+        return (
+            format_logs(obs.get("logs", [])),
+            f"{action} → {reward}",
+            "Done" if done else "Running",
+        )
 
-    return (
-        format_logs(current_logs),
-        "\n".join(current_actions),
-        status
-    )
+    except Exception as e:
+        return ("Error occurred", str(e), "Failed", None)
 
 
 def format_logs(logs):