Update app.py

app.py

@@ -65,6 +65,8 @@ st.title("Hugging Face Agent")
 message_input = st.text_input("Enter your message:", "")
 
 # Checkboxes for the tools to be used by the agent
+
+        
 tool_checkboxes = [st.checkbox(f"Use {tool}") for tool in tools]
 
 # Submit button
@@ -74,7 +76,13 @@ tool_checkboxes = [st.checkbox(f"Use {tool}") for tool in tools]
 def handle_submission():
     # Get the user's message and the selected tools
     message = st.text_input("Enter your message:", "")
-    selected_tools = [tool for tool, checkbox in tool_checkboxes]
+    
+    selected_tools = []
+    for tool, checkbox in tool_checkboxes:
+        if checkbox:
+            selected_tools.append(tool)
+        
+    #selected_tools = [tool for tool, checkbox in tool_checkboxes]
 
     # Initialize the agent
     agent = CustomHfAgent(url_endpoint="https://api-inference.huggingface.co/models/bigcode/starcoder", token=os.environ['HF_token'])