Update app.py

app.py

@@ -6,7 +6,9 @@ from llama_index import (
     VectorStoreIndex,
 )
 from llama_index.llms import OpenAI
-import openai
+from openai import OpenAI
+
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 # Define Streamlit layout and interaction
 st.title("Grounded Generations")
@@ -59,20 +61,17 @@ content_type = st.selectbox("Select content type:", ["Blog", "Tweet"]) # make so
 if st.button("Generate") and content_type:
     with st.spinner('Generating text...'):
         # Generate text using OpenAI API
-        openai.api_key = os.getenv("OPENAI_API_KEY")
         try:
             if content_type == "Blog":
                 prompt = f"Write a blog about 500 words in length using the {st.session_state['retrieved_text']}" #uses content from the stored state
             elif content_type == "Tweet":
                 prompt = f"Compose a tweet using the {st.session_state['retrieved_text']}"
-            response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo-16k",
-                messages=[
-                    {"role": "system", "content": "You are a helpful assistant."},
-                    {"role": "user", "content": prompt}
-                ]
-            )
-            generated_text = response['choices'][0]['message']['content']
+            response = client.chat.completions.create(model="gpt-3.5-turbo-16k",
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": prompt}
+            ])
+            generated_text = response.choices[0].message.content
             st.write(f"Generated Text: {generated_text}")
         except Exception as e:
             st.write(f"An error occurred: {e}")