Update app.py

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
 import random
 from smolagents import GradioUI, CodeAgent, InferenceClientModel
-from smolagents import LiteLLMModel, ToolCallingAgent
+from smolagents import LiteLLMModel
+from smolagents import ToolCallingAgent, PythonInterpreterTool, TOOL_CALLING_SYSTEM_PROMPT
+
+agent = ToolCallingAgent(tools=[PythonInterpreterTool()], model=model, system_prompt=modified_prompt)
 import os
 from PIL import Image
 #from smolagents.src.smolagents.prompts 
@@ -28,6 +31,7 @@ evaluate_consumption = evaluate_consumption()
 evaluate_consumption_example = evaluate_consumption_example()
 FrugalAI_methods=FrugalAI_methods()
 
+modified_prompt = TOOL_CALLING_SYSTEM_PROMPT
 modified_system_prompt = (
     "\nYou are an AI agent that receives Python code from a manager. "
     "Your task is to test its resource consumption using a custom tool. "
@@ -42,7 +46,7 @@ alfred = ToolCallingAgent(
     tools=[evaluate_consumption, evaluate_consumption_example, FrugalAI_methods], 
     model=model,
     #additional_authorized_imports=['os'],
-    system_prompt=modified_system_prompt,
+    system_prompt=modified_prompt,
     #add_base_tools=True,  # Add any additional base tools
    # planning_interval=10   
 )