Update app.py

app.py

@@ -1,47 +1,114 @@
 import streamlit as st
-import openai
+from openai import OpenAI
 from tenacity import retry, stop_after_attempt, wait_fixed
+import os
 
-# Ensure your OpenAI API key is set in your environment variables or Streamlit secrets
-openai.api_key = st.secrets.get("OPENAI_API_KEY", "your_openai_api_key_here")
+# Initialize OpenAI client
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-# Retry decorator to handle potential API call failures
 @retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
 def call_openai_api(messages):
-    return openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
+    """
+    Make API call to OpenAI with retry logic for reliability.
+    """
+    return client.chat.completions.create(
+        model="gpt-4",
         messages=messages,
-        max_tokens=150,
-        n=1,
-        stop=None
+        max_tokens=400,
+        temperature=0.9  # Higher creativity for hook generation
     )
 
 def generate_video_hooks(script):
+    """
+    Generate compelling video hooks based on the provided script.
+    """
     messages = [{
-        "role": "system", "content": """You are a real estate marketing video copywriter, 
-        and your task is to generate creative, intriguing opening lines for video scripts. 
-        You will receive a script and your response should be 10 new opening lines for that video. 
-        Each opening line should be less than 15 words, spark the viewer's curiosity, and be related to the topic of the video script. 
-        The best opening lines are questions or statements that tease or summarize the content while making people want to know more.  
-        They can be 2 sentences if that helps you create ones that fit better into the script.
-        Now, here's the script you need to create opening lines for: """
+        "role": "system", 
+        "content": """You are an expert at creating viral video hooks for real estate content. 
+        Your specialty is writing opening lines that stop scrollers and make viewers watch the entire video.
+        
+        Your hooks should:
+        - Be 15 words or less (can be 2 short sentences if needed)
+        - Create immediate curiosity or urgency
+        - Use power words that grab attention
+        - Avoid generic openings like "Hey guys" or "In today's video"
+        - Be specific and intriguing
+        - Make viewers think "I need to know this"
+        
+        Generate exactly 10 different hook options that are each unique in style:
+        - Question hooks that create curiosity
+        - Shocking statement hooks
+        - "What if" scenario hooks  
+        - Contrarian/surprising hooks
+        - Fear-based urgency hooks
+        - Benefit-focused hooks
+        - Story teaser hooks
+        - Number/list hooks
+        - Problem-focused hooks
+        - Secret/insider knowledge hooks
+        
+        Format your response as a numbered list with each hook on its own line."""
     }, {
-        "role": "user", "content": script
+        "role": "user", 
+        "content": f"Script: {script}"
     }]
+    
     response = call_openai_api(messages)
-    return response.choices[0].message['content']
+    return response.choices[0].message.content
 
-# Streamlit user interface
-st.title("Video Hook Generator")
+# Streamlit setup
+st.set_page_config(layout="wide", page_title="Video Hook Generator")
 
-# Text input for user to enter the script or video content theme
-user_input = st.text_area("Enter the script or theme for your video:", placeholder="Paste your script here or describe the theme.")
+# Initialize session state
+if "generated_hooks" not in st.session_state:
+    st.session_state["generated_hooks"] = None
+if "show_notice" not in st.session_state:
+    st.session_state["show_notice"] = False
 
-# Button to generate video hooks
-if st.button('Generate Hooks'):
-    with st.spinner('Generating creative hooks...'):
-        hooks = generate_video_hooks(user_input)
-        st.markdown("### Generated Hooks")
-        st.write(hooks)
+# Header
+st.markdown("<h1 style='text-align: center; color: black;'>🎣 Video Hook Generator</h1>", unsafe_allow_html=True)
+st.markdown("<p style='text-align: center; color: gray; font-size: 18px;'>Turn any script into scroll-stopping content</p>", unsafe_allow_html=True)
 
-# Note: Replace "your_openai_api_key_here" with your actual OpenAI API key or ensure it's set in st.secrets["OPENAI_API_KEY"].
+# Simple input section
+st.markdown("<h2 style='text-align: center; color: black;'>Paste Your Script</h2>", unsafe_allow_html=True)
+
+# Create a clean, centered input area
+col1, col2, col3 = st.columns([1, 3, 1])
+
+with col2:
+    user_script = st.text_area(
+        "",
+        placeholder="Paste your complete video script here...\n\nThe more detailed your script, the better hooks I can create for you.",
+        height=200
+    )
+    
+    generate_button = st.button('🎯 Generate 10 Hooks', use_container_width=True)
+
+# Process results
+if generate_button and user_script.strip():
+    st.session_state["show_notice"] = True
+    
+    with st.spinner("🎣 Creating irresistible hooks..."):
+        st.session_state["generated_hooks"] = generate_video_hooks(user_script)
+        st.session_state["show_notice"] = False
+
+elif generate_button and not user_script.strip():
+    st.error("Please paste your script first!")
+
+# Display waiting notice
+if st.session_state["show_notice"]:
+    st.info("🎬 Analyzing your script and generating hooks... This will just take a moment!")
+
+# Display results
+if st.session_state["generated_hooks"]:
+    st.markdown("---")
+    st.markdown("<h2 style='text-align: center; color: black;'>🔥 Your Hook Options</h2>", unsafe_allow_html=True)
+    
+    # Display hooks in a clean format
+    st.markdown("""
+    <div style='background-color: #f8f9fa; padding: 25px; border-radius: 10px; border-left: 4px solid #FF6B35; margin: 20px 0;'>
+    """, unsafe_allow_html=True)
+    
+    st.markdown(st.session_state["generated_hooks"])
+    
+    st.markdown("</div>", unsafe_allow_html=True)