Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
-import openai
+from openai import OpenAI
 import os
 
-# Set up OpenAI API key
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Set up OpenAI client
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 st.title("LeetCode to Real-World Interview Question Generator")
 
@@ -22,13 +22,13 @@ for message in st.session_state.messages[1:]:  # Skip the system message
 
 # Function to generate response using OpenAI API
 def generate_response(prompt):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4",
         messages=st.session_state.messages + [{"role": "user", "content": prompt}],
         max_tokens=1000,
         temperature=0.7,
     )
-    return response.choices[0].message['content']
+    return response.choices[0].message.content
 
 # React to user input
 if prompt := st.chat_input("Enter a LeetCode question to transform:"):