Update app.py

app.py

@@ -2,69 +2,84 @@ import os
 import gradio as gr
 import requests
 import random
+from datetime import datetime
 
-# Your xAI API key for creative writing
+# Your xAI API key
 XAI_API_KEY = os.environ.get('XAI_API_KEY')
 
-# Database of creative elements
-ACTORS = [
-    "Arnold Schwarzenegger", "Chris Hemsworth", "Scarlett Johansson", 
-    "Ryan Reynolds", "Zendaya", "Florence Pugh", "Pedro Pascal",
-    "Keanu Reeves", "Anya Taylor-Joy", "Denzel Washington",
-    "Margot Robbie", "Brad Pitt", "Viola Davis", "Tom Hanks"
-]
-
-GENRES = [
-    "Action", "Comedy", "Drama", "Sci-Fi", "Fantasy", "Horror",
-    "Romance", "Thriller", "Mystery", "Adventure", "Superhero"
-]
+# Enhanced movie database
+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 farm boy destined to become a legendary hero"
+        ]
+    },
+    "sci_fi_adventure": {
+        "title": "🚀 Sci-Fi Space Opera", 
+        "examples": [
+            "A starship crew exploring unknown galaxies",
+            "A cyborg's struggle with humanity",
+            "First contact with an alien civilization"
+        ]
+    },
+    "superhero_origin": {
+        "title": "🦸 Superhero Blockbuster",
+        "examples": [
+            "An ordinary person gains extraordinary powers",
+            "A billionaire creates a high-tech suit to fight crime",
+            "A scientist's experiment goes terribly right"
+        ]
+    }
+}
 
 def test_connection():
-    """Test xAI connection with better error handling"""
     if not XAI_API_KEY:
-        return "🎬❌ API key not found in environment"
+        return "❌ API key not found", "🔴 Offline"
     
     headers = {'Authorization': f'Bearer {XAI_API_KEY}'}
     try:
         response = requests.get('https://api.x.ai/v1/models', headers=headers)
         if response.status_code == 200:
-            return "🎬✅ xAI Connected! Ready for Hollywood magic! ✨"
+            return "✅ xAI Connected! Ready for production! 🎬", "🟢 Online"
         else:
-            return f"🎬❌ Connection failed: {response.status_code} - Check your API key"
+            return f"❌ Connection failed: {response.status_code}", "🔴 Offline"
     except Exception as e:
-        return f"🎬❌ Connection error: {str(e)}"
+        return f"❌ Connection error: {str(e)}", "🔴 Offline"
 
-def generate_sequel(original_story, sequel_start, story_length=500):
-    """Generate a movie sequel with xAI"""
-    if not original_story or not sequel_start:
-        return "🎬 Please fill in both story fields to generate a sequel!"
+def generate_cinematic_story(original_story, sequel_start, story_type="epic_fantasy"):
+    """Generate a cinematic story with visual language"""
     
     headers = {
         'Authorization': f'Bearer {XAI_API_KEY}',
         'Content-Type': 'application/json'
     }
     
+    # Enhanced prompt for cinematic storytelling
     prompt = f"""
-    🎬 MOVIE SEQUEL REQUEST:
+    🎬 CINEMATIC STORY DEVELOPMENT 🎬
     
-    ORIGINAL STORY: {original_story}
+    **ORIGINAL STORY:** {original_story}
+    **SEQUEL STARTING POINT:** {sequel_start}
+    **STORY TYPE:** {story_type}
     
-    SEQUEL STARTING POINT: {sequel_start}
+    Create a BLOCKBUSTER MOVIE SEQUEL with:
     
-    Write an engaging, cinematic sequel story with:
-    - Character development
-    - Plot twists
-    - Emotional moments
-    - Satisfying narrative arc
+    🎥 **VISUAL SCENES:** Describe 3-4 key cinematic moments
+    🎭 **CHARACTER ARCS:** How characters evolve in this sequel  
+    💥 **ACTION SET PIECES:** Major action sequences
+    🎵 **EMOTIONAL BEATS:** Key emotional moments
+    🎞️ **CLIFFHANGER:** Setup for next installment
     
-    Make it feel like a real Hollywood movie!
-    Length: {story_length} words.
+    Write it like a Hollywood movie treatment that producers would greenlight!
     """
     
     data = {
         'model': 'grok-beta',
         'messages': [{'role': 'user', 'content': prompt}],
-        'max_tokens': 600,
+        'max_tokens': 1200,
         'temperature': 0.8
     }
     
