Update app.py

app.py

@@ -1,55 +1,46 @@
 import streamlit as st
 import openai
+from tenacity import retry, wait_fixed, stop_after_attempt
 
-# Access the OpenAI API key from Hugging Face Spaces secrets
+# Assuming your OpenAI API key is stored in Streamlit's secrets or set as an environment variable
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-st.title("Valentine's Day Video Scripter for Small Businesses")
-
-# User inputs
-st.subheader("About Your Business")
-business_description = st.text_area("Describe Your Business", placeholder="What does your business do? What are its unique features?")
-target_audience = st.text_area("Target Audience", placeholder="Describe the kinds of people you want to attract (e.g., demographics, interests)")
-
-st.subheader("Video Type and Platform")
-video_type = st.text_input("Video Type/Platform", placeholder="E.g., TikTok, Instagram Reels, YouTube")
-
-st.subheader("Initial Video Ideas")
-initial_ideas = st.text_area("Initial Video Ideas", placeholder="Any initial Valentine's Day themes or ideas you have in mind for the video?")
-
-if st.button('Generate Video Script'):
-    # Detailed prompt for AI focusing on Valentine's Day
-    ai_instructions = """
-    Create a playful and engaging Valentine's Day themed video script for the specified platform, focusing on content that resonates with the target audience for Valentine's promotions. The script should creatively promote the business while incorporating Valentine's Day elements.
-    Steps:
-    1. Review the business description, target audience, and initial video ideas, then brainstorm 20 potential video concepts with compelling Valentine's Day themed hooks.
-    2. Select the best concept that is playful and aligns with Valentine's Day. Develop a script under 200 words, suitable for short-form video formats. Consider 10 varied 
-    ideas for a compelling hook, the first sentence or two of the video. This should grab the viewer's attention and spark their curiosity to watch the rest of the video.
-    3. Include a strong and relevant call to action for Valentine's Day, encouraging viewers to engage with the business.
-    4. Provide the script first, followed by additional filming and editing tips suitable for the chosen platform, emphasizing Valentine's Day aesthetics.
-    
-    The script should be ready to use in a teleprompter, free of shot directions or speaker references, and tailored to Valentine's Day promotional activities.
-    """
-
-    # Construct the prompt for the AI
-    prompt_text = f"{ai_instructions}\nBusiness description: {business_description}. Target audience: {target_audience}. Video type/platform: {video_type}. Initial video ideas: {initial_ideas}."
-
-    # Call the OpenAI API for text generation
-    try:
-        response_text = openai.ChatCompletion.create(
-            model="gpt-4",
-            messages=[
-                {"role": "system", "content": ai_instructions},
-                {"role": "user", "content": prompt_text}
-            ]
-        )
-        script = response_text.choices[0].message['content']
-    except Exception as e:
-        script = f"Error in generating video script: {e}"
-
-    # Display the video script
-    st.markdown("### Your Valentine's Day Video Script")
+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(stop=stop_after_attempt(3), wait=wait_fixed(1))
+def call_openai_api(messages):
+    return openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages
+    )
+
+def custom_chat_gpt(user_input):
+    messages = initial_messages.copy()
+    messages.append({"role": "user", "content": user_input})
+    response = call_openai_api(messages)
+    chat_gpt_reply = response.choices[0].message['content']
+    return chat_gpt_reply
+
+st.title("Real Estate Video Script Writer")
+
+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')
+
+if generate_button:
+    script = custom_chat_gpt(user_input)
+    st.markdown("### Generated Video Script")
     st.write(script)
-
-# Disclaimer
-st.write("Disclaimer: This script is AI-generated. Please review and customize it to fit your specific business needs and video platform, especially for Valentine's Day.")