Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+ import streamlit as st
 import openai
 
 # Access the OpenAI API key from Hugging Face Spaces secrets
@@ -8,9 +8,9 @@ st.title("2024 Video Marketing Plan Generator with Class Recommendations")
 
 # Function to recommend classes based on video-making experience
 def recommend_classes(video_experience):
-    # Class list from the Small Business Video Roadmap
+    # Full class list from the Small Business Video Roadmap
     class_list = [
-        """Course #1
+""" Course #1
 Getting Started with Video Using The 6-Step BVS Process
 Class #1: Why Video Works and Outlining Your 12-Month Plan
 Class #2: 6 Simple Steps to Creating a Video (Part 1)
@@ -311,25 +311,51 @@ Develop Your 2023 Real Estate Marketing Plan (Part 2)
 Implement Tristan's SOI Video Touch System (Part 1)
 Implement Tristan's SOI Video Touch System (Part 2)
 """
-        # Assuming the list is ordered from beginner to advanced
+        # Add your full class list here
     ]
 
     if video_experience < 3:
         # Recommend earlier classes for beginners
-        return class_list[:3]
+        return class_list[:3]  # Adjust indices as needed
     else:
         # Recommend more advanced classes for experienced users
-        return class_list[-3:]  # Assuming the last three classes are more advanced
+        return class_list[-3:]  # Adjust indices as needed
 
 # User inputs for the marketing plan
-# ... [rest of your code for inputs] ...
+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 Efforts")
+current_marketing = st.text_area("Current Marketing Strategies", placeholder="Describe your ongoing marketing activities.")
+
+st.subheader("Your Video Making Experience")
+video_experience = st.number_input("How many videos have you made in the past month?", min_value=0, max_value=100, step=1)
 
 if st.button('Generate My Video Marketing Plan and Class Recommendations'):
     # Construct the prompt for text generation
-    # ... [rest of your code to generate the marketing plan] ...
+    prompt_text = (
+        f"Generate a 2024 video marketing plan for a {business_type} targeting an audience characterized as: {target_audience}. "
+        f"Include up to four video ideas for each month and 10 specific distribution strategies to maximize video views, "
+        f"based on current marketing efforts: {current_marketing}."
+    )
+
+    # Call the OpenAI API for text generation
+    try:
+        response_text = openai.ChatCompletion.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "You are an AI specializing in marketing strategy."},
+                {"role": "user", "content": prompt_text}
+            ]
+        )
+        marketing_plan = response_text.choices[0].message['content']
+    except Exception as e:
+        marketing_plan = f"Error in generating marketing plan: {e}"
 
     # Display the marketing plan
-    # ... [rest of your code to display the marketing plan] ...
+    st.markdown("### Your Customized Video Marketing Plan")
+    st.write(marketing_plan)
 
     # Recommend classes based on video-making experience
     class_recommendations = recommend_classes(video_experience)
@@ -337,6 +363,10 @@ if st.button('Generate My Video Marketing Plan and Class Recommendations'):
     st.write(", ".join(class_recommendations))
 
     # Copy to clipboard button
-    # ... [rest of your code for the copy to clipboard button] ...
+    if marketing_plan:
+        full_text = marketing_plan + "\n\nRecommended Classes: " + ", ".join(class_recommendations)
+        st.text_area("Copy and paste the plan and recommendations:", full_text, height=300, key="text_area")
+        st.button("Copy to Clipboard", key="copy-button")
 
 # No additional code needed beyond this point for your application
+