Update app.py

app.py

@@ -17,7 +17,7 @@ uploaded_file = st.sidebar.file_uploader("Choose a CSV file", type="csv")
 if uploaded_file is not None:
     st.session_state["df"] = pd.read_csv(uploaded_file)
 
-# Example function to generate a response from OpenAI's chat model
+# Function to generate a response from OpenAI's chat model
 def ask_openai(prompt):
     try:
         response = openai.ChatCompletion.create(
@@ -27,17 +27,18 @@ def ask_openai(prompt):
         )
         return response.choices[0].message["content"]
     except Exception as e:
-        print(f"Error in generating response: {e}")
-        return "Sorry, I couldn't generate a response. Please try again."
-
-# Example chat interaction in Streamlit
-if "chat_history" not in st.session_state:
-    st.session_state["chat_history"] = []
-
-if prompt := st.text_input("Ask me anything about the data:"):
-    st.session_state["chat_history"].append({"role": "user", "content": prompt})
-    response = ask_openai(prompt)
-    st.session_state["chat_history"].append({"role": "assistant", "content": response})
-
-for chat in st.session_state["chat_history"]:
-    st_message(chat["content"], is_user=True if chat["role"] == "user" else False)
+        st.error(f"Error in generating response: {e}")
+        return "I encountered an error. Please try again."
+
+# Input for new messages
+user_input = st.text_input("Ask me anything:", key="chat_input")
+
+# Process input on submission
+if user_input:
+    with st.chat_message("user"):
+        st.write(user_input)
+    
+    # Generate and display response
+    ai_response = ask_openai(user_input)
+    with st.chat_message("assistant"):
+        st.write(ai_response)