Update app.py

app.py

@@ -1,23 +1,38 @@
 """
-Simple Gradio demo to interact with the baseline agent locally.
+Gradio interface for the smolagents CodeAgent.
 
-Run with:
-```powershell
-python app.py
-```
-Then open the printed local URL or the Gradio link.
+By default this script will NOT launch a local Gradio server. To allow
+local runs (for testing) set the environment variable `RUN_LOCAL=1` or pass
+`--run-local` on the command line. This prevents accidental local launches
+when you intended to deploy to Hugging Face Spaces.
+
+Run locally (explicit):
+  RUN_LOCAL=1 python app.py
+  or
+  python app.py --run-local
+
+When running in a Space you can set `RUN_LOCAL=1` in Space secrets if desired,
+but Spaces typically run the app directly; set `RUN_LOCAL` only when you want
+explicit local behavior.
 """
+import os
+import sys
+import logging
 import gradio as gr
 from code_agent import run_agent
 
+logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.INFO)
+
 
-def respond(prompt: str):
-    # Run the real smolagents CodeAgent (or raise if not available)
+def respond(prompt: str) -> str:
+    if not prompt or not prompt.strip():
+        return "Please provide a prompt."
     try:
-        ans = run_agent(prompt)
-        return ans
+        return run_agent(prompt)
     except Exception as e:
-        return f"(agent error) {e}"
+        logger.error("Agent failed: %s", e)
+        return f"Agent error: {type(e).__name__}: {str(e)[:200]}"
 
 
 demo = gr.Interface(
@@ -29,5 +44,22 @@ demo = gr.Interface(
 )
 
 
+def _should_launch_local() -> bool:
+    # Explicit CLI flag takes precedence
+    if "--run-local" in sys.argv:
+        return True
+    # Environment variable opt-in
+    if os.environ.get("RUN_LOCAL", "0") == "1":
+        return True
+    return False
+
+
 if __name__ == "__main__":
-    demo.launch()
+    if _should_launch_local():
+        # Launch only when explicitly allowed
+        logger.info("Launching Gradio locally (RUN_LOCAL=1 or --run-local detected)")
+        demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    else:
+        logger.warning(
+            "Local Gradio launch prevented. To run locally set RUN_LOCAL=1 or pass --run-local."
+        )