Update app.py

app.py

@@ -1,52 +1,146 @@
 import os
 import gradio as gr
 import requests
+import random
 
 # Your xAI API key for creative writing
 XAI_API_KEY = os.environ.get('XAI_API_KEY')
 
-def generate_sequel(original_story, sequel_start, story_length=500):
+# 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"
+]
+
+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, creative sequel story with:
-    - Character development
-    - Plot twists  
-    - Emotional depth
-    - Satisfying narrative arc
-    
-    Story length: approximately {story_length} words.
+    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  # More creative
+        'temperature': 0.8
     }
     
     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)
+        if response.status_code == 200:
+            result = response.json()
+            return result['choices'][0]['message']['content']
+        else:
+            return f"🎬 {rating}-rated sequel casting {main_actor}!"
+    except Exception as e:
+        return f"✨ {rating}-rated sequel featuring {main_actor}!"
+
+def generate_original_movie(genre, main_actor, hero_type, villain_type, plot_trope, supporting_actors, rating, custom_prompt=""):
+    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}
+    """
+    
+    if supporting_actors:
+        story_elements += f"SUPPORTING CAST: {supporting_actors}"
+    if custom_prompt:
+        story_elements += f"ADDITIONAL NOTES: {custom_prompt}"
+    
+    prompt = f"""
+    Create an original movie story based on these elements:
+    {story_elements}
+    
+    Include:
+    - Compelling logline (1-2 sentences)
+    - Detailed story summary
+    - Character descriptions
+    - Key plot points
+    - Thematic elements
+    
+    Write in a professional screenplay treatment style.
+    """
+    
+    data = {
+        'model': 'grok-beta',
+        'messages': [{'role': 'user', 'content': prompt}],
+        'max_tokens': 1000,
+        'temperature': 0.9
+    }
+    
+    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']
         else:
-            return f"🚀 Sequel generated! Based on '{original_story}', continuing from '{sequel_start}'. The story unfolds with new characters, unexpected twists, and deeper character development."
-            
+            return f"🎭 Original {genre} movie created! Starring {main_actor} as {hero_type} against {villain_type}."
     except Exception as e:
-        return f"✨ Creative sequel: The story continues with fresh adventures, building on the original narrative while introducing compelling new elements."
+        return f"✨ New {genre} story generated featuring {main_actor}!"
+
+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:
@@ -56,59 +150,159 @@ def test_connection():
     try:
         response = requests.get('https://api.x.ai/v1/models', headers=headers)
         if response.status_code == 200:
-            return "✅ xAI Connected! Ready for creative storytelling! 🎭"
+            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)}"
 
-# IdeaForge Creative Interface
-with gr.Blocks(title="IdeaForge Studio - AI Storytelling") as demo:
+# Ultimate IdeaForge Interface
+with gr.Blocks(title="IdeaForge Studio - Complete Creative Suite") as demo:
     gr.Markdown("# 🎬 IdeaForge Studio")
-    gr.Markdown("### **Voice your vision. Forge the future.**")
-    gr.Markdown("*Powered by xAI Grok with Active Credits!* ✨")
+    gr.Markdown("### **Voice your vision. Cast your stars. Create worlds.**")
+    gr.Markdown("*Complete AI Storytelling Platform* ✨")
     
-    with gr.Tab("📖 Generate Sequel"):
-        gr.Markdown("Create amazing movie sequels with AI")
+    with gr.Tab("🎞️ Sequel Generator"):
+        gr.Markdown("Continue existing stories with new sequels")
+        
+        with gr.Row():
+            with gr.Column():
+                original_input = gr.Textbox(
+                    label="Original Story/Plot", 
+                    placeholder="Describe the original movie or story...",
+                    lines=3
+                )
+                sequel_input = gr.Textbox(
+                    label="Sequel Starting Point",
+                    placeholder="Where does the sequel begin?...", 
+                    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)")
         
-        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'",
+        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"
+                )
+                
+            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"
+                )
+        
+        supporting_original = gr.Textbox(
+            label="Supporting Cast (optional)",
+            placeholder="Add other actors and their roles...",
             lines=2
         )
         
-        sequel_input = gr.Textbox(
-            label="Sequel Starting Point",
-            placeholder="e.g., 'Years after the ring's destruction, a new shadow grows in the South'", 
+        custom_prompt = gr.Textbox(
+            label="Additional Story Notes (optional)",
+            placeholder="Any specific elements, twists, or settings you want included...",
             lines=2
         )
         
-        length_slider = gr.Slider(100, 2000, value=500, label="Story Length (words)")
+        with gr.Row():
+            random_btn = gr.Button("🎲 Randomize Story Elements")
+            generate_original_btn = gr.Button("✨ Generate Original Movie", variant="primary")
         
-        generate_btn = gr.Button("🎭 Generate Sequel Story", variant="primary")
-        story_output = gr.Textbox(label="Your Sequel Story", lines=8)
+        original_output = gr.Textbox(label="Your Original Movie Treatment", lines=12)
         
-        generate_btn.click(
-            generate_sequel, 
-            [original_input, sequel_input, length_slider], 
-            story_output
+        # Randomize function
+        def update_random_elements():
+            random_elements = randomize_story()
+            return [
+                random_elements["genre"],
+                random_elements["main_actor"], 
+                random_elements["hero_type"],
+                random_elements["villain_type"],
+                random_elements["plot_trope"],
+                random_elements["rating"]
+            ]
+        
+        random_btn.click(
+            update_random_elements,
+            outputs=[genre_dropdown, main_actor_dropdown, hero_dropdown, villain_dropdown, plot_dropdown, rating_original]
+        )
+        
+        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
         )
     
+    with gr.Tab("🎤 Voice Input"):
+        gr.Markdown("**Voice Input Coming Soon!**")
+        gr.Markdown("""
+        Future features:
+        - 🎤 Record your story ideas with voice
+        - 🎧 Audio-to-text conversion  
+        - 🗣️ Character voice samples
+        - 🎵 Musical theme suggestions
+        """)
+        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)
-    
-    with gr.Tab("🎥 Examples"):
-        gr.Markdown("""
-        **Quick Start Examples:**
-        
-        | Original Story | Sequel Starting Point |
-        |---------------|----------------------|
-        | A hobbit's quest to destroy a powerful ring | Years later, a new shadow grows in the South |
-        | A young wizard discovers his magical heritage | The next generation faces an ancient evil |
-        | A superhero loses powers but gains wisdom | New threats require different kinds of strength |
-        """)
 
 demo.launch()