Update app.py

app.py

@@ -56,6 +56,19 @@ def crewai_process(research_topic):
 
     )
 
+    illustrator = Agent(
+        role='Children Story Illustrator',
+        goal="""Your primary goal is to bring children's stories to life through captivating and age-appropriate illustrations. . """,
+        backstory="""You have a passion for drawing and storytelling. As a child, you loved reading fairy tales and imagining vivid 
+        worlds filled with adventure and wonder. This love for stories and art grew over the years. You realize that the true magic 
+        happens when the words on a page were paired with enchanting illustrations.  """,
+        verbose=True,
+        allow_delegation=True,
+        llm = gemini_llm
+
+    )
+
+    
     artist = Agent(
         role='Storybook Illustrator',
         goal="""Visually bring stories to life. Create images that complement and enhance the text, 
@@ -135,28 +148,34 @@ def crewai_process(research_topic):
     )
 
     task2 = Task(
+        description="""Give each of the 8 parts a name and output the story.
+        """,
+        agent=illustrator
+    )  
+
+    task3 = Task(
         description="""Divide the story into 8 parts and ensure itis engaging and resonates with children, making it a valuable and enjoyable reading experience.
         """,
         agent=editor
     )  
 
     
-    task3 = Task(
+    task4 = Task(
         description="""Summarize the author story into an image prompt.""",
         agent=artist
     )
 
-    task4 = Task(
+    task5 = Task(
         description="""create a rhyming version of the story created by the author""",
         agent=poet
     )
 
-    task5 = Task(
+    task6 = Task(
         description="""create a rhyming version of the story created by the author""",
         agent=reader
     )
     
-    task6 = Task(
+    task7 = Task(
         description="""output both the story created by the author and the poem created by the poet, 
         make sure you label the story and the poem so you can tell them apart, name the story and poem appropriately """,
         agent=finalizer
@@ -167,7 +186,7 @@ def crewai_process(research_topic):
 
     # Instantiate your crew with a sequential process
     crew = Crew(
-        agents=[author, editor],
+        agents=[author, illustrator,
         tasks=[task1, task2],
         verbose=2,
         process=Process.sequential