diff --git a/libs/agent/agent/ui/gradio/app.py b/libs/agent/agent/ui/gradio/app.py
index bafa6c3e..f52a3222 100644
--- a/libs/agent/agent/ui/gradio/app.py
+++ b/libs/agent/agent/ui/gradio/app.py
@@ -133,8 +133,11 @@ class GradioChatScreenshotHandler(DefaultCallbackHandler):
 is_mac = platform.system().lower() == "darwin"
 
 # Detect if lume is available (host device is macOS)
-is_arm64 = platform.machine().lower() == "arm64"
-is_lume_available = is_mac or (is_arm64 and os.environ.get("LUME_HOST", "localhost") != "localhost")
+is_lume_available = is_mac or (os.environ.get("PYLUME_HOST", "localhost") != "localhost")
+
+print("PYLUME_HOST: ", os.environ.get("PYLUME_HOST", "localhost"))
+print("is_mac: ", is_mac)
+print("Lume available: ", is_lume_available)
 
 # Map model names to specific provider model names
 MODEL_MAPPINGS = {