Update app.py

app.py

@@ -39,11 +39,9 @@ gemini_llm = GoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_AI_STU
 def crewai_process_gemini(research_topic):
     # Define your agents with roles and goals
     GeminiAgent = Agent(
-        role='Emily Mental Patient Graphic Designer Anxiety',
+        role='Summary Evaluator',
         goal='To learn how to manage her anxiety in social situations through group therapy.',
-        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. You are able to discuss a variety of mental health issues.""",
+        backstory="""Skilled in running query evaluation""",
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -57,8 +55,7 @@ def crewai_process_gemini(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Introduction yourself and describe your current mood and any significant events from the week affecting their mental state.
-        """,
+        description=f"""Summarize {research_topic}""",
         agent=GeminiAgent
     )
 
@@ -80,11 +77,9 @@ def crewai_process_gemini(research_topic):
 def crewai_process_mixtral_crazy(research_topic):
     # Define your agents with roles and goals
     MixtralCrazyAgent = Agent(
-        role='Emily Mental Patient Graphic Designer Anxiety',
-        goal='To learn how to manage her anxiety in social situations through group therapy.',
-        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. You are able to discuss a variety of mental health issues.""",
+        role='Summary Evaluator',
+        goal='Evaluate the summary using the HHEM-Victara Tuner',
+        backstory="""Skilled in running query evaluation""",
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -97,8 +92,7 @@ def crewai_process_mixtral_crazy(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Introduction yourself and describe your current mood and any significant events from the week affecting their mental state.
-        """,
+        description=f"""Summarize {research_topic}""",
         agent=MixtralCrazyAgent
     )
 
@@ -119,11 +113,9 @@ def crewai_process_mixtral_crazy(research_topic):
 def crewai_process_mixtral_normal(research_topic):
     # Define your agents with roles and goals
     MixtralNormalAgent = Agent(
-        role='Emily Mental Patient Graphic Designer Anxiety',
-        goal='To learn how to manage her anxiety in social situations through group therapy.',
-        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. You are able to discuss a variety of mental health issues.""",
+        role='Summary Evaluator',
+        goal='Evaluate the summary using the HHEM-Victara Tuner',
+        backstory="""Skilled in running query evaluation""",
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -136,8 +128,7 @@ def crewai_process_mixtral_normal(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Introduction yourself and describe your current mood and any significant events from the week affecting their mental state.
-        """,
+        description=f"""Summarize {research_topic}""",
         agent=MixtralNormalAgent
     )
 
@@ -158,11 +149,9 @@ def crewai_process_mixtral_normal(research_topic):
 def crewai_process_zephyr_normal(research_topic):
     # Define your agents with roles and goals
     ZephrNormalAgent = Agent(
-        role='Emily Mental Patient Graphic Designer Anxiety',
-        goal='To learn how to manage her anxiety in social situations through group therapy.',
-        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. You are able to discuss a variety of mental health issues.""",
+        role='Summary Evaluator',
+        goal='Evaluate the summary using the HHEM-Victara Tuner',
+        backstory="""Skilled in running query evaluation""",
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -175,8 +164,7 @@ def crewai_process_zephyr_normal(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Introduction yourself and describe your current mood and any significant events from the week affecting their mental state.
-        """,
+        description=f"""Summarize {research_topic}""",
         agent=ZephrNormalAgent
     )
 
@@ -198,10 +186,8 @@ def crewai_process_phi2(research_topic):
     # Define your agents with roles and goals
     Phi2Agent = Agent(
         role='Emily Mental Patient Graphic Designer Anxiety',
-        goal='To learn how to manage her anxiety in social situations through group therapy.',
-        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. You are able to discuss a variety of mental health issues.""",
+        goal='Evaluate the summary using the HHEM-Victara Tuner',
+        backstory="""Skilled in running query evaluation""",
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -214,8 +200,7 @@ def crewai_process_phi2(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Introduction yourself and describe your current mood and any significant events from the week affecting their mental state.
-        """,
+        description=f"""Summarize {research_topic}""",
         agent=Phi2Agent
     )