import streamlit as st
import openai

# Access the OpenAI API key from Hugging Face Spaces secrets
openai.api_key = st.secrets["OPENAI_API_KEY"]

st.title("2024 Video Content Calendar Generator")

# User inputs for the 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 Efforts")
current_marketing = st.text_area("Current Marketing Strategies", placeholder="Describe your ongoing marketing activities.")

if st.button('Generate My Custom 2024 Video Content Plan'):
    # Construct the prompt for text generation
    prompt_text = f"""
        Generate a 2024 video marketing plan for a {business_type} targeting an audience characterized as: {target_audience}.
        Include up to four video ideas for each month of 2024 with a short description of each, ensuring a full plan for all 12 months, try to incorporate current 
        marketing efforts when it makes sense to do so: {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. You are specialized in creating a diverse range of video topic ideas for businesses. 
                    Your ideas are a creative blend of topics that focus on the interests or pain points of the client. They are often community oriented 
                    but 1 in 4 ideas is promotional in nature and focuses on helping the business attract customers. 
                """},
                {"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
    st.markdown("### Your Customized Video Marketing Plan")
    st.write(marketing_plan)

    # Suggestion to enroll in Business Video School
    st.markdown("### Take Your Video Marketing to the Next Level!")
    st.write("""
        To successfully execute your new video marketing plan, consider enrolling in Business Video School. 
        Our One Day Challenge and Video Workspaces are designed to keep you accountable and help you stay on track with your plan. 
        Join now to transform your video marketing strategy!
    """)

# No additional code needed beyond this point for your application