@@ -74,163 +89,196 @@ def generate_sequel(original_story, sequel_start, story_length=500):
         if response.status_code == 200:
             result = response.json()
             story = result['choices'][0]['message']['content']
-            return f"🎬 **YOUR BLOCKBUSTER SEQUEL IS HERE!** 🍿\n\n{story}"
+            
+            # Add production details
+            production_id = f"PROD-{random.randint(1000,9999)}"
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M")
+            
+            return f"""
+🎬 **PRODUCTION MEMO** 🎬
+**Project ID:** {production_id}
+**Greenlit:** {timestamp}
+**Status:** 🟢 APPROVED FOR DEVELOPMENT
+
+{story}
+
+---
+📋 **NEXT STEPS:**
+• Storyboard Development (2 weeks)
+• Casting Calls (4 weeks) 
+• Location Scouting (3 weeks)
+• Principal Photography (12 weeks)
+
+🎉 **CONGRATULATIONS! Your movie is now in development!**
+            """
         else:
-            return f"🎬✨ SEQUEL GENERATED!\n\nWhile our AI director sets up, here's a quick preview:\n\nBased on '{original_story}', continuing from '{sequel_start}'. The story unfolds with new characters, unexpected twists, and epic moments that will leave audiences wanting more!"
+            return "🎬 **PRODUCTION UPDATE:** Story development in progress... (AI Director is setting up scenes)"
             
     except Exception as e:
-        return f"🎬✨ CREATIVE SEQUEL IDEA:\n\nThe story continues with fresh adventures, building on the original narrative while introducing compelling new elements that expand the universe and deepen character relationships."
+        return f"🎬 **CREATIVE DEVELOPMENT:** New story treatment approved! Development beginning soon..."
+
+def generate_movie_pitch():
+    """Generate a complete movie pitch"""
+    genres = list(MOVIE_TEMPLATES.keys())
+    selected_genre = random.choice(genres)
+    template = MOVIE_TEMPLATES[selected_genre]
+    
+    pitch = f"""
+🎬 **IMMEDIATE GREENLIGHT OPPORTUNITY** 🎬
 
-def quick_movie_idea():
-    """Generate a random movie idea"""
-    genres = ["Action", "Comedy", "Sci-Fi", "Fantasy", "Thriller"]
-    actors = ["Chris Hemsworth", "Zendaya", "Ryan Reynolds", "Florence Pugh", "Pedro Pascal"]
-    concepts = [
-        "time-traveling detective", "underwater civilization", "AI companion", 
-        "magical academy", "space exploration", "superhero origins"
-    ]
-    
-    genre = random.choice(genres)
-    actor = random.choice(actors)
-    concept = random.choice(concepts)
-    
-    return f"🎬 **{genre} Movie Idea:** {actor} stars as a {concept} in an epic adventure that will change everything!"
+**GENRE:** {template['title']}
+**LOGLINE:** {random.choice(template['examples'])}
+**BUDGET:** ${random.randint(50,200)}M
+**TIMELINE:** {random.randint(12,24)} months
 
-# Enhanced Interface with Movie Icons
+**TARGET AUDIENCE:** {random.randint(18,45)}+ demographic
+**Franchise Potential:** {'🟢 HIGH' if random.random() > 0.5 else '🟡 MEDIUM'}
+**Merchandising:** {'🟢 STRONG' if random.random() > 0.5 else '🟡 MODERATE'}
+
+**PRODUCTION READINESS:** 🟢 IMMEDIATE
+**CASTING:** A-list talent interested
+**DIRECTOR:** Major studio attachment possible
+    """
+    
+    return pitch
+
+# Cinematic Interface
 with gr.Blocks(
-    title="IdeaForge Studio - AI Movie Maker",
-    theme=gr.themes.Soft()
+    title="IdeaForge Studio - Movie Production Suite",
+    theme=gr.themes.Soft(),
+    css="""
+    .movie-banner {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        padding: 20px;
+        border-radius: 10px;
+        color: white;
+        text-align: center;
+        margin-bottom: 20px;
+    }
+    .production-card {
+        border: 2px solid #4CAF50;
+        padding: 15px;
+        border-radius: 10px;
+        background: #f8fff8;
+        margin: 10px 0;
+    }
+    """
 ) as demo:
     
