Update app.py

app.py

@@ -36,10 +36,46 @@ TITLE3 = """<h3 align="center">"To see active group discussion click on logs dur
 
 def crewai_process(research_topic):
     # Define your agents with roles and goals
-    crazy1 = Agent(
-        role='Homeless Person',
-        goal=f'Discuss issue with {research_topic} and come to a deeper understanding',
-        backstory="""Not doing well mentally living on the street""",
+    Emily = Agent(
+        role='Mental Patient',
+        goal='To learn how to manage her anxiety in social situations.',
+        backstory="""Emily is a 28-year-old graphic designer. She has always struggled with social anxiety, 
+        making it difficult for her to participate in group settings. She joined the therapy group to improve 
+        her social skills and manage her anxiety.""",
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                MixtralSearchTools.mixtral_normal,
+                GeminiSearchTools.gemini_search
+                
+      ]
+
+    )
+
+    David = Agent(
+        role='Mental Patient',
+        goal='To gain insights into managing his bipolar disorder.',
+        backstory="""David, a 35-year-old musician, has been living with bipolar disorder for over a decade. 
+        His condition has impacted his career and personal life. He seeks to understand his emotions better 
+        and find stability through the group sessions.""",
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                ZephyrSearchTools.zephyr_crazy,
+                GeminiSearchTools.gemini_search
+                
+      ]
+
+    )    
+
+    Sarah = Agent(
+        role='Mental Patient',
+        goal='To find strategies to cope with her depression.',
+        backstory="""Sarah, 42, is a former teacher who has been battling depression for several years. 
+        The illness has led her to leave her job. She hopes to find new coping mechanisms and rediscover 
+        her passion for teaching.""",
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -49,39 +85,78 @@ def crewai_process(research_topic):
                 
       ]
 
+    )  
+
+    Michael = Agent(
+        role='Mental Patient',
+        goal='To overcome his PTSD and return to a normal life.',
+        backstory="""Michael is a 30-year-old ex-soldier. He developed PTSD following his service. 
+        Struggling with flashbacks and anxiety, he joined the group to seek support and ways to 
+        return to civilian life smoothly.""",
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                ZephyrSearchTools.zephyr_normal,
+                GeminiSearchTools.gemini_search
+                
+      ]
+
     )
-    crazy2 = Agent(
-        role='Distressed Mom',
-        goal='Resond to answers',
-        backstory="""Struggling with depression and loss of a child""",
+
+    Lisa  = Agent(
+        role='Therapist',
+        goal='To facilitate the group progress and assist each member in their personal goals.',
+        backstory=""" Dr. Thompson is a seasoned psychologist specializing in group therapy. 
+        With over 15 years of experience, she is skilled at creating a safe space for her patients 
+        to explore and address their mental health challenges.""",
         verbose=True,
-        allow_delegation=True,
+        allow_delegation=False,
         llm = gemini_llm,
         tools=[
                 MixtralSearchTools.mixtral_normal,
+                ZephyrSearchTools.zephyr_normal, 
                 GeminiSearchTools.gemini_search
+                
       ]
-     
-        # Add tools and other optional parameters as needed
+
     )
+    
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Discuss the  {research_topic}. Use Mixtral
-        do 10 rounds of chat with. react to crazy2 """,
-        agent=crazy1
+        description=f"""Discuss the  {research_topic} from your backstory perspective. Use Mixtral
+        """,
+        agent=Emily
     )
 
     task2 = Task(
-        description=f"""Discuss the  {research_topic}. Use Mixtral
-        do 10 rounds of chat with. react to crazy1 """,
-        agent=crazy2
+        description=f"""Discuss the  {research_topic} from your backstory perspective. Use Zephyr
+        """,
+        agent=David
+    )
+
+    task3 = Task(
+        description=f"""Discuss the  {research_topic} from your backstory perspective. Use Mixtral
+        """,
+        agent=Sarah
+    )
+    task4 = Task(
+        description=f"""Discuss the  {research_topic} from your backstory perspective. Use Zephyr
+        """,
+        agent=Michael
+    )
+    task4 = Task(
+        description=f"""Discuss the  {research_topic} from your backstory perspective. 
+        Use Gemini, Mixtral or Zephyr what ever is appropriate.
+        """,
+        agent=Lisa
     )
 
     # Instantiate your crew with a sequential process
     crew = Crew(
-        agents=[crazy1, crazy2],
-        tasks=[task1, task2],
+        agents=[Emily, David, Sarah, Michael, Lisa ],
+        tasks=[task1, task2, task3, task4, task5],
         verbose=2,
         process=Process.sequential
     )