Create app.py

app.py

@@ -0,0 +1,77 @@
+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("cuda")
+
+# Initialize the Groq client with your API key
+client = Groq(api_key="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")
+
+# Define the system message for the model
+system_message = {
+    "role": "system",
+    "content": "You are an experienced scriptwriter who generates engaging and creative movie content based on a given theme. Provide main characters, a storyline, and a movie script."
+}
+
+def generate_movie_content(theme):
+    # Generate main characters and storyline using Groq
+    setup_prompt = f"Generate main characters and a storyline for a movie with the theme: '{theme}'."
+    response = client.chat(messages=[
+        system_message,
+        {"role": "user", "content": setup_prompt}
+    ])
+    setup = response["choices"][0]["message"]["content"]
+
+    # Extract characters and storyline
+    if "Main Characters:" in setup and "Storyline:" in setup:
+        characters_start = setup.index("Main Characters:") + len("Main Characters:")
+        storyline_start = setup.index("Storyline:") + len("Storyline:")
+        characters = setup[characters_start:storyline_start].strip()
+        storyline = setup[storyline_start:].strip()
+    else:
+        characters = "Unknown characters"
+        storyline = "Unknown storyline"
+
+    # Generate movie script using Groq
+    script_prompt = (
+        f"Write a movie script based on the theme: '{theme}', with main characters: {characters}, and storyline: {storyline}."
+    )
+    script_response = client.chat(messages=[
+        system_message,
+        {"role": "user", "content": script_prompt}
+    ])
+    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
+
+def main_interface(theme):
+    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"
+description = "Generate movie scripts, characters, and thematic images based on a single theme."
+
+demo = gr.Interface(
+    fn=main_interface,
+    inputs=[
+        gr.Textbox(label="Theme", placeholder="Enter the movie theme"),
+    ],
+    outputs=[
+        gr.Textbox(label="Generated Content"),
+        gr.Image(label="Generated Image"),
+    ],
+    title=title,
+    description=description,
+)
+
+if __name__ == "__main__":
+    demo.launch()