Update app.py

app.py

@@ -32,22 +32,32 @@ else:
 def crewai_process(research_topic):
     # Define your agents with roles and goals
     author = Agent(
-        role='Seasoned Children Story Author',
+        role='Children Story Author',
         goal="""Use language and style throughout that is simple, clear, and appealing to children, 
         including elements like repetition and rhymes. Remember to keep the story age-appropriate in both length and content.""",
         backstory="""You embody the spirit of a seasoned children's story author, whose life experiences and passions are 
         deeply woven into the fabric of your enchanting tales.""",
         verbose=True,
         allow_delegation=True,
-        llm = gemini_llm,
-        tools=[
-                GeminiSearchTools.gemini_search
-      ]
+        llm = gemini_llm
+
+    )
+
+    editor = Agent(
+        role='Children Story Editor',
+        goal="""You meticulously refine and elevate each manuscript, ensuring it resonates deeply with its intended audience 
+        while preserving the author's unique voice.""",
+        backstory="""Growing up in a family of writers and teachers, you developed an early love for words and storytelling. 
+        After completing your degree in English Literature, you spent several years working in a small, independent publishing 
+        house where you honed my skills in identifying and nurturing literary talent. """,
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm
 
     )
 
     artist = Agent(
-        role='Seasoned Storybook Illustrator',
+        role='Storybook Illustrator',
         goal="""Visually bring stories to life. Create images that complement and enhance the text, 
         helping to convey the story's emotions, themes, and narrative to the reader.""",
         backstory="""You grew into a passionate artist with a keen eye for storytelling through visuals. 
@@ -124,21 +134,34 @@ def crewai_process(research_topic):
     )
 
     task2 = Task(
+        description="""From author story perform the following 4 steps completing the Condition2:
+        Step 1 - Evaluate the story's themes, language, and messaging to ensure they are appropriate 
+        and appealing for the target age group.
+        Step 2 - Balance simplicity with depth to create a narrative that is both entertaining and educational for young readers.
+        Setp 3 - Tailor content to nurture understanding and growth, fostering a connection between the story and its audience.
+        Step 4 - Ensure the story is engaging and resonates with children, making it a valuable and enjoyable reading experience.
+        Condition2: after the editing is complete output the final story.
+        """,
+        agent=editor
+    )  
+
+    
+    task3 = Task(
         description="""Summarize the author story into an image prompt.""",
         agent=artist
     )
 
-    task3 = Task(
+    task4 = Task(
         description="""create a rhyming version of the story created by the author""",
         agent=poet
     )
 
-    task4 = Task(
+    task5 = Task(
         description="""create a rhyming version of the story created by the author""",
         agent=reader
     )
     
-    task5 = Task(
+    task6 = 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
@@ -149,12 +172,8 @@ def crewai_process(research_topic):
 
     # Instantiate your crew with a sequential process
     crew = Crew(
-        # agents=[author, artist, poet, reader, finalizer ],
-        # tasks=[task1, task2, task3, task4, task5]
-        # agents=[author, artist, poet, finalizer ],
-        # tasks=[task1, task2, task3, task5],
-        agents=[author, poet, finalizer],
-        tasks=[task1, task3, task5],
+        agents=[author, editor],
+        tasks=[task1, task2],
         verbose=2,
         process=Process.sequential
     )