Update app.py

app.py

@@ -7,24 +7,55 @@ from datetime import datetime
 # Your xAI API key
 XAI_API_KEY = os.environ.get('XAI_API_KEY')
 
-# Enhanced movie database
+# Expanded movie database with more genres
 MOVIE_TEMPLATES = {
     "epic_fantasy": {
         "title": "🏰 Epic Fantasy Quest",
         "examples": [
             "A hobbit's quest to destroy a powerful ring",
-            "A young wizard's journey to defeat a dark lord", 
+            "A young wizard's journey to defeat a dark lord",
         ]
     },
     "sci_fi_adventure": {
-        "title": "🚀 Sci-Fi Space Opera", 
+        "title": "🚀 Sci-Fi Space Opera",
         "examples": [
             "A starship crew exploring unknown galaxies",
             "First contact with an alien civilization"
         ]
+    },
+    "action_thriller": {
+        "title": "💥 High-Octane Action",
+        "examples": [
+            "A rogue agent's global chase against a shadow cabal",
+            "Undercover cop vs. cartel kingpin in neon-lit streets"
+        ]
+    },
+    "horror_mystery": {
+        "title": "👻 Chilling Horror",
+        "examples": [
+            "A haunted house awakens ancient curses",
+            "Small-town vanishings reveal a cult's dark secret"
+        ]
+    },
+    "rom_com": {
+        "title": "❤️ Romantic Comedy",
+        "examples": [
+            "Fake dating turns real amid wedding chaos",
+            "Rival chefs spark love in a culinary showdown"
+        ]
     }
 }
 
+# Top 2025 actors (sourced from box office & buzz)
+TOP_ACTORS = [
+    "Tom Holland", "Dwayne Johnson", "Tom Cruise", "Chris Hemsworth", "Pedro Pascal",
+    "Robert Downey Jr.", "Will Smith", "Jason Statham", "Rachel Zegler", "Eiza González",
+    "Monica Barbaro", "Sophie Wilde", "Joseph Quinn", "David Corenswet"
+]
+
+# MPAA ratings
+MPAA_RATINGS = ["G", "PG", "PG-13", "R", "NC-17"]
+
 def test_connection():
     """Test xAI connection with better error handling"""
     if not XAI_API_KEY:
@@ -40,32 +71,48 @@ def test_connection():
     except Exception as e:
         return f"❌ Connection error: {str(e)}", "🔴 Offline"
 
-def generate_cinematic_story(original_story, sequel_start, story_type="epic_fantasy"):
-    """Generate a cinematic story with proper error handling"""
+def generate_cinematic_story(original_story, sequel_start, selected_actors, extras, story_type="epic_fantasy"):
+    """Generate a cinematic story with casting, extras, and rating"""
     
     if not original_story or not sequel_start:
         return "❌ Please fill in both story fields!"
     
+    # Cast lead/support
+    lead_actor = random.choice(selected_actors) if selected_actors else random.choice(TOP_ACTORS)
+    support_actors = [a for a in selected_actors if a != lead_actor][:2]
+    
+    # Random rating based on genre
+    rating = random.choice(["PG-13", "R"] if "horror" in story_type or "action" in story_type else MPAA_RATINGS)
+    
     headers = {
         'Authorization': f'Bearer {XAI_API_KEY}',
         'Content-Type': 'application/json'
     }
     
