Update app.py

app.py

@@ -1,34 +1,39 @@
 import streamlit as st
 import openai
-import requests
-from PIL import Image
-from io import BytesIO
 
 # Access the OpenAI API key from Hugging Face Spaces secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-st.title("Advanced 2024 Video Marketing Plan Generator")
+st.title("2024 Video Marketing Plan Generator")
 
 # User inputs for the marketing plan
-st.subheader("Define Your Target Audience")
-customer_profile = st.text_area("Describe Your Ideal Customer", placeholder="Age, interests, demographics, etc.")
-
-st.subheader("Your Business Highlights")
-business_differentiators = st.text_input("Key Differentiators", placeholder="What makes your business unique?")
-
-st.subheader("Current Marketing Efforts")
-current_marketing_efforts = st.text_area("Current Marketing Strategies", placeholder="Describe your ongoing marketing activities.")
-
-st.subheader("Budget Considerations")
-marketing_budget = st.text_input("Marketing Budget", placeholder="Estimated budget for video marketing")
-
-if st.button('Generate My Detailed Video Marketing Plan'):
+st.subheader("Define Your Business and Audience")
+business_type = st.text_input("Your Business Type", placeholder="e.g., Cafe, Yoga Studio")
+target_audience = st.text_area("Describe Your Target Audience", placeholder="e.g., demographics, interests")
+
+st.subheader("Current Marketing and Budget")
+current_marketing = st.text_area("Current Marketing Efforts", placeholder="What are your ongoing marketing activities?")
+marketing_budget = st.text_input("Annual Marketing Budget", placeholder="Enter your budget for the year")
+
+# Test data button
+if st.button('Load Test Data'):
+    st.session_state['business_type'] = "Fitness Studio"
+    st.session_state['target_audience'] = "Young adults interested in health and wellness"
+    st.session_state['current_marketing'] = "Social media campaigns, local community events"
+    st.session_state['marketing_budget'] = "50000"
+
+# Use session state for input fields
+business_type = st.text_input("Your Business Type", value=st.session_state.get('business_type', ''), placeholder="e.g., Cafe, Yoga Studio")
+target_audience = st.text_area("Describe Your Target Audience", value=st.session_state.get('target_audience', ''), placeholder="e.g., demographics, interests")
+current_marketing = st.text_area("Current Marketing Efforts", value=st.session_state.get('current_marketing', ''), placeholder="What are your ongoing marketing activities?")
+marketing_budget = st.text_input("Annual Marketing Budget", value=st.session_state.get('marketing_budget', ''), placeholder="Enter your budget for the year")
+
+if st.button('Generate My Video Marketing Plan'):
     # Construct the prompt for text generation
     prompt_text = (
-        f"Create a comprehensive and detailed 2024 video marketing plan for a business targeting customers with the following profile: {customer_profile}. "
-        f"Include fleshed-out video ideas, partial script suggestions, and strategic advice. "
-        f"Key differentiators: {business_differentiators}. Current marketing efforts: {current_marketing_efforts}. "
-        f"Budget: {marketing_budget}."
+        f"Generate a 2024 video marketing plan for a {business_type} targeting an audience characterized as: {target_audience}. "
+        f"Include content pillars, and monthly video recommendations based on the current marketing efforts: {current_marketing} "
+        f"and an annual marketing budget of {marketing_budget}."
     )
 
     # Call the OpenAI API for text generation
@@ -36,16 +41,14 @@ if st.button('Generate My Detailed Video Marketing Plan'):
         response_text = openai.ChatCompletion.create(
             model="gpt-4",
             messages=[
-                {"role": "system", "content": "You are an expert marketing strategist."},
+                {"role": "system", "content": "You are an AI specializing in marketing strategy."},
                 {"role": "user", "content": prompt_text}
             ]
         )
-        detailed_marketing_plan = response_text.choices[0].message['content']
+        marketing_plan = response_text.choices[0].message['content']
     except Exception as e:
-        detailed_marketing_plan = f"Error in generating detailed marketing plan: {e}"
-
-    # Display the detailed marketing plan
-    st.markdown("### Your Comprehensive Video Marketing Plan")
-    st.write(detailed_marketing_plan)
+        marketing_plan = f"Error in generating marketing plan: {e}"
 
-    # Additional features like image generation can be added here if relevant
+    # Display the marketing plan
+    st.markdown("### Your Customized Video Marketing Plan")
+    st.write(marketing_plan)