Update app.py

app.py

@@ -19,290 +19,218 @@ GENRES = [
     "Romance", "Thriller", "Mystery", "Adventure", "Superhero"
 ]
 
-HERO_ARCHETYPES = [
-    "Reluctant Hero", "Chosen One", "Anti-Hero", "Underdog",
-    "Tragic Hero", "Epic Hero", "Everyman Hero", "Super Soldier"
-]
-
-VILLAIN_TYPES = [
-    "Mad Scientist", "Ancient Evil", "Corrupt Leader", "Supervillain",
-    "Betrayed Friend", "Corporate Villain", "Supernatural Threat"
-]
-
-PLOT_TROPES = [
-    "Race Against Time", "Quest for Artifact", "Revenge Story",
-    "Save the World", "Personal Redemption", "Discovery of Powers",
-    "Fish Out of Water", "Buddy Adventure", "Heist Plot"
-]
-
-def generate_sequel(original_story, sequel_start, main_actor, supporting_actors, rating, story_length=500):
-    headers = {
-        'Authorization': f'Bearer {XAI_API_KEY}',
-        'Content-Type': 'application/json'
-    }
-    
-    cast_text = f"Starring {main_actor} as the main character."
-    if supporting_actors:
-        cast_text += f" Supporting cast: {supporting_actors}."
-    
-    rating_guides = {
-        "G": "Family-friendly, no violence, no strong language",
-        "PG": "Mild violence, minimal language, some thematic elements",
-        "PG-13": "Moderate violence, some language, intense sequences",
-        "R": "Strong violence, language, adult themes",
-        "NC-17": "Explicit content, adult-only themes"
-    }
-    
-    prompt = f"""
-    ORIGINAL STORY: {original_story}
-    SEQUEL STARTING POINT: {sequel_start}
-    CAST: {cast_text}
-    RATING: {rating} - {rating_guides[rating]}
-    
-    Write an engaging sequel story with character development, plot twists, and content appropriate for {rating} rating.
-    Length: {story_length} words.
-    """
-    
-    data = {
-        'model': 'grok-beta',
-        'messages': [{'role': 'user', 'content': prompt}],
-        'max_tokens': 800,
-        'temperature': 0.8
-    }
+def test_connection():
+    """Test xAI connection with better error handling"""
+    if not XAI_API_KEY:
+        return "🎬❌ API key not found in environment"
     
+    headers = {'Authorization': f'Bearer {XAI_API_KEY}'}
     try:
-        response = requests.post('https://api.x.ai/v1/chat/completions', headers=headers, json=data)
+        response = requests.get('https://api.x.ai/v1/models', headers=headers)
         if response.status_code == 200:
-            result = response.json()
-            return result['choices'][0]['message']['content']
+            return "🎬✅ xAI Connected! Ready for Hollywood magic! ✨"
         else:
-            return f"🎬 {rating}-rated sequel casting {main_actor}!"
+            return f"🎬❌ Connection failed: {response.status_code} - Check your API key"
     except Exception as e:
-        return f"✨ {rating}-rated sequel featuring {main_actor}!"
+        return f"🎬❌ Connection error: {str(e)}"
 
-def generate_original_movie(genre, main_actor, hero_type, villain_type, plot_trope, supporting_actors, rating, custom_prompt=""):
+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!"
+    
     headers = {
         'Authorization': f'Bearer {XAI_API_KEY}',
         'Content-Type': 'application/json'
     }
     
-    # Build story elements
-    story_elements = f"""
-    GENRE: {genre}
-    MAIN ACTOR: {main_actor} as a {hero_type}
-    VILLAIN: {villain_type}
-    PLOT: {plot_trope}
-    RATING: {rating}
-    """
+    prompt = f"""
+    🎬 MOVIE SEQUEL REQUEST:
     
-    if supporting_actors:
-        story_elements += f"SUPPORTING CAST: {supporting_actors}"
-    if custom_prompt:
-        story_elements += f"ADDITIONAL NOTES: {custom_prompt}"
+    ORIGINAL STORY: {original_story}
     
-    prompt = f"""
-    Create an original movie story based on these elements:
-    {story_elements}
+    SEQUEL STARTING POINT: {sequel_start}
     
-    Include:
-    - Compelling logline (1-2 sentences)
-    - Detailed story summary
-    - Character descriptions
-    - Key plot points
-    - Thematic elements
+    Write an engaging, cinematic sequel story with:
+    - Character development
+    - Plot twists
+    - Emotional moments
+    - Satisfying narrative arc
     