-    # Simpler, more reliable prompt
+    # Enhanced prompt with actors, extras, genre
+    extras_str = f" Include these elements: {extras}." if extras else ""
+    actors_str = f" Cast {lead_actor} as the lead hero. Feature {', '.join(support_actors)} in supporting roles.{extras_str}"
+    
     prompt = f"""
-    Create a movie sequel story:
+    Create a blockbuster movie sequel in the {story_type} genre (MPAA rating: {rating}):
 
     Original: {original_story}
     Sequel starts: {sequel_start}
+    {actors_str}
 
-    Write an engaging sequel with character development and exciting scenes.
+    Write an engaging, cinematic sequel synopsis (300-400 words) with:
+    - Strong character arcs for the lead ({lead_actor})
+    - Key scenes: 3-5 high-stakes action/emotional beats
+    - Plot twists from extras
+    - Blockbuster ending hook
+    Keep it vivid, dialogue-snappy, and ready for the big screen!
     """
     
     data = {
         'model': 'grok-beta',
         'messages': [{'role': 'user', 'content': prompt}],
-        'max_tokens': 500,
-        'temperature': 0.7
+        'max_tokens': 600,
+        'temperature': 0.8  # Bumped for creativity
     }
     
     try:
@@ -73,138 +120,197 @@ def generate_cinematic_story(original_story, sequel_start, story_type="epic_fant
             'https://api.x.ai/v1/chat/completions',
             headers=headers,
             json=data,
-            timeout=30  # Added timeout
+            timeout=30
         )
         
         if response.status_code == 200:
             result = response.json()
             story = result['choices'][0]['message']['content']
+        else:
+            story = _generate_fallback_story(original_story, sequel_start, lead_actor, support_actors, extras, story_type, rating)
             
-            production_id = f"PROD-{random.randint(1000,9999)}"
-            
-            return f"""
-🎬 **PRODUCTION MEMO** 🎬
-**Project ID:** {production_id}
-**Status:** 🟢 STORY APPROVED
+    except Exception as e:
+        story = _generate_fallback_story(original_story, sequel_start, lead_actor, support_actors, extras, story_type, rating)
+    
+    production_id = f"PROD-{random.randint(10000,99999)}-{datetime.now().strftime('%m%d')}"
+    
+    cast_section = f"**CAST HIGHLIGHTS:**\n• Lead: {lead_actor}\n" + "\n".join([f"• Support: {a}" for a in support_actors]) if support_actors else ""
+    
+    return f"""
+🎬 **PRODUCTION MEMO: {story_type.upper()} SEQUEL** 🎬
+**Project ID:** {production_id} | **Rating:** {rating} | **Genre:** {MOVIE_TEMPLATES[story_type]['title']}
+
+**{lead_actor}'s LEGEND CONTINUES...**
 
 {story}
 
 ---
-✨ **Your Conan sequel is ready!**
-            """
-        else:
-            return f"""
-🎬 **PRODUCTION UPDATE** 🎬
-**Status:** 🟡 IN DEVELOPMENT
-
-While our AI director sets up the epic scenes, here's your Conan sequel concept:
+**CAST SHEET:**
+{cast_section}
 
