Update app.py

app.py

@@ -1,23 +1,25 @@
 import streamlit as st
-import openai
+from openai import OpenAI
 import os
 from dotenv import load_dotenv
 
 load_dotenv()
 
 # Set up OpenAI API key
-openai.api_key = os.getenv("OPENAI_API_KEY")  # Replace with your OpenAI API key if not using an environment variable
+api_key = os.getenv("OPENAI_API_KEY")  # Replace with your API key if not using an environment variable
+client = OpenAI(api_key=api_key)
 
 # Function to query OpenAI
 def query_openai(prompt, model="gpt-4o-mini"):
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model=model,
-            messages=prompt,
-            max_tokens=200,
-            temperature=0.7,
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant. Respond in 20 words."},
+                {"role": "user", "content": prompt}
+            ]
         )
-        return response["choices"][0]["message"]["content"]
+        return response.choices[0].message['content']
     except Exception as e:
         return f"Error: {e}"
 
@@ -45,7 +47,7 @@ if st.button("Send"):
         st.session_state.messages.append({"role": "user", "content": user_input})
 
         # Get bot response
-        bot_response = query_openai(st.session_state.messages)
+        bot_response = query_openai(user_input)
 
         # Add bot response to chat history
         st.session_state.messages.append({"role": "assistant", "content": bot_response})