Update app.py

app.py

@@ -82,10 +82,11 @@ def handle_submission():
     # Get the user's message and the selected tools
     #message = st.text_input("Enter your message:", "")
     
-   # selected_tools = []
-   # for tool, checkbox in tool_checkboxes:
-   #     if checkbox:
-   #         selected_tools.append(tool)
+    selected_tools = []
+    for tool, checkbox in tool_checkboxes:
+        if checkbox:
+            print("checked {tool.name}")
+            selected_tools.append(tool)
         
     #selected_tools = [tool for tool, checkbox in tool_checkboxes]
 
@@ -93,8 +94,8 @@ def handle_submission():
     agent = CustomHfAgent(url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", token=os.environ['HF_token'])
 
     # Run the agent with the user's message and selected tools
-    #response = agent.run(message, tools=selected_tools)
-    response = agent.chat(message)
+    response = agent.run(message, tools=selected_tools)
+    #response = agent.chat(message)
 
     print(response)