Update app.py

app.py

@@ -29,16 +29,20 @@ 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)
 
-            # Input user question
+            # Continous chat
             question = st.text_input("Ask a question about the data:")
-            if st.button("Submit"):
-                if question:
-                    response = query_engine.query(question)
-                    # Displaying the response
-                    if response.startswith('http'):
-                        st.image(response, use_column_width=True)
-                    else:
+            while True:
+                if st.button("Submit") or st.session_state.is_enter_pressed:
+                    if question:
+                        response = query_engine.query(question)
                         st.markdown(f"**Response:** {response}")
 
+                # For continuous chat
+                st.text_input("Ask another question:", key="question")
+                st.session_state.is_enter_pressed = False
+
+                if st.session_state.is_enter_pressed:
+                    break
+
 if __name__ == "__main__":
     main()