Update app.py

app.py

@@ -2,15 +2,15 @@ import streamlit as st
 import openai
 from tenacity import retry, stop_after_attempt, wait_fixed
 
-# Set the OpenAI API key from Hugging Face Spaces secrets
-openai.api_key = st.secrets["OPENAI_API_KEY"]
+# Ensure your OpenAI API key is set in your environment variables or Streamlit secrets
+openai.api_key = st.secrets.get("OPENAI_API_KEY", "your_openai_api_key_here")
 
 # Retry decorator to handle potential API call failures
 @retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
-def call_openai_api(prompt):
+def call_openai_api(messages):
     return openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
-        prompt=prompt,
+        messages=messages,
         max_tokens=150,
         n=1,
         stop=None
@@ -29,7 +29,7 @@ def generate_video_hooks(script):
         "role": "user", "content": script
     }]
     response = call_openai_api(messages)
-    return response.choices[0].text
+    return response.choices[0].message['content']
 
 # Streamlit user interface
 st.title("Video Hook Generator")
@@ -44,4 +44,4 @@ if st.button('Generate Hooks'):
         st.markdown("### Generated Hooks")
         st.write(hooks)
 
-# Note: Ensure your OpenAI API key is correctly set in st.secrets or your environment variables.
+# Note: Replace "your_openai_api_key_here" with your actual OpenAI API key or ensure it's set in st.secrets["OPENAI_API_KEY"].