Update app.py

app.py

@@ -100,13 +100,7 @@ class CustomHfAgent(Agent):
 
 st.title("Hugging Face Agent and tools")
 
-# Initialize the agent
-agent = CustomHfAgent(
-    url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder",
-    token=os.environ['HF_token'],
-    additional_tools=[],
-    input_params={"max_new_tokens": 192},
-)
+
 
 # Display a welcome message
 with st.chat_message("assistant"):
@@ -124,9 +118,17 @@ submit_button = st.button("Submit")
 # Define the callback function to handle the form submission
 def handle_submission():
     selected_tools = [tools[idx] for idx, checkbox in enumerate(tool_checkboxes) if checkbox]
-    agent.tools = selected_tools
-
-    response = agent.chat(user_message)
+    
+    # Initialize the agent
+    agent = CustomHfAgent(
+        url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder",
+        token=os.environ['HF_token'],
+        additional_tools=selected_tools,
+        input_params={"max_new_tokens": 192},
+    )
+    #agent.tools = selected_tools
+
+    response = agent.run(user_message)
 
     print("Agent Response\n {}".format(response))