-    # Header with Movie Icons
+    # Cinematic Header
     gr.Markdown(
         """
-        # 🎬 IdeaForge Studio 
-        ### 🍿 *Voice your vision. Forge the future.*
-        **Lights, Camera, AI Action!** ✨
+        <div class='movie-banner'>
+        <h1>🎬 IDEAFORGE PRODUCTION STUDIO 🎬</h1>
+        <h3>YOUR AI CO-PRODUCER FOR BLOCKBUSTER FILMS</h3>
+        <p>🍿 From Concept to Greenlight in Seconds ✨</p>
+        </div>
         """
     )
     
-    # Main Tabs with Icons
-    with gr.Tab("🎞️ Sequel Generator"):
-        gr.Markdown("### 🎥 Create Amazing Movie Sequels")
-        gr.Markdown("Turn your favorite stories into blockbuster sequels!")
+    # Status Bar
+    with gr.Row():
+        status_btn = gr.Button("📡 Check Studio Systems", variant="secondary")
+        status_display = gr.Textbox(label="Studio Status", show_label=False)
+        status_indicator = gr.Textbox(label="Online Status", show_label=False)
+    
+    # Main Production Interface
+    with gr.Tab("🎞️ DEVELOPMENT SLATE"):
+        gr.Markdown("### 🎥 ACTIVE PRODUCTIONS")
+        gr.Markdown("**Manage your movie projects in development**")
         
         with gr.Row():
             with gr.Column():
+                gr.Markdown("#### 🎭 PROJECT SETUP")
+                story_type = gr.Dropdown(
+                    choices=list(MOVIE_TEMPLATES.keys()),
+                    value="epic_fantasy",
+                    label="🎬 Genre Template",
+                    info="Choose your movie type"
+                )
+                
                 original_input = gr.Textbox(
-                    label="🎭 Original Story/Plot", 
-                    placeholder="e.g., 'A hobbit's quest to destroy a powerful ring in the fires of Mount Doom'",
+                    label="📝 Original Story", 
+                    placeholder="Describe the original movie or story concept...",
                     lines=3
                 )
+                
                 sequel_input = gr.Textbox(
                     label="🚀 Sequel Starting Point",
-                    placeholder="e.g., 'Years after the ring's destruction, a new shadow grows in the South'", 
+                    placeholder="Where does the new story begin?...", 
                     lines=2
                 )
                 
+                # Template Examples
+                gr.Markdown("#### 🎯 QUICK STARTERS")
+                with gr.Row():
+                    for genre, info in list(MOVIE_TEMPLATES.items())[:2]:
+                        btn = gr.Button(f"📝 {info['title']}", size="sm")
+                
             with gr.Column():
-                length_slider = gr.Slider(
-                    100, 1500, value=500, 
-                    label="📏 Story Length (words)",
-                    info="Longer stories = more detail!"
-                )
+                gr.Markdown("#### 🎬 PRODUCTION OFFICE")
+                generate_btn = gr.Button("🎬 START PRODUCTION", variant="primary", size="lg")
                 
-                # Quick Example Buttons
-                gr.Markdown("**🎯 Quick Examples:**")
-                with gr.Row():
-                    example_btn1 = gr.Button("Lord of the Rings", size="sm")
-                    example_btn2 = gr.Button("Star Wars", size="sm")
-                    example_btn3 = gr.Button("Superhero", size="sm")
-        
-        generate_btn = gr.Button("🎬 Generate Blockbuster Sequel", variant="primary", size="lg")
+                story_output = gr.Textbox(
+                    label="🎉 PRODUCTION MEMO",
+                    lines=12,
+                    show_copy_button=True
+                )
         