-**CONAN: THE FROZEN THRONE**
-
-Years after becoming king, Conan (Arnold Schwarzenegger) faces an ancient ice demon from the north. The frozen threat forces the aging barbarian to wield his sword once more in an epic battle for Aquilonia.
-
-**Key Scenes:**
-• Conan's council debate about the supernatural threat
-• Journey through frozen wastelands  
-• Epic battle with ice creatures
-• Final confrontation with the ice demon
-
-**Character Arc:** Older, wiser Conan learns that true strength comes from wisdom, not just muscle.
-            """
-            
-    except Exception as e:
-        return f"""
-🎬 **CREATIVE DEVELOPMENT** 🎬
-**Status:** 🟡 CONCEPT PHASE
-
-**CONAN RETURNS: FROZEN LEGACY**
+✨ Lights, camera, sequel! Export to script? 🚀
+    """
 
-Arnold Schwarzenegger reprises his role as Conan, now King of Aquilonia. When an ancient ice demon awakens and begins freezing kingdoms, Conan must journey north to confront this supernatural threat.
+def _generate_fallback_story(original_story, sequel_start, lead_actor, support_actors, extras, story_type, rating):
+    """Rich fallback story generation"""
+    extras_str = f" with twists like {extras}" if extras else ""
+    support_str = f" alongside {', '.join(support_actors)}" if support_actors else ""
+    
+    # Genre-tailored plot
+    if "fantasy" in story_type:
+        plot = f"Years after {original_story}, {lead_actor} reprises the scarred warrior-king, facing a cataclysmic rift summoning shadow beasts{support_str}. {extras_str} In blistering battles across fractured realms, {lead_actor} uncovers a betrayal that tests loyalty and reignites old flames. Key scenes: A throne-room ambush, a forbidden portal chase, and an earth-shaking demon duel. Arc: From jaded ruler to unbreakable legend."
+    elif "sci_fi" in story_type:
+        plot = f"Post-{original_story}, {lead_actor} commands a ragtag fleet against an AI overlord devouring stars{support_str}. {extras_str} Quantum jumps and zero-G dogfights ensue, revealing a personal clone conspiracy. Key scenes: Black hole evasion, holographic interrogations, epic wormhole assault. Arc: Doubtful captain to galaxy's defiant savior."
+    elif "action" in story_type:
+        plot = f"After {original_story}, {lead_actor} hunts a rogue syndicate in rain-slicked megacities{support_str}. {extras_str} Explosive set-pieces include subway heists and rooftop pursuits. Key scenes: Double-cross diner shootout, armored convoy takedown, skyline finale. Arc: Lone wolf to team-forged avenger."
+    elif "horror" in story_type:
+        plot = f"Haunted by {original_story}, {lead_actor} investigates viral nightmares plaguing the world{support_str}. {extras_str} Creeping dread builds to possessions and ritual showdowns. Key scenes: Foggy asylum lockdown, mirror-world traps, blood-moon ritual. Arc: Skeptic to harbinger of the damned."
+    else:  # rom_com
+        plot = f"Following {original_story}, {lead_actor} navigates love's chaos at a luxury resort{support_str}. {extras_str} Mishaps like mistaken identities and beachside confessions spark hilarity. Key scenes: Awkward speed-dating fiasco, midnight dance mix-up, heartfelt airport dash. Arc: Commitment-phobe to forever optimist."
+    
+    return f"**SYNOPSIS:** {plot}\n\n**Why It Works:** High-stakes{ ' + emotional depth' if 'rom' in story_type else ''}, perfect for {rating} crowds. Budget: $150M. Greenlight now!"
 
-The sequel explores:
-- Leadership vs. Warrior instincts
-- Supernatural threats in the Hyborian Age  
-- Conan's legacy as both king and barbarian
-- New allies and enemies in frozen lands
+def generate_movie_pitch(selected_actors=None):
+    """Generate a movie pitch with random cast if none selected"""
+    if not selected_actors:
+        selected_actors = random.sample(TOP_ACTORS, 3)
+    
+    genres = list(MOVIE_TEMPLATES.keys())
+    selected_genre = random.choice(genres)
+    template = MOVIE_TEMPLATES[selected_genre]
+    
+    lead = selected_actors[0]
+    supports = selected_actors[1:]
+    rating = random.choice(MPAA_RATINGS)
+    
+    logline = random.choice(template['examples'])
+    franchise_potential = '🟢 HIGH' if random.random() > 0.5 else '🟡 MEDIUM'
+    
+    pitch = f"""
+🎬 **GREENLIGHT PITCH DECK** 🎬
 
-Perfect for Arnold's return to the iconic role!
-        """
+**GENRE:** {template['title']} | **RATING:** {rating}
+**LOGLINE:** {logline} — Starring {lead} as the unbreakable hero, with {', '.join(supports)} lighting up the ensemble.
 
-def generate_movie_pitch():
-    """Generate a movie pitch with fallback"""
-    try:
-        genres = list(MOVIE_TEMPLATES.keys())
-        selected_genre = random.choice(genres)
-        template = MOVIE_TEMPLATES[selected_genre]
-        
-        pitch = f"""
-🎬 **GREENLIGHT OPPORTUNITY** 🎬
+**BUDGET:** ${random.randint(80,250)}M
+**TIMELINE:** {random.randint(18,36)} months to cameras
+**TARGET AUDIENCE:** {random.randint(13,35)}+ thrill-seekers/romantics
+**FRANCHISE POTENTIAL:** {franchise_potential}
+**HOOK:** Explosive opener + twisty mid-act + unforgettable finale. Dir. rec: Taika Waititi for {selected_genre}.
 
