Update app.py

app.py

@@ -24,9 +24,6 @@ tool_checkboxes = [st.checkbox(f"Use {tool}") for tool in tools]
 # Submit button
 #submit_button = st.button("Submit")
 
-# Add the callback function to the Streamlit app
-submit_button = st.button("Submit", on_click=handle_submission)
-
 
 # Define the callback function to handle the form submission
 def handle_submission():
@@ -45,7 +42,11 @@ def handle_submission():
    # outputs = agent(inputs['input_ids'], attention_mask=inputs['attention_mask'])
 
     # Display the agent's response
- #   response = agent.run(message)
- #   st.text(f"{response:.4f}")
+    response = agent.run(message)
+    st.text(f"{response:.4f}")
     return "done"
 
+
+# Add the callback function to the Streamlit app
+submit_button = st.button("Submit", on_click=handle_submission)
+