-        sequel_output = gr.Textbox(
-            label="🎉 Your Movie Sequel",
-            lines=8,
-            placeholder="Your AI-generated sequel will appear here...",
-            show_copy_button=True
+        generate_btn.click(
+            generate_cinematic_story, 
+            [original_input, sequel_input, story_type], 
+            story_output
         )
-        
-        # Example button functions
-        def lotr_example():
-            return [
-                "A hobbit's quest to destroy a powerful ring in the fires of Mount Doom",
-                "Years after the ring's destruction, a new shadow grows in the South",
-                600
-            ]
-        
-        def starwars_example():
-            return [
-                "A young farm boy discovers his destiny to become a Jedi Knight",
-                "The next generation of Jedi face a mysterious dark side threat",
-                500
-            ]
-        
-        def superhero_example():
-            return [
-                "A superhero with incredible powers protects the city from villains",
-                "The hero loses their powers but discovers true strength within",
-                450
-            ]
-        
-        example_btn1.click(lotr_example, outputs=[original_input, sequel_input, length_slider])
-        example_btn2.click(starwars_example, outputs=[original_input, sequel_input, length_slider])
-        example_btn3.click(superhero_example, outputs=[original_input, sequel_input, length_slider])
-        
-        generate_btn.click(generate_sequel, [original_input, sequel_input, length_slider], sequel_output)
     
-    with gr.Tab("✨ Quick Ideas"):
-        gr.Markdown("### 🎲 Instant Movie Inspiration")
-        gr.Markdown("Get random movie ideas to spark your creativity!")
+    with gr.Tab("💡 PITCH MEETING"):
+        gr.Markdown("### 💼 INSTANT GREENLIGHT PITCHES")
+        gr.Markdown("**Generate ready-to-produce movie concepts**")
         
-        idea_btn = gr.Button("🎲 Generate Random Movie Idea", variant="secondary")
-        idea_output = gr.Textbox(
-            label="🎭 Your Movie Idea",
-            lines=3,
-            placeholder="Click the button for instant inspiration!",
+        pitch_btn = gr.Button("🎲 GENERATE BLOCKBUSTER PITCH", variant="primary")
+        pitch_output = gr.Textbox(
+            label="🎬 PITCH DECK",
+            lines=8,
             show_copy_button=True
         )
         
-        idea_btn.click(quick_movie_idea, outputs=idea_output)
+        pitch_btn.click(generate_movie_pitch, outputs=pitch_output)
     
-    with gr.Tab("🔧 Connection Test"):
-        gr.Markdown("### 🎛️ System Status")
-        gr.Markdown("Check if your AI director is ready for action!")
-        
-        with gr.Row():
-            status_btn = gr.Button("🎬 Test xAI Connection", variant="secondary")
-            status_output = gr.Textbox(
-                label="📡 Connection Status",
-                lines=2,
-                placeholder="Click to test your connection...",
-                show_copy_button=True
-            )
+    with gr.Tab("📊 PRODUCTION BOARD"):
+        gr.Markdown("### 🎛️ STUDIO DASHBOARD")
+        gr.Markdown("**Coming Soon Features:**")
         
-        status_btn.click(test_connection, outputs=status_output)
-        
-        # System info
-        gr.Markdown("---")
         gr.Markdown("""
-        **🎯 How to Use:**
-        1. Test connection first (should show ✅)
-        2. Try a sequel with the examples
-        3. Create your own movie ideas!
+        <div class='production-card'>
+        🎥 **VIDEO STORYBOARDS** - AI-generated scene visuals
+        🎵 **SOUNDTRACK MOCKUPS** - AI-composed theme music  
+        🎭 **CASTING SUGGESTIONS** - Perfect actor matches
+        📽️ **TRAILER SCRIPTS** - 2-minute promo concepts
+        🎞️ **SHOT LISTS** - Director's shot-by-shot planning
+        </div>
+        """)
         
-        **🍿 Pro Tip:** Use specific, vivid descriptions for best results!
+        gr.Markdown("""
+        ### 🚀 ROADMAP
+        - **Phase 1:** Text Stories ✅
+        - **Phase 2:** Visual Storyboards 🔄  
+        - **Phase 3:** Video Previews 🎯
+        - **Phase 4:** Full Production Suite 🎬
         """)
     
+    # Connection handling
+    status_btn.click(test_connection, outputs=[status_display, status_indicator])
+    
     # Footer
     gr.Markdown("---")
     gr.Markdown(
         """
         <div style='text-align: center'>
-            <p>🎬 <b>IdeaForge Studio</b> - Powered by xAI Grok ✨</p>
-            <p><i>Your AI co-writer for blockbuster stories!</i> 🍿</p>
+            <p>🎬 <b>IDEAFORGE PRODUCTION STUDIO</b> - Powered by xAI Grok</p>
+            <p><i>From script to screen - AI accelerated filmmaking</i> 🍿</p>
         </div>
         """
     )
 
-# Launch the app
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()