Update app.py

app.py

@@ -108,13 +108,12 @@ tools.append(space_search_tool)
 model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
 
 # Create the agent
-# python_globals here are for fixed globals. Dynamic ones will be handled per-run.
+# Removed python_globals from constructor
 agent = CodeAgent(
     tools=tools,
     model=model,
     additional_authorized_imports=['PIL', 'Pillow', 'os', 'sys', 'numpy', 'huggingface_hub', 'gradio_client', 'uuid'],
-    add_base_tools=True,
-    python_globals=None # Explicitly setting to None or an empty dict if preferred
+    add_base_tools=True
 )
 
 AGENT_INSTRUCTIONS = """You are a highly capable AI assistant. Your primary goal is to accomplish tasks using a variety of tools, prioritizing Hugging Face Spaces.
@@ -175,7 +174,6 @@ def gradio_interface(user_prompt, input_image_path, input_audio_path, input_vide
         progress(0, desc="Initializing Agent...")
         full_prompt_with_instructions = f"{AGENT_INSTRUCTIONS}\n\nUSER PROMPT: {user_prompt}"
         
-        # Prepare a dictionary of dynamic global variables for the agent's execution scope
         dynamic_globals_for_run = {}
         if input_image_path: dynamic_globals_for_run["input_image_path"] = str(input_image_path)
         if input_audio_path: dynamic_globals_for_run["input_audio_path"] = str(input_audio_path)
@@ -183,26 +181,38 @@ def gradio_interface(user_prompt, input_image_path, input_audio_path, input_vide
         if input_3d_model_path: dynamic_globals_for_run["input_3d_model_path"] = str(input_3d_model_path)
         if input_file_path: dynamic_globals_for_run["input_file_path"] = str(input_file_path)
 
-        # Temporarily update the agent's python_globals for this specific run
-        # These globals are used by the CodeAgent when it executes Python code.
-        original_python_globals = agent.python_globals.copy() if agent.python_globals is not None else {}
-        
-        # Ensure agent.python_globals is a dict before updating
-        if agent.python_globals is None:
-            agent.python_globals = {}
-        
-        agent.python_globals.update(dynamic_globals_for_run)
+        # Access the agent's python_interpreter's globals
+        # The python_interpreter tool is usually named 'python_interpreter' or 'python'
+        # We need to ensure it exists and has a 'globals' attribute.
+        interpreter_tool = None
+        if hasattr(agent, 'python_interpreter') and agent.python_interpreter is not None: # Common attribute name
+            interpreter_tool = agent.python_interpreter
+        elif 'python_interpreter' in agent.tools and agent.tools['python_interpreter'] is not None:
+            interpreter_tool = agent.tools['python_interpreter']
+        elif 'python' in agent.tools and agent.tools['python'] is not None: # Another common name for the tool
+             interpreter_tool = agent.tools['python']
+
+
+        original_interpreter_globals = {}
+        if interpreter_tool and hasattr(interpreter_tool, 'globals') and isinstance(interpreter_tool.globals, dict):
+            original_interpreter_globals = interpreter_tool.globals.copy()
+            interpreter_tool.globals.update(dynamic_globals_for_run)
+            print(f"Updated agent.python_interpreter.globals with: {dynamic_globals_for_run}")
+        else:
+            print("Warning: Could not find or update python_interpreter globals on the agent.")
+
 
         progress(0.2, desc="Agent processing request...")
         result = None
         try:
-            # Call agent.run() without the unexpected keyword arguments.
-            # The dynamic_globals_for_run are now part of agent.python_globals for this execution.
             result = agent.run(full_prompt_with_instructions)
         finally:
-            # Restore the agent's original python_globals
-            agent.python_globals = original_python_globals
-            print(f"Restored agent.python_globals to: {agent.python_globals}")
+            # Restore the agent's original python_interpreter globals
+            if interpreter_tool and hasattr(interpreter_tool, 'globals'):
+                interpreter_tool.globals = original_interpreter_globals
+                print(f"Restored agent.python_interpreter.globals.")
+            else:
+                print("Warning: Could not restore python_interpreter globals.")
 
 
         progress(0.8, desc="Processing result...")