Update app.py

app.py

@@ -4,29 +4,33 @@ import openai
 # Access the OpenAI API key from Hugging Face Spaces secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-st.title("Holiday Video Script Generator for Realtors")
-
-# User inputs
-agent_name = st.text_input("Your Name", placeholder="Enter your name")
-community_focus = st.text_input("Community Focus", placeholder="Enter the community or area you focus on")
-holiday_theme = st.selectbox("Holiday Theme", ["Christmas", "Hanukkah", "New Year's", "General Holiday Season"])
-personal_message = st.text_area("Personal Message", placeholder="Enter any personal message or story you want to include")
-
-if st.button('Generate Script'):
-    # Formulate the user input for the script
-    user_input = (
-        f"""You are an AI assistant that helps real estate agents create compelling holiday video scripts. Create a script that is festive, engaging, and approximately one minute long. Include elements like community engagement, local highlights, market insights, home decoration tips, and a personalized message from {agent_name}, focusing on the {community_focus} community. The holiday theme is {holiday_theme}. Include this personal message: {personal_message}. End with notes on suitable B-roll shots and music.
-        """
-    )
+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')
 
-    # Call the OpenAI API with the chat model
-    response = openai.ChatCompletion.create(
-        model="gpt-4",  # Use the appropriate GPT-4 model
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": user_input}
-        ]
+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.
+    """
+
+    # Call the OpenAI API
+    response = openai.Completion.create(
+        model="gpt-4",
+        prompt=user_input,
+        max_tokens=300
     )
 
-    # Display the response from the API
-    st.write(response.choices[0].message['content'])
+    # Display the script
+    script = response.choices[0].text.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.")