Update app.py

app.py

@@ -90,8 +90,10 @@ agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True,handle_par
 # agent = initialize_agent(
 #     tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
 # )
+prompt = st.chat_input("Say something")
+if prompt:
 
-with st.chat_message("assistant"):
-    st_callback = StreamlitCallbackHandler(st.container())
-    response = agent_executor.invoke(callbacks=[st_callback])
-    st.write(response)
+    with st.chat_message("assistant"):
+        st_callback = StreamlitCallbackHandler(st.container())
+        response = agent_executor.invoke({"input":prompt},callbacks=[st_callback])
+        st.write(response)