Update app.py

app.py

@@ -29,6 +29,10 @@ def main():
             llm = Anthropic(temperature=0.0, model='claude-3-opus-20240229', api_key=api_key)
             query_engine = PandasQueryEngine(df=df, llm=llm, verbose=True)
 
+            # Initialize session state
+            if "is_enter_pressed" not in st.session_state:
+                st.session_state.is_enter_pressed = False
+
             # Continous chat
             question = st.text_input("Ask a question about the data:")
             while True:
@@ -38,8 +42,8 @@ def main():
                         st.markdown(f"**Response:** {response}")
 
                 # For continuous chat
-                st.text_input("Ask another question:", key="question")
-                st.session_state.is_enter_pressed = False
+                question = st.text_input("Ask another question:")
+                st.session_state.is_enter_pressed = st.button("Enter")
 
                 if st.session_state.is_enter_pressed:
                     break