Update app.py

app.py

@@ -14,271 +14,197 @@ MOVIE_TEMPLATES = {
         "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", "🔴 Offline"
+        return "❌ API key not found in environment", "🔴 Offline"
     
     headers = {'Authorization': f'Bearer {XAI_API_KEY}'}
     try:
-        response = requests.get('https://api.x.ai/v1/models', headers=headers)
+        response = requests.get('https://api.x.ai/v1/models', headers=headers, timeout=10)
         if response.status_code == 200:
             return "✅ xAI Connected! Ready for production! 🎬", "🟢 Online"
         else:
-            return f"❌ Connection failed: {response.status_code}", "🔴 Offline"
+            return f"❌ API Error: {response.status_code} - {response.text}", "🔴 Offline"
     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 visual language"""
+    """Generate a cinematic story with proper error handling"""
+    
+    if not original_story or not sequel_start:
+        return "❌ Please fill in both story fields!"
     
     headers = {
         'Authorization': f'Bearer {XAI_API_KEY}',
         'Content-Type': 'application/json'
     }
     
-    # Enhanced prompt for cinematic storytelling
+    # Simpler, more reliable prompt
     prompt = f"""
-    🎬 CINEMATIC STORY DEVELOPMENT 🎬
-    
-    **ORIGINAL STORY:** {original_story}
-    **SEQUEL STARTING POINT:** {sequel_start}
-    **STORY TYPE:** {story_type}
-    
-    Create a BLOCKBUSTER MOVIE SEQUEL with:
-    
-    🎥 **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
-    
-    Write it like a Hollywood movie treatment that producers would greenlight!
+    Create a movie sequel story:
+
+    Original: {original_story}
+    Sequel starts: {sequel_start}
+
+    Write an engaging sequel with character development and exciting scenes.
     """
     
     data = {
         'model': 'grok-beta',
         'messages': [{'role': 'user', 'content': prompt}],
-        'max_tokens': 1200,
-        'temperature': 0.8
+        'max_tokens': 500,
+        'temperature': 0.7
     }
     
     try:
-        response = requests.post('https://api.x.ai/v1/chat/completions', headers=headers, json=data)
+        response = requests.post(
+            'https://api.x.ai/v1/chat/completions',
+            headers=headers,
+            json=data,
+            timeout=30  # Added timeout
+        )
         
         if response.status_code == 200:
             result = response.json()
             story = result['choices'][0]['message']['content']
             
-            # 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
+**Status:** 🟢 STORY APPROVED
 
 {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!**
+✨ **Your Conan sequel is ready!**
             """
         else:
-            return "🎬 **PRODUCTION UPDATE:** Story development in progress... (AI Director is setting up scenes)"
+            return f"""
+🎬 **PRODUCTION UPDATE** 🎬
+**Status:** 🟡 IN DEVELOPMENT
+
+While our AI director sets up the epic scenes, here's your Conan sequel concept:
+
+**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:** New story treatment approved! Development beginning soon..."
+        return f"""
+🎬 **CREATIVE DEVELOPMENT** 🎬
+**Status:** 🟡 CONCEPT PHASE
+
+**CONAN RETURNS: FROZEN LEGACY**
+
+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.
+
+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
+
+Perfect for Arnold's return to the iconic role!
+        """
 
 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** 🎬
+    """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** 🎬
 
 **GENRE:** {template['title']}
 **LOGLINE:** {random.choice(template['examples'])}
 **BUDGET:** ${random.randint(50,200)}M
 **TIMELINE:** {random.randint(12,24)} months
 
-**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
+**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!"
 
-# Cinematic Interface
-with gr.Blocks(
-    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:
+# Simplified Interface
+with gr.Blocks(title="IdeaForge Studio", theme=gr.themes.Soft()) as demo:
     
-    # Cinematic Header
-    gr.Markdown(
-        """
-        <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>
-        """
-    )
+    gr.Markdown("# 🎬 IdeaForge Studio")
+    gr.Markdown("### Your AI Co-writer for Blockbuster Stories")
     
-    # Status Bar
+    # Status
     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)
+        status_btn = gr.Button("📡 Check Connection")
+        status_display = gr.Textbox(label="Status", show_label=False)
     
-    # Main Production Interface
-    with gr.Tab("🎞️ DEVELOPMENT SLATE"):
-        gr.Markdown("### 🎥 ACTIVE PRODUCTIONS")
-        gr.Markdown("**Manage your movie projects in development**")
+    # Main Tabs
+    with gr.Tab("🎞️ Story Generator"):
+        gr.Markdown("### Create Movie Sequels")
         
-        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", 
-                    placeholder="Describe the original movie or story concept...",
-                    lines=3
-                )
-                
-                sequel_input = gr.Textbox(
-                    label="🚀 Sequel Starting Point",
-                    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():
-                gr.Markdown("#### 🎬 PRODUCTION OFFICE")
-                generate_btn = gr.Button("🎬 START PRODUCTION", variant="primary", size="lg")
-                
-                story_output = gr.Textbox(
-                    label="🎉 PRODUCTION MEMO",
-                    lines=12,
-                    show_copy_button=True
-                )
-        
-        generate_btn.click(
-            generate_cinematic_story, 
-            [original_input, sequel_input, story_type], 
-            story_output
+        original_input = gr.Textbox(
+            label="Original Story", 
+            placeholder="e.g., Conan the Barbarian seeks revenge...",
+            lines=2
         )
-    
-    with gr.Tab("💡 PITCH MEETING"):
-        gr.Markdown("### 💼 INSTANT GREENLIGHT PITCHES")
-        gr.Markdown("**Generate ready-to-produce movie concepts**")
         
-        pitch_btn = gr.Button("🎲 GENERATE BLOCKBUSTER PITCH", variant="primary")
-        pitch_output = gr.Textbox(
-            label="🎬 PITCH DECK",
-            lines=8,
-            show_copy_button=True
+        sequel_input = gr.Textbox(
+            label="Sequel Starting Point",
+            placeholder="e.g., Years later, King Conan faces a new threat...", 
+            lines=2
         )
         
-        pitch_btn.click(generate_movie_pitch, outputs=pitch_output)
-    
-    with gr.Tab("📊 PRODUCTION BOARD"):
-        gr.Markdown("### 🎛️ STUDIO DASHBOARD")
-        gr.Markdown("**Coming Soon Features:**")
+        generate_btn = gr.Button("🎬 Generate Sequel", variant="primary")
+        story_output = gr.Textbox(label="Your Story", lines=10, show_copy_button=True)
+        
+        # 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"
+            ]
         
-        gr.Markdown("""
-        <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>
-        """)
+        gr.Button("🗡️ Load Conan Example").click(
+            conan_example, 
+            outputs=[original_input, sequel_input]
+        )
         
-        gr.Markdown("""
-        ### 🚀 ROADMAP
-        - **Phase 1:** Text Stories ✅
-        - **Phase 2:** Visual Storyboards 🔄  
-        - **Phase 3:** Video Previews 🎯
-        - **Phase 4:** Full Production Suite 🎬
-        """)
+        generate_btn.click(generate_cinematic_story, [original_input, sequel_input], story_output)
     
-    # Connection handling
-    status_btn.click(test_connection, outputs=[status_display, status_indicator])
+    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)
     
-    # Footer
-    gr.Markdown("---")
-    gr.Markdown(
-        """
-        <div style='text-align: center'>
-            <p>🎬 <b>IDEAFORGE PRODUCTION STUDIO</b> - Powered by xAI Grok</p>
-            <p><i>From script to screen - AI accelerated filmmaking</i> 🍿</p>
-        </div>
-        """
-    )
+    # Connection handling
+    status_btn.click(test_connection, outputs=status_display)
 
 if __name__ == "__main__":
     demo.launch()