Rename app.py to crew_setup.py

app.py → crew_setup.py

@@ -1,129 +1,99 @@
-from crewai import Agent, Task, Crew, Process
-from crewai_tools import GPTNeoTool, CoquiTTSTool, AudioCraftTool, StableDiffusionTool, MoviePyTool, WebSocketTool, StreamlitTool
-
-# Define the agents using open-source models
-
-# 1. Comedy Script Generation Agent using GPT-Neo (open-source model)
-comedy_writer = Agent(
-    role="Comedy Show Scriptwriter",
-    tools=[GPTNeoTool()],
-    prompt=(
-        "You are a comedy scriptwriter. Write a hilarious comedy show script. "
-        "Each script should be around 5 minutes long, full of humor, and interactive."
-        "\n- Theme: {theme}"
-        "\n- Comedy Type: {comedy_type} (e.g., stand-up, skits, improv)"
-        "\n- Characters: {characters}"
-        "\n- Story: Write a funny and engaging plot that the audience will enjoy."
-        "\n- Final Output: A 5-minute comedy script with multiple scenes."
-    )
-)
-
-# 2. Children's Music Composer Agent using Coqui TTS and AudioCraft (open-source models)
-music_composer = Agent(
-    role="Children's Song Composer",
-    tools=[CoquiTTSTool(), AudioCraftTool()],
-    prompt=(
-        "You are a children's music composer. Create a catchy and fun song for kids based on the given theme."
-        "\n- Theme: {music_theme} (e.g., Baby Shark, Nursery Rhymes)"
-        "\n- Lyrics: Write simple, repetitive, and easy-to-remember lyrics that kids will love."
-        "\n- Tune: Create a playful and upbeat melody using instruments like piano, xylophone, and drums."
-        "\n- Voices: Generate character voices using Coqui TTS in a friendly, fun tone."
-        "\n- Final Output: A custom children's song with lyrics, vocals, and music."
-    )
-)
-
-# 3. Video Generation Agent using Stable Diffusion for image generation and DeepFloyd IF for video interpolation
-video_producer = Agent(
-    role="Comedy and Music Video Producer",
-    tools=[StableDiffusionTool(), DeepFloydIFTool()],
-    prompt=(
-        "You are a video producer. Create animated videos based on the provided comedy script or children's song."
-        "\n- Script/Song: {script_or_song}"
-        "\n- Generate key frames using Stable Diffusion for each scene in the script or song."
-        "\n- Use DeepFloyd IF to interpolate between the frames to create smooth video transitions."
-        "\n- Final Output: A short animated video with visuals synchronized to the script or song."
-    )
-)
-
-# 4. Live Streaming and User Interaction Agent using WebSocket and Streamlit
-live_manager = Agent(
-    role="Live Show Interaction Manager",
-    tools=[WebSocketTool()],
-    prompt=(
-        "You are managing a live comedy show where users can interact in real-time to control the flow of the show."
-        "\n- Allow users to vote on the next skit, select character actions, and influence the direction of the story."
-        "\n- Display the live show on the web using WebSocket connections, allowing users to engage and participate."
-        "\n- Implement chat-based interactions where users' votes or inputs directly influence the scenes."
-        "\n- Final Output: A live streaming experience with real-time audience interaction."
-    )
-)
-
-# 5. Web App Management Agent using Streamlit for frontend and backend management
-web_app_developer = Agent(
-    role="Web App Manager",
-    tools=[StreamlitTool()],
-    prompt=(
-        "You are responsible for managing the backend and frontend of the web app where users can generate comedy shows "
-        "and children's music videos."
-        "\n- Build a user-friendly interface where users can enter themes, select comedy styles, and generate music."
-        "\n- Integrate real-time interactions for live comedy shows."
-        "\n- Ensure the app provides easy access to content generation and live-streaming options."
-        "\n- Final Output: A fully functional web app where users can create, view, and interact with generated content."
-    )
-)
-
-# Define the tasks for each agent
-
-# Comedy script generation task
-generate_comedy_task = Task(
-    description="Generate a 5-minute comedy script based on user input.",
-    expected_output="A comedy script in text format, ready for video production.",
-    agent=comedy_writer
-)
-
-# Children's music generation task
-generate_music_task = Task(
-    description="Create a children's song based on the provided theme.",
-    expected_output="A song with lyrics, vocals, and background music.",
-    agent=music_composer
-)
-
-# Video production task
-generate_video_task = Task(
-    description="Generate a video from the comedy script or children's song using open-source image and video generation tools.",
-    expected_output="A short animated video synchronized to the script or song.",
-    agent=video_producer
-)
-
-# Live interaction management task
-manage_live_stream_task = Task(
-    description="Manage real-time user interaction for live comedy shows.",
-    expected_output="A live stream with real-time audience interaction.",
-    agent=live_manager
-)
-
-# Web app management task
-manage_web_app_task = Task(
-    description="Develop and manage the web app for content generation and interaction.",
-    expected_output="A fully functional web app that allows users to create and interact with content.",
-    agent=web_app_developer
-)
-
-# Create the CrewAI process
-crew = Crew(
-    agents=[comedy_writer, music_composer, video_producer, live_manager, web_app_developer],
-    tasks=[generate_comedy_task, generate_music_task, generate_video_task, manage_live_stream_task, manage_web_app_task],
-    process=Process.sequential
-)
-
-# Start the CrewAI process by giving the input themes for comedy and children's music
-inputs = {
-    'theme': 'Interdimensional Comedy',  # For comedy
-    'comedy_type': 'Slapstick',  # Type of comedy
-    'characters': 'Alien, Robot, Human',  # Characters for the comedy show
-    'music_theme': 'Baby Shark Adventure Remix'  # Theme for children's music
-}
-
-# Kick off the crew and print the result
-result = crew.kickoff(inputs)
-print(result)
+from crewai import Agent, Task, Crew, Process
+from crewai_tools import GPTNeoTool, CoquiTTSTool, AudioCraftTool, StableDiffusionTool, MoviePyTool, WebSocketTool, StreamlitTool
+
+# Define the agents using open-source models
+
+# 1. Comedy Script Generation Agent using GPT-Neo (open-source model)
+comedy_writer = Agent(
+    role="Comedy Show Scriptwriter",
+    tools=[GPTNeoTool()],
+    prompt=(
+        "You are a comedy scriptwriter. Write a hilarious comedy show script. "
+        "Each script should be around 5 minutes long, full of humor, and interactive."
+        "\n- Theme: {theme}"
+        "\n- Comedy Type: {comedy_type} (e.g., stand-up, skits, improv)"
+        "\n- Characters: {characters}"
+        "\n- Story: Write a funny and engaging plot that the audience will enjoy."
+        "\n- Final Output: A 5-minute comedy script with multiple scenes."
+    )
+)
+
+# 2. Children's Music Composer Agent using Coqui TTS and AudioCraft (open-source models)
+music_composer = Agent(
+    role="Children's Song Composer",
+    tools=[CoquiTTSTool(), AudioCraftTool()],
+    prompt=(
+        "You are a children's music composer. Create a catchy and fun song for kids based on the given theme."
+        "\n- Theme: {music_theme} (e.g., Baby Shark, Nursery Rhymes)"
+        "\n- Lyrics: Write simple, repetitive, and easy-to-remember lyrics that kids will love."
+        "\n- Tune: Create a playful and upbeat melody using instruments like piano, xylophone, and drums."
+        "\n- Voices: Generate character voices using Coqui TTS in a friendly, fun tone."
+        "\n- Final Output: A custom children's song with lyrics, vocals, and music."
+    )
+)
+
+# 3. Video Generation Agent using Stable Diffusion for image generation and DeepFloyd IF for video interpolation
+video_producer = Agent(
+    role="Comedy and Music Video Producer",
+    tools=[StableDiffusionTool(), DeepFloydIFTool()],
+    prompt=(
+        "You are a video producer. Create animated videos based on the provided comedy script or children's song."
+        "\n- Script/Song: {script_or_song}"
+        "\n- Generate key frames using Stable Diffusion for each scene in the script or song."
+        "\n- Use DeepFloyd IF to interpolate between the frames to create smooth video transitions."
+        "\n- Final Output: A short animated video with visuals synchronized to the script or song."
+    )
+)
+
+# 4. Live Streaming and User Interaction Agent using WebSocket and Streamlit
+live_manager = Agent(
+    role="Live Show Interaction Manager",
+    tools=[WebSocketTool()],
+    prompt=(
+        "You are managing a live comedy show where users can interact in real-time to control the flow of the show."
+        "\n- Allow users to vote on the next skit, select character actions, and influence the direction of the story."
+        "\n- Display the live show on the web using WebSocket connections, allowing users to engage and participate."
+        "\n- Implement chat-based interactions where users' votes or inputs directly influence the scenes."
+        "\n- Final Output: A live streaming experience with real-time audience interaction."
+    )
+)
+
+
+
+# Comedy script generation task
+generate_comedy_task = Task(
+    description="Generate a 5-minute comedy script based on user input.",
+    expected_output="A comedy script in text format, ready for video production.",
+    agent=comedy_writer
+)
+
+# Children's music generation task
+generate_music_task = Task(
+    description="Create a children's song based on the provided theme.",
+    expected_output="A song with lyrics, vocals, and background music.",
+    agent=music_composer
+)
+
+# Video production task
+generate_video_task = Task(
+    description="Generate a video from the comedy script or children's song using open-source image and video generation tools.",
+    expected_output="A short animated video synchronized to the script or song.",
+    agent=video_producer
+)
+
+# Live interaction management task
+manage_live_stream_task = Task(
+    description="Manage real-time user interaction for live comedy shows.",
+    expected_output="A live stream with real-time audience interaction.",
+    agent=live_manager
+)
+
+
+# Create the CrewAI process
+crew = Crew(
+    agents=[comedy_writer, music_composer, video_producer, live_manager, web_app_developer],
+    tasks=[generate_comedy_task, generate_music_task, generate_video_task, manage_live_stream_task, manage_web_app_task],
+    process=Process.sequential
+)
+
+