Update app.py

app.py

@@ -6,37 +6,34 @@ openai.api_key = st.secrets["YOUR_OPENAI_API_KEY"]
 
 st.title("Holiday Video Script Generator for Real Estate Agents")
 
-# User inputs for personalization
-agent_name = st.text_input("Agent's Name", placeholder="Enter your name")
-neighborhood = st.text_input("Neighborhood or Area", placeholder="Enter the neighborhood or area you focus on")
-unique_selling_points = st.text_input("Unique Selling Points", placeholder="Mention any unique selling points or achievements from the year")
-personal_message = st.text_area("Personal Message", placeholder="Write a personal message you'd like to include", height=100)
-
-generate_script_button = st.button('Generate Script')
-
-if generate_script_button:
-    # Create the user input for the AI model
-    user_input = f"""
-        Create a festive and engaging one-minute holiday video script for a real estate agent named {agent_name}. The script should 
-        focus on the neighborhood of {neighborhood}, incorporate unique selling points: {unique_selling_points}, and include the personal 
-        message: {personal_message}. The script should be warm, welcoming, and build a sense of community. Keep it professional yet personal.
-        The final script should only contain the words the agent should read into the camera. It should not contain different shots or scenes.
-        You can make recommendations on backgrounds or b-roll after the script if you have ideas for those.
-    """
-
-    # Call the OpenAI API with the correct endpoint for chat models
+# User inputs
+agent_name = st.text_input("Your Name", placeholder="Enter your name")
+local_area = st.text_input("Your Local Area", placeholder="Enter the area you serve (e.g., Southwest Garden, St. Louis)")
+holiday_theme = st.text_input("Holiday Theme", placeholder="Enter a specific holiday theme or message (e.g., community spirit, gratitude, holiday events)")
+call_to_action = st.text_input("Call to Action", placeholder="Enter your desired call to action (e.g., visit your website, schedule a consultation)")
+
+initial_messages = [{
+    "role": "system",
+    "content": """Act as a real estate marketing video script writer specialized in holiday-themed scripts. Respond with 
+    fully written video scripts that contain only the words to be read out loud into the camera by a real estate agent. 
+    These scripts should not include shot directions, references to who is speaking, or any other extraneous notes. 
+    Create distinctive, succinct, and compelling scripts ideal for short social media videos. Start with engaging opening lines, 
+    avoid generic greetings, and conclude with a strong call to action based on the agent's input. Base the script on the specific 
+    holiday theme or topic provided by the agent."""
+}]
+
+if st.button('Generate Script'):
+    # Process the inputs and call the OpenAI API
+    user_input = f"Create a holiday video script for a real estate agent named {agent_name}, serving the {local_area} area. The holiday theme is '{holiday_theme}'. The script should end with a call to action: '{call_to_action}'."
+    
+    # Call the OpenAI API with the chat model
     response = openai.ChatCompletion.create(
-        model="gpt-4",
+        model="gpt-4",  # Replace with the GPT-4 model you are using
         messages=[
-            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "system", "content": initial_messages[0]["content"]},
             {"role": "user", "content": user_input}
         ]
     )
 
-    # Display the script
-    script = response.choices[0].message['content'].strip()
-    st.write("### Generated Script")
-    st.write(script)
-
-    st.write("### Notes for B-roll and Music (if applicable)")
-    st.write("Include here any notes on B-roll footage or background music that could enhance the video.")
+    # Display the response from the API
+    st.write(response.choices[0].message['content'])