Update app.py

app.py

@@ -1,46 +1,47 @@
 import streamlit as st
 import openai
-from tenacity import retry, wait_fixed, stop_after_attempt
+from tenacity import retry, stop_after_attempt, wait_fixed
 
-# Assuming your OpenAI API key is stored in Streamlit's secrets or set as an environment variable
+# Set the OpenAI API key from Hugging Face Spaces secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-initial_messages = [{
-    "role": "system", 
-    "content": """Act as a real estate marketing video script writer. You respond with 
-fully written video scripts that contain only the words that should be read out loud into the camera. A real estate agent should be 
-able to take the response you give and immediately read it word-for-word into a camera without editing it. The scripts you create do not include
-shot directions, references to who is speaking, or any other extraneous notes that are not the actual words that should be read out oud.
-As a real estate video marketing expert you have studied
-the most effective marketing and social media videos made by real estate agents. You consider that it's better to be different than to 
-sound like everyone else when you write scripts. The scripts you write are succinct and compelling. They work well as short social media
-videos shared by real estate agents. They always begin with engaging opening lines that tease what the rest of the video is about and they end
-with a single strong call to action. If the script is a list the video starts with at least a single sentence explaining what that list 
-contains. They never start with the first item on the list. 
-They never include someone saying hi or introducing themselves. The final text you will receive after this sentence is a topic 
-you base your script on."""
-}]
-
+# Retry decorator to handle potential API call failures
 @retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
-def call_openai_api(messages):
+def call_openai_api(prompt):
     return openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
-        messages=messages
+        prompt=prompt,
+        max_tokens=150,
+        n=1,
+        stop=None
     )
 
-def custom_chat_gpt(user_input):
-    messages = initial_messages.copy()
-    messages.append({"role": "user", "content": user_input})
+def generate_video_hooks(script):
+    messages = [{
+        "role": "system", "content": """You are a real estate marketing video copywriter, 
+        and your task is to generate creative, intriguing opening lines for video scripts. 
+        You will receive a script and your response should be 10 new opening lines for that video. 
+        Each opening line should be less than 15 words, spark the viewer's curiosity, and be related to the topic of the video script. 
+        The best opening lines are questions or statements that tease or summarize the content while making people want to know more.  
+        They can be 2 sentences if that helps you create ones that fit better into the script.
+        Now, here's the script you need to create opening lines for: """
+    }, {
+        "role": "user", "content": script
+    }]
     response = call_openai_api(messages)
-    chat_gpt_reply = response.choices[0].message['content']
-    return chat_gpt_reply
+    return response.choices[0].text
+
+# Streamlit user interface
+st.title("Video Hook Generator")
 
-st.title("Real Estate Video Script Writer")
+# Text input for user to enter the script or video content theme
+user_input = st.text_area("Enter the script or theme for your video:", placeholder="Paste your script here or describe the theme.")
 
-user_input = st.text_area("Enter your video topic or theme:", placeholder="Enter a topic or theme for the video script")
-generate_button = st.button('Generate Script')
+# Button to generate video hooks
+if st.button('Generate Hooks'):
+    with st.spinner('Generating creative hooks...'):
+        hooks = generate_video_hooks(user_input)
+        st.markdown("### Generated Hooks")
+        st.write(hooks)
 
-if generate_button:
-    script = custom_chat_gpt(user_input)
-    st.markdown("### Generated Video Script")
-    st.write(script)
+# Note: Ensure your OpenAI API key is correctly set in st.secrets or your environment variables.