Update app.py

app.py

@@ -2,12 +2,15 @@ import os
 import gradio as gr
 import requests
 import random
+import time
 from datetime import datetime
+from typing import Optional
 
-# Your xAI API key
+# API Keys
 XAI_API_KEY = os.environ.get('XAI_API_KEY')
+RUNWAY_API_KEY = os.environ.get('RUNWAY_API_KEY')
 
-# Expanded movie database with more genres
+# Expanded movie database (unchanged)
 MOVIE_TEMPLATES = {
     "epic_fantasy": {
         "title": "🏰 Epic Fantasy Quest",
@@ -46,18 +49,18 @@ MOVIE_TEMPLATES = {
     }
 }
 
-# Top 2025 actors (sourced from box office & buzz)
+# Top 2025 actors (unchanged)
 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 (unchanged)
 MPAA_RATINGS = ["G", "PG", "PG-13", "R", "NC-17"]
 
 def test_connection():
-    """Test xAI connection with better error handling"""
+    """Test xAI connection (unchanged)"""
     if not XAI_API_KEY:
         return "❌ API key not found in environment", "🔴 Offline"
     
@@ -71,9 +74,25 @@ def test_connection():
     except Exception as e:
         return f"❌ Connection error: {str(e)}", "🔴 Offline"
 
-def generate_cinematic_story(original_story, sequel_start, selected_actors, extras, story_type="epic_fantasy"):
-    """Generate a cinematic story with casting, extras, and rating"""
+def test_runway_connection():
+    """Test Runway ML connection"""
+    if not RUNWAY_API_KEY:
+        return "❌ Runway API key not found", "🔴 Offline"
     
+    headers = {'Authorization': f'Bearer {RUNWAY_API_KEY}', 'Content-Type': 'application/json'}
+    base_url = 'https://api.runwayml.com/v1'
+    try:
+        # Simple ping: GET /tasks (or models if available; using a safe endpoint)
+        response = requests.get(f"{base_url}/tasks", headers=headers, timeout=10)
+        if response.status_code == 200 or response.status_code == 401:  # 401 ok for auth test
+            return "✅ Runway Connected! Videos incoming... 🎞️", "🟢 Online"
+        else:
+            return f"❌ Runway Error: {response.status_code}", "🔴 Offline"
+    except Exception as e:
+        return f"❌ Runway Connection error: {str(e)}", "🔴 Offline"
+
+def generate_cinematic_story(original_story, sequel_start, selected_actors, extras, story_type="epic_fantasy"):
+    """Generate story (unchanged from v2)"""
     if not original_story or not sequel_start:
         return "❌ Please fill in both story fields!"
     
@@ -112,7 +131,7 @@ def generate_cinematic_story(original_story, sequel_start, selected_actors, extr
         'model': 'grok-beta',
         'messages': [{'role': 'user', 'content': prompt}],
         'max_tokens': 600,
-        'temperature': 0.8  # Bumped for creativity
+        'temperature': 0.8
     }
     
     try:
@@ -136,6 +155,9 @@ def generate_cinematic_story(original_story, sequel_start, selected_actors, extr
     
     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 story with trailer prompt extraction (key scene for video)
+    trailer_prompt = story.split('Key scenes:')[1].split('.')[0].strip() if 'Key scenes:' in story else f"Epic {story_type} battle featuring {lead_actor}"
+    
     return f"""
 🎬 **PRODUCTION MEMO: {story_type.upper()} SEQUEL** 🎬
 **Project ID:** {production_id} | **Rating:** {rating} | **Genre:** {MOVIE_TEMPLATES[story_type]['title']}
@@ -148,15 +170,15 @@ def generate_cinematic_story(original_story, sequel_start, selected_actors, extr
 **CAST SHEET:**
 {cast_section}
 
-✨ Lights, camera, sequel! Export to script? 🚀
-    """
+✨ Lights, camera, sequel! Ready for trailer? 🚀
+**Trailer Prompt:** {trailer_prompt}
+    """, trailer_prompt
 
 def _generate_fallback_story(original_story, sequel_start, lead_actor, support_actors, extras, story_type, rating):
-    """Rich fallback story generation"""
+    """Fallback (unchanged)"""
     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:
@@ -168,10 +190,10 @@ def _generate_fallback_story(original_story, sequel_start, lead_actor, support_a
     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!"
+    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!\n**Key scenes:** {plot.split('Key scenes:')[1] if 'Key scenes:' in plot else 'Epic showdown.'}"
 
 def generate_movie_pitch(selected_actors=None):
-    """Generate a movie pitch with random cast if none selected"""
+    """Pitch (unchanged)"""
     if not selected_actors:
         selected_actors = random.sample(TOP_ACTORS, 3)
     
@@ -186,6 +208,8 @@ def generate_movie_pitch(selected_actors=None):
     logline = random.choice(template['examples'])
     franchise_potential = '🟢 HIGH' if random.random() > 0.5 else '🟡 MEDIUM'
     
+    trailer_prompt = f"{logline} - cinematic trailer shot with dramatic music."
+    
     pitch = f"""
 🎬 **GREENLIGHT PITCH DECK** 🎬
 
@@ -204,27 +228,88 @@ def generate_movie_pitch(selected_actors=None):
 • {supports[1] if len(supports) > 1 else 'TBD'}: Emotional anchor
 
 🟢 **READY FOR DEVELOPMENT?** This one's a box-office beast!
+**Trailer Prompt:** {trailer_prompt}
     """
-    return pitch
+    return pitch, trailer_prompt
 
-# Polished Interface
-with gr.Blocks(title="IdeaForge Studio v2", theme=gr.themes.Base(primary_hue="amber")) as demo:
+def generate_video_trailer(prompt: str, duration: int = 5) -> str:
+    """Generate video trailer using Runway ML API"""
+    if not RUNWAY_API_KEY:
+        # Fallback: Placeholder video URL (e.g., stock trailer)
+        return "https://example.com/placeholder-trailer.mp4", "No Runway key—using demo clip. Add RUNWAY_API_KEY for real gen!"
+    
+    headers = {
+        'Authorization': f'Bearer {RUNWAY_API_KEY}',
+        'Content-Type': 'application/json'
+    }
+    base_url = 'https://api.runwayml.com/v1'
+    
+    # Create task (text-to-video params based on Runway docs)
+    create_data = {
+        "model": "gen3a_turbo",  # Latest text-to-video model (2025)
+        "text_prompt": prompt + " | Cinematic trailer style, dramatic lighting, high-energy action, epic music swell.",
+        "seconds": duration,
+        "aspect_ratio": "16:9",
+        "turbo": True,  # Faster gen
+        "explore_mode": True  # More creative
+    }
+    
+    try:
+        # Start task
+        create_resp = requests.post(f"{base_url}/text_to_video", headers=headers, json=create_data, timeout=30)
+        create_resp.raise_for_status()
+        task_data = create_resp.json()
+        task_id = task_data.get('id') or task_data.get('task_id')
+        if not task_id:
+            raise ValueError("No task ID in response")
+        
+        # Poll for completion (every 10s, up to 5 min)
+        status = "processing"
+        for _ in range(30):  # ~5 min max
+            time.sleep(10)
+            status_resp = requests.get(f"{base_url}/tasks/{task_id}", headers=headers, timeout=10)
+            status_resp.raise_for_status()
+            status_data = status_resp.json()
+            status = status_data.get('status')
+            
+            if status == 'succeeded':
+                # Extract video URL
+                output = status_data.get('output', {})
+                video_url = output.get('video_url') or output.get('assets', [{}])[0].get('url')
+                if video_url:
+                    return video_url, f"✅ Trailer generated! ({duration}s clip)"
+                else:
+                    raise ValueError("No video URL in output")
+            elif status == 'failed':
+                error_msg = status_data.get('error', 'Unknown error')
+                raise Exception(f"Generation failed: {error_msg}")
+        
+        raise TimeoutError("Generation timed out after 5 min")
+        
+    except Exception as e:
+        return None, f"❌ Error: {str(e)}. Check key/credits. Fallback: Imagine epic footage here! 🎞️"
+
+# Polished Interface (v3 with Video Tab)
+with gr.Blocks(title="IdeaForge Studio v3", theme=gr.themes.Base(primary_hue="amber")) as demo:
     
     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.*
+    # 🎬 **IdeaForge Studio v3** 
+    ### AI Blockbusters: Stories, Pitches & Now VIDEO TRAILERS!
+    *xAI for text + Runway ML for clips | Oct 2025 Edition.*
     """)
     
-    # Status Row
+    # Status Row (added Runway test)
     with gr.Row():
         with gr.Column(scale=1):
-            status_btn = gr.Button("📡 Test xAI Connection", variant="secondary")
+            xai_btn = gr.Button("📡 Test xAI", variant="secondary")
+            runway_btn = gr.Button("🎞️ Test Runway", 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)
+            xai_msg = gr.Textbox(label="xAI Log", show_label=False, lines=1)
+            xai_icon = gr.Textbox(label="xAI Status", show_label=False, max_lines=1)
+            runway_msg = gr.Textbox(label="Runway Log", show_label=False, lines=1)
+            runway_icon = gr.Textbox(label="Runway Status", show_label=False, max_lines=1)
     
-    # Main Tabs
+    # Story Tab (updated to return prompt for trailer)
     with gr.Tab("🎞️ Sequel Forge"):
         gr.Markdown("### Craft Epic Sequels with Custom Cast")
         
@@ -281,14 +366,16 @@ with gr.Blocks(title="IdeaForge Studio v2", theme=gr.themes.Base(primary_hue="am
                 )
         
         generate_btn = gr.Button("🎬 Generate Full Sequel!", variant="primary", size="lg")
-        story_output = gr.Markdown(label="Your Blockbuster Bible", show_label=True)
+        story_output = gr.Markdown(label="Your Blockbuster Bible")
+        trailer_prompt_hidden = gr.Textbox(visible=False)  # Hidden for chaining
         
         generate_btn.click(
             generate_cinematic_story,
             inputs=[original_input, sequel_input, actor_select, extras_input, story_type_dropdown],
-            outputs=story_output
+            outputs=[story_output, trailer_prompt_hidden]
         )
     
+    # Pitch Tab (updated to return prompt)
     with gr.Tab("💡 Pitch Roulette"):
         gr.Markdown("### Instant Greenlight Concepts")
         
@@ -305,12 +392,47 @@ with gr.Blocks(title="IdeaForge Studio v2", theme=gr.themes.Base(primary_hue="am
             )
         
         pitch_btn = gr.Button("✨ Generate Pitch Deck!", variant="primary")
-        pitch_output = gr.Markdown(label="Hot Pitch", show_label=True)
+        pitch_output = gr.Markdown(label="Hot Pitch")
+        pitch_prompt_hidden = gr.Textbox(visible=False)
         
-        pitch_btn.click(generate_movie_pitch, inputs=[pitch_actor_select], outputs=pitch_output)
+        pitch_btn.click(generate_movie_pitch, inputs=[pitch_actor_select], outputs=[pitch_output, pitch_prompt_hidden])
+    
+    # NEW: Video Trailer Tab
+    with gr.Tab("🎥 Trailer Forge"):
+        gr.Markdown("### AI-Generated Video Trailers via Runway ML")
+        gr.Markdown("*Paste a story prompt below—watch it come alive! ~2-5 min gen time.*")
+        
+        trailer_prompt_input = gr.Textbox(
+            label="Trailer Prompt",
+            placeholder="e.g., King Conan battles ice demons in frozen wastelands, epic sword fight...",
+            lines=3,
+            value="Cinematic fantasy battle in snowy mountains"
+        )
+        duration_slider = gr.Slider(5, 10, value=5, step=1, label="Clip Length (seconds)")
+        gen_video_btn = gr.Button("🚀 Generate Trailer!", variant="primary")
+        
+        video_output = gr.Video(label="Your AI Trailer", show_download_button=True)
+        video_status = gr.Textbox(label="Status", show_label=False, lines=1)
+        
+        # Chain from other tabs? Optional examples
+        gr.Examples(
+            examples=[
+                ["Epic space battle with starships exploding in nebula clouds"],
+                ["Romantic comedy meet-cute in rainy Paris streets, slow-motion smiles"],
+                ["Horror chase through haunted mansion, flickering lights and shadows"]
+            ],
+            inputs=trailer_prompt_input
+        )
+        
+        gen_video_btn.click(
+            generate_video_trailer,
+            inputs=[trailer_prompt_input, duration_slider],
+            outputs=[video_output, video_status]
+        )
     
-    # Connection
-    status_btn.click(test_connection, outputs=[status_msg, status_icon])
+    # Connections
+    xai_btn.click(test_connection, outputs=[xai_msg, xai_icon])
+    runway_btn.click(test_runway_connection, outputs=[runway_msg, runway_icon])
 
 if __name__ == "__main__":
-    demo.launch(share=True)  # Enable public link for easy sharing
+    demo.launch(share=True)