Update app.py

app.py

@@ -30,11 +30,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, Agent, LocalAgent
 # print(agent.run("Is the following `text` (in Spanish) positive or negative?", text="¡Este es un API muy agradable!"))
 
 # Load tools
-random-character-tool = load_tool("Chris4K/random-character-tool")
-text-generation-tool = load_tool("Chris4K/text-generation-tool")
+random_character_tool = load_tool("Chris4K/random-character-tool")
+text_generation_tool = load_tool("Chris4K/text-generation-tool")
 
 
-tools = [random-character-tool, text-generation-tool]
+tools = [random_character_tool, text_generation_tool]
 
 # Define the custom HfAgent class
 class CustomHfAgent(Agent):
@@ -87,6 +87,15 @@ tool_checkboxes = [st.checkbox(f"Use {tool.name} --- {tool.description} ") for t
 # Submit button
 #submit_button = st.button("Submit")
 
+# 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},  # Set the desired value
+)
+
+
 # Define the callback function to handle the form submission
 def handle_submission():
     # Get the user's message and the selected tools
@@ -94,19 +103,15 @@ def handle_submission():
     
     #selected_tools = []
     selected_tools = [tool for idx, tool in enumerate(tools) if tool_checkboxes[idx]]
+
     print(selected_tools)
 
-    # 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},  # Set the desired value
-    )
+    agent.tools = selected_tools
 
+    
     # Run the agent with the user's message and selected tools
-    response = agent.run(message)
-    #response = agent.chat(message)
+    #response = agent.run(message)
+    response = agent.chat(message)
 
     print("Response " + response)