Update app.py

app.py

@@ -108,11 +108,13 @@ 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.
 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
 )
 
 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.
@@ -173,19 +175,35 @@ 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 global variables for the agent's execution scope
-        # These will be available as global variables in the agent's Python execution environment
-        # when agent.run is called with the code_globals parameter.
-        agent_code_globals = {}
-        if input_image_path: agent_code_globals["input_image_path"] = str(input_image_path)
-        if input_audio_path: agent_code_globals["input_audio_path"] = str(input_audio_path)
-        if input_video_path: agent_code_globals["input_video_path"] = str(input_video_path)
-        if input_3d_model_path: agent_code_globals["input_3d_model_path"] = str(input_3d_model_path)
-        if input_file_path: agent_code_globals["input_file_path"] = str(input_file_path)
+        # 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)
+        if input_video_path: dynamic_globals_for_run["input_video_path"] = str(input_video_path)
+        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)
 
         progress(0.2, desc="Agent processing request...")
-        # Pass the global variables using the code_globals parameter
-        result = agent.run(full_prompt_with_instructions, code_globals=agent_code_globals)
+        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}")
+
 
         progress(0.8, desc="Processing result...")
         outputs = {