-**GENRE:** {template['title']}
-**LOGLINE:** {random.choice(template['examples'])}
-**BUDGET:** ${random.randint(50,200)}M
-**TIMELINE:** {random.randint(12,24)} months
+**QUICK CAST NOTES:**
+• {lead}: Lead (raw charisma + action chops)
+• {supports[0] if supports else 'TBD'}: Comic relief/foil
+• {supports[1] if len(supports) > 1 else 'TBD'}: Emotional anchor
 
-**AUDIENCE:** {random.randint(18,45)}+ demographic
-**FRANCHISE:** {'🟢 HIGH' if random.random() > 0.5 else '🟡 MEDIUM'}
-**READINESS:** 🟢 IMMEDIATE
-        """
-        return pitch
-    except:
-        return "🎬 New movie concept ready for development!"
+🟢 **READY FOR DEVELOPMENT?** This one's a box-office beast!
+    """
+    return pitch
 
-# Simplified Interface
-with gr.Blocks(title="IdeaForge Studio", theme=gr.themes.Soft()) as demo:
+# Polished Interface
+with gr.Blocks(title="IdeaForge Studio v2", theme=gr.themes.Base(primary_hue="amber")) as demo:
     
-    gr.Markdown("# 🎬 IdeaForge Studio")
-    gr.Markdown("### Your AI Co-writer for Blockbuster Stories")
+    gr.Markdown("""
+    # 🎬 **IdeaForge Studio v2** 
+    ### AI-Powered Blockbuster Builder | Cast Stars, Rate It, Pitch It!
+    *Powered by xAI | Generate sequels & pitches with 2025's top talent.*
+    """)
     
-    # Status
+    # Status Row
     with gr.Row():
-        status_btn = gr.Button("📡 Check Connection")
-        status_display = gr.Textbox(label="Status", show_label=False)
+        with gr.Column(scale=1):
+            status_btn = gr.Button("📡 Test xAI Connection", variant="secondary")
+        with gr.Column(scale=4):
+            status_msg = gr.Textbox(label="Connection Log", show_label=False, lines=1)
+            status_icon = gr.Textbox(label="Status", show_label=False, max_lines=1)
     
     # Main Tabs
-    with gr.Tab("🎞️ Story Generator"):
-        gr.Markdown("### Create Movie Sequels")
+    with gr.Tab("🎞️ Sequel Forge"):
+        gr.Markdown("### Craft Epic Sequels with Custom Cast")
         
-        original_input = gr.Textbox(
-            label="Original Story", 
-            placeholder="e.g., Conan the Barbarian seeks revenge...",
-            lines=2
-        )
+        with gr.Row():
+            with gr.Column(scale=1):
+                original_input = gr.Textbox(
+                    label="Original Movie Logline",
+                    placeholder="e.g., Conan slays Thulsa Doom in barbaric revenge...",
+                    lines=2
+                )
+                sequel_input = gr.Textbox(
+                    label="Sequel Kickoff",
+                    placeholder="e.g., As King, Conan battles an ice demon's curse...",
+                    lines=2
+                )
+                story_type_dropdown = gr.Dropdown(
+                    choices=list(MOVIE_TEMPLATES.keys()),
+                    value="epic_fantasy",
+                    label="Genre Vibes"
+                )
+                extras_input = gr.Textbox(
+                    label="Add Twists/Elements",
+                    placeholder="e.g., time-travel betrayal, zombie allies",
+                    lines=1
+                )
+                actor_select = gr.CheckboxGroup(
+                    choices=TOP_ACTORS,
+                    label="Cast Your Stars (pick 1-3)",
+                    value=[TOP_ACTORS[0]],
+                    interactive=True
+                )
+                
+            with gr.Column(scale=1):
+                gr.Markdown("### Quick Load")
+                gr.Button("🗡️ Conan Classic").click(
+                    lambda: ([
+                        "Conan the Barbarian seeks revenge against Thulsa Doom for destroying his village",
+                        "Years later, King Conan faces an ancient ice demon awakening in the frozen north",
+                        "epic_fantasy",
+                        "magical artifacts, betrayal by advisor",
+                        ["Dwayne Johnson", "Pedro Pascal"]
+                    ]),
+                    outputs=[original_input, sequel_input, story_type_dropdown, extras_input, actor_select]
+                )
+                gr.Button("🚀 Star Wars Vibes").click(
+                    lambda: ([
+                        "A farm boy joins rebels to destroy the Death Star",
+                        "Post-victory, the hero uncovers a Sith family secret",
+                        "sci_fi_adventure",
+                        "force ghost mentors, twin reveal",
+                        ["Tom Holland", "Joseph Quinn"]
+                    ]),
+                    outputs=[original_input, sequel_input, story_type_dropdown, extras_input, actor_select]
+                )
         
-        sequel_input = gr.Textbox(
-            label="Sequel Starting Point",
-            placeholder="e.g., Years later, King Conan faces a new threat...", 
-            lines=2
+        generate_btn = gr.Button("🎬 Generate Full Sequel!", variant="primary", size="lg")
+        story_output = gr.Markdown(label="Your Blockbuster Bible", show_label=True)
+        
+        generate_btn.click(
+            generate_cinematic_story,
+            inputs=[original_input, sequel_input, actor_select, extras_input, story_type_dropdown],
+            outputs=story_output
         )
+    
+    with gr.Tab("💡 Pitch Roulette"):
+        gr.Markdown("### Instant Greenlight Concepts")
         
-        generate_btn = gr.Button("🎬 Generate Sequel", variant="primary")
-        story_output = gr.Textbox(label="Your Story", lines=10, show_copy_button=True)
+        with gr.Row():
+            pitch_actor_select = gr.CheckboxGroup(
+                choices=TOP_ACTORS,
+                label="Dream Cast (optional)",
+                value=[],
+                interactive=True
+            )
+            gr.Button("🎲 Roll Random Cast", variant="secondary").click(
+                lambda: random.sample(TOP_ACTORS, random.randint(1,3)),
+                outputs=pitch_actor_select
+            )
         
-        # Pre-fill Conan example
-        def conan_example():
-            return [
-                "Conan the Barbarian seeks revenge against Thulsa Doom for destroying his village",
-                "Years later, King Conan faces an ancient ice demon awakening in the frozen north"
-            ]
+        pitch_btn = gr.Button("✨ Generate Pitch Deck!", variant="primary")
+        pitch_output = gr.Markdown(label="Hot Pitch", show_label=True)
         
-        gr.Button("🗡️ Load Conan Example").click(
-            conan_example, 
-            outputs=[original_input, sequel_input]
-        )
-        
-        generate_btn.click(generate_cinematic_story, [original_input, sequel_input], story_output)
-    
-    with gr.Tab("💡 Quick Pitches"):
-        gr.Markdown("### Instant Movie Concepts")
-        pitch_btn = gr.Button("🎲 Generate Pitch")
-        pitch_output = gr.Textbox(label="Pitch Deck", lines=6, show_copy_button=True)
-        pitch_btn.click(generate_movie_pitch, outputs=pitch_output)
+        pitch_btn.click(generate_movie_pitch, inputs=[pitch_actor_select], outputs=pitch_output)
     
-    # Connection handling
-    status_btn.click(test_connection, outputs=status_display)
+    # Connection
+    status_btn.click(test_connection, outputs=[status_msg, status_icon])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)  # Enable public link for easy sharing