-    Write in a professional screenplay treatment style.
+    Make it feel like a real Hollywood movie!
+    Length: {story_length} words.
     """
     
     data = {
         'model': 'grok-beta',
         'messages': [{'role': 'user', 'content': prompt}],
-        'max_tokens': 1000,
-        'temperature': 0.9
+        'max_tokens': 600,
+        'temperature': 0.8
     }
     
     try:
         response = requests.post('https://api.x.ai/v1/chat/completions', headers=headers, json=data)
+        
         if response.status_code == 200:
             result = response.json()
-            return result['choices'][0]['message']['content']
+            story = result['choices'][0]['message']['content']
+            return f"🎬 **YOUR BLOCKBUSTER SEQUEL IS HERE!** 🍿\n\n{story}"
         else:
-            return f"🎭 Original {genre} movie created! Starring {main_actor} as {hero_type} against {villain_type}."
+            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!"
+            
     except Exception as e:
-        return f"✨ New {genre} story generated featuring {main_actor}!"
+        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."
 
-def randomize_story():
-    """Generate random story elements for inspiration"""
-    return {
-        "genre": random.choice(GENRES),
-        "main_actor": random.choice(ACTORS),
-        "hero_type": random.choice(HERO_ARCHETYPES),
-        "villain_type": random.choice(VILLAIN_TYPES),
-        "plot_trope": random.choice(PLOT_TROPES),
-        "rating": random.choice(["PG-13", "R"])
-    }
-
-def test_connection():
-    if not XAI_API_KEY:
-        return "❌ API key not found"
-    
-    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 cinematic storytelling! 🎭"
-        else:
-            return f"❌ Connection test failed: {response.status_code}"
-    except Exception as e:
-        return f"❌ Connection error: {str(e)}"
+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!"
 
-# Ultimate IdeaForge Interface
-with gr.Blocks(title="IdeaForge Studio - Complete Creative Suite") as demo:
-    gr.Markdown("# 🎬 IdeaForge Studio")
-    gr.Markdown("### **Voice your vision. Cast your stars. Create worlds.**")
-    gr.Markdown("*Complete AI Storytelling Platform* ✨")
-    
+# Enhanced Interface with Movie Icons
+with gr.Blocks(
+    title="IdeaForge Studio - AI Movie Maker",
+    theme=gr.themes.Soft()
+) as demo:
+    
+    # Header with Movie Icons
+    gr.Markdown(
+        """
+        # 🎬 IdeaForge Studio 
+        ### 🍿 *Voice your vision. Forge the future.*
+        **Lights, Camera, AI Action!** ✨
+        """
+    )
+    
+    # Main Tabs with Icons
     with gr.Tab("🎞️ Sequel Generator"):
-        gr.Markdown("Continue existing stories with new sequels")
+        gr.Markdown("### 🎥 Create Amazing Movie Sequels")
+        gr.Markdown("Turn your favorite stories into blockbuster sequels!")
         
         with gr.Row():
             with gr.Column():
                 original_input = gr.Textbox(
-                    label="Original Story/Plot", 
-                    placeholder="Describe the original movie or story...",
+                    label="🎭 Original Story/Plot", 
+                    placeholder="e.g., 'A hobbit's quest to destroy a powerful ring in the fires of Mount Doom'",
                     lines=3
                 )
                 sequel_input = gr.Textbox(
-                    label="Sequel Starting Point",
-                    placeholder="Where does the sequel begin?...", 
+                    label="🚀 Sequel Starting Point",
+                    placeholder="e.g., 'Years after the ring's destruction, a new shadow grows in the South'", 
                     lines=2
                 )
                 
             with gr.Column():
-                main_actor_input = gr.Dropdown(
-                    choices=ACTORS,
-                    value="Arnold Schwarzenegger",
-                    label="Main Actor",
-                    allow_custom_value=True
-                )
-                supporting_actors_input = gr.Textbox(
-                    label="Supporting Cast",
-                    placeholder="e.g., Javier Bardem as villain, Zendaya as love interest",
-                    lines=2
-                )
-        
-        with gr.Row():
-            rating_dropdown = gr.Dropdown(
-                choices=["G", "PG", "PG-13", "R", "NC-17"],
-                value="R",
-                label="MPAA Rating"
-            )
-            length_slider = gr.Slider(100, 2000, value=500, label="Story Length (words)")
-        
-        sequel_btn = gr.Button("🎭 Generate Sequel", variant="primary")
-        sequel_output = gr.Textbox(label="Your Sequel Story", lines=10)
-        
-        sequel_btn.click(
-            generate_sequel, 
-            [original_input, sequel_input, main_actor_input, supporting_actors_input, rating_dropdown, length_slider], 
-            sequel_output
-        )
-    
-    with gr.Tab("✨ Original Movie Creator"):
-        gr.Markdown("Build completely new movies from templates")
-        
-        with gr.Row():
-            with gr.Column():
-                genre_dropdown = gr.Dropdown(
-                    choices=GENRES,
-                    value="Action",
-                    label="Genre"
-                )
-                main_actor_dropdown = gr.Dropdown(
-                    choices=ACTORS,
-                    value="Chris Hemsworth",
-                    label="Main Actor",
-                    allow_custom_value=True
-                )
-                hero_dropdown = gr.Dropdown(
-                    choices=HERO_ARCHETYPES,
-                    value="Reluctant Hero",
-                    label="Hero Type"
+                length_slider = gr.Slider(
+                    100, 1500, value=500, 
+                    label="📏 Story Length (words)",
+                    info="Longer stories = more detail!"
                 )
                 
-            with gr.Column():
-                villain_dropdown = gr.Dropdown(
-                    choices=VILLAIN_TYPES,
-                    value="Ancient Evil",
-                    label="Villain Type"
-                )
-                plot_dropdown = gr.Dropdown(
-                    choices=PLOT_TROPES,
-                    value="Save the World",
-                    label="Plot Trope"
-                )
-                rating_original = gr.Dropdown(
-                    choices=["G", "PG", "PG-13", "R", "NC-17"],
-                    value="PG-13",
-                    label="MPAA Rating"
-                )
+                # 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")
         
-        supporting_original = gr.Textbox(
-            label="Supporting Cast (optional)",
-            placeholder="Add other actors and their roles...",
-            lines=2
-        )
+        generate_btn = gr.Button("🎬 Generate Blockbuster Sequel", variant="primary", size="lg")
         
-        custom_prompt = gr.Textbox(
-            label="Additional Story Notes (optional)",
-            placeholder="Any specific elements, twists, or settings you want included...",
-            lines=2
+        sequel_output = gr.Textbox(
+            label="🎉 Your Movie Sequel",
+            lines=8,
+            placeholder="Your AI-generated sequel will appear here...",
+            show_copy_button=True
         )
         
-        with gr.Row():
-            random_btn = gr.Button("🎲 Randomize Story Elements")
-            generate_original_btn = gr.Button("✨ Generate Original Movie", variant="primary")
+        # 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
+            ]
         
-        original_output = gr.Textbox(label="Your Original Movie Treatment", lines=12)
+        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
+            ]
         
-        # Randomize function
-        def update_random_elements():
-            random_elements = randomize_story()
+        def superhero_example():
             return [
-                random_elements["genre"],
-                random_elements["main_actor"], 
-                random_elements["hero_type"],
-                random_elements["villain_type"],
-                random_elements["plot_trope"],
-                random_elements["rating"]
+                "A superhero with incredible powers protects the city from villains",
+                "The hero loses their powers but discovers true strength within",
+                450
             ]
         
-        random_btn.click(
-            update_random_elements,
-            outputs=[genre_dropdown, main_actor_dropdown, hero_dropdown, villain_dropdown, plot_dropdown, rating_original]
-        )
+        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!")
         
-        generate_original_btn.click(
-            generate_original_movie,
-            [genre_dropdown, main_actor_dropdown, hero_dropdown, villain_dropdown, plot_dropdown, supporting_original, rating_original, custom_prompt],
-            original_output
+        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!",
+            show_copy_button=True
         )
+        
+        idea_btn.click(quick_movie_idea, outputs=idea_output)
     
-    with gr.Tab("🎤 Voice Input"):
-        gr.Markdown("**Voice Input Coming Soon!**")
+    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
+            )
+        
+        status_btn.click(test_connection, outputs=status_output)
+        
+        # System info
+        gr.Markdown("---")
         gr.Markdown("""
-        Future features:
-        - 🎤 Record your story ideas with voice
-        - 🎧 Audio-to-text conversion  
-        - 🗣️ Character voice samples
-        - 🎵 Musical theme suggestions
+        **🎯 How to Use:**
+        1. Test connection first (should show ✅)
+        2. Try a sequel with the examples
+        3. Create your own movie ideas!
+        
+        **🍿 Pro Tip:** Use specific, vivid descriptions for best results!
         """)
-        gr.Audio(label="Voice Recording (Coming Soon)", interactive=False)
     
-    with gr.Tab("🔗 Connection Test"):
-        gr.Markdown("Verify your xAI integration")
-        test_btn = gr.Button("Test xAI Connection")
-        status_output = gr.Textbox(label="Connection Status")
-        test_btn.click(test_connection, outputs=status_output)
+    # 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>
+        </div>
+        """
+    )
 
-demo.launch()
+# Launch the app
+if __name__ == "__main__":
+    demo.launch(share=True)