Update app.py

app.py

@@ -5,41 +5,35 @@ from streamlit_chat import message
 import pandas as pd
 from streamlit_chat import message as st_message  # Ensure streamlit_chat is installed
 
-# Load data function remains the same
-def load_data(path):
-    return pd.read_csv(path)
 
-# Assuming you've set your OpenAI API key in the environment variables
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Ensure you have the OpenAI API key set in your environment variables
+openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-
-# File uploader and data loading logic can remain the same
-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)
-
-# Function to generate a response from OpenAI's chat model
+# Function to generate a response from OpenAI's GPT model using the updated API
 def ask_openai(prompt):
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",  # Adjust model as needed
-            messages=[{"role": "system", "content": "You are a helpful assistant."}, 
-                      {"role": "user", "content": prompt}],
+        response = openai.Completion.create(
+            model="text-davinci-003",  # Example model, adjust as needed
+            prompt=prompt,
+            temperature=0.7,
+            max_tokens=150,
+            n=1,
+            stop=None,
         )
-        return response.choices[0].message["content"]
+        return response.choices[0].text.strip()
     except Exception as e:
         st.error(f"Error in generating response: {e}")
         return "I encountered an error. Please try again."
 
-# Input for new messages
+# UI for input and displaying the chat
 user_input = st.text_input("Ask me anything:", key="chat_input")
 
-# Process input on submission
 if user_input:
+    # Display user input
     with st.chat_message("user"):
         st.write(user_input)
     
-    # Generate and display response
+    # Generate and display AI response
     ai_response = ask_openai(user_input)
     with st.chat_message("assistant"):
-        st.write(ai_response)
+        st.write(ai_response)