Update app.py

app.py

@@ -1,7 +1,13 @@
 import gradio as gr
 import torch
+from diffusers import StableDiffusionPipeline
 from groq import Groq
 
+# Initialize the Stable Diffusion model
+model_id = "stabilityai/stable-diffusion-2"  # Example model
+sd_pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+sd_pipe.to("cpu")
+
 # Initialize the Groq client with your API key
 client = Groq(api_key="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")
 
@@ -40,14 +46,18 @@ def generate_movie_content(theme):
     ])
     script = script_response["choices"][0]["message"]["content"]
 
-    return characters, storyline, script
+    # Generate image
+    image_prompt = f"A cinematic illustration of the theme: {theme}"
+    image = sd_pipe(image_prompt).images[0]
+
+    return characters, storyline, script, image
 
 def main_interface(theme):
-    characters, storyline, script = generate_movie_content(theme)
-    return f"Main Characters: {characters}\n\nStoryline: {storyline}\n\nScript: {script}"
+    characters, storyline, script, image = generate_movie_content(theme)
+    return f"Main Characters: {characters}\n\nStoryline: {storyline}\n\nScript: {script}", image
 
 # Gradio Interface
-title = "Movie Generator"
+title = "Movie Script and Image Generator"
 description = "Generate movie scripts, characters, and thematic images based on a single theme."
 
 demo = gr.Interface(
@@ -56,7 +66,8 @@ demo = gr.Interface(
         gr.Textbox(label="Theme", placeholder="Enter the movie theme"),
     ],
     outputs=[
-        gr.Textbox(label="Generated Content")
+        gr.Textbox(label="Generated Content"),
+        gr.Image(label="Generated Image"),
     ],
     title=title,
     description=description,