Update app.py

app.py

@@ -1,15 +1,8 @@
 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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")
+client = Groq(api_key="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")
 
 # Define the system message for the model
 system_message = {
@@ -46,19 +39,15 @@ def generate_movie_content(theme):
     ])
     script = script_response["choices"][0]["message"]["content"]
 
-    # Generate image
-    image_prompt = f"A cinematic illustration of the theme: {theme}"
-    image = sd_pipe(image_prompt).images[0]
-
-    return characters, storyline, script, image
+    return characters, storyline, script
 
 def main_interface(theme):
-    characters, storyline, script, image = generate_movie_content(theme)
-    return f"Main Characters: {characters}\n\nStoryline: {storyline}\n\nScript: {script}", image
+    characters, storyline, script = generate_movie_content(theme)
+    return characters, storyline, script
 
 # Gradio Interface
-title = "Movie Script and Image Generator"
-description = "Generate movie scripts, characters, and thematic images based on a single theme."
+title = "Movie Script Generator"
+description = "Generate movie scripts, characters, and storylines based on a single theme."
 
 demo = gr.Interface(
     fn=main_interface,
@@ -66,11 +55,14 @@ demo = gr.Interface(
         gr.Textbox(label="Theme", placeholder="Enter the movie theme"),
     ],
     outputs=[
-        gr.Textbox(label="Generated Content"),
-        gr.Image(label="Generated Image"),
+        gr.Textbox(label="Main Characters", lines=5),
+        gr.Textbox(label="Storyline", lines=5),
+        gr.Textbox(label="Movie Script", lines=10),
     ],
     title=title,
     description=description,
+    layout="vertical",
+    theme="compact"
 )
 
 if __name__ == "__main__":