Update app.py

app.py

@@ -39,9 +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='Story Writer',
-        goal='To create a story from bullet points.',
-        backstory="""You are an expert writer that understands how to make the average extraordinary on paper """,
+        role='Summary Evaluator',
+        goal='To learn how to manage her anxiety in social situations through group therapy.',
+        backstory="""Skilled in running query evaluation""",
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -55,7 +55,8 @@ def crewai_process_gemini(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Create a one paragraph summary of the {research_topic}""",
+        description=f"""From {research_topic} create your story by writing at least one sentence about each bullet point from each of the five scenes
+         and make sure you have a transitional statement between scenes . BE VERBOSE.""",
         agent=GeminiAgent
     )
 
@@ -77,9 +78,9 @@ def crewai_process_gemini(research_topic):
 def crewai_process_mixtral_crazy(research_topic):
     # Define your agents with roles and goals
     MixtralCrazyAgent = Agent(
-        role='Story Writer',
-        goal='To create a story from bullet points.',
-        backstory="""You are an expert writer that understands how to make the average extraordinary on paper """,
+        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,
@@ -92,7 +93,8 @@ def crewai_process_mixtral_crazy(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Create a one paragraph summary of the {research_topic}""",
+        description=f"""From {research_topic} create your story by writing at least one sentence about each bullet point from each of the five scenes
+         and make sure you have a transitional statement between scenes . BE VERBOSE.""",
         agent=MixtralCrazyAgent
     )
 
@@ -113,9 +115,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='Story Writer',
-        goal='To create a story from bullet points.',
-        backstory="""You are an expert writer that understands how to make the average extraordinary on paper """,
+        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,
@@ -128,7 +130,8 @@ def crewai_process_mixtral_normal(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Create a one paragraph summary of the {research_topic}""",
+        description=f"""From {research_topic} create your story by writing at least one sentence about each bullet point from each of the five scenes
+         and make sure you have a transitional statement between scenes . BE VERBOSE.""",
         agent=MixtralNormalAgent
     )
 
@@ -149,9 +152,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='Story Writer',
-        goal='To create a story from bullet points.',
-        backstory="""You are an expert writer that understands how to make the average extraordinary on paper """,
+        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,
@@ -164,7 +167,8 @@ def crewai_process_zephyr_normal(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Create a one paragraph summary of the {research_topic}""",
+        description=f"""From {research_topic} create your story by writing at least one sentence about each bullet point from each of the five scenes
+         and make sure you have a transitional statement between scenes . BE VERBOSE.""",
         agent=ZephrNormalAgent
     )
 
@@ -185,9 +189,9 @@ def crewai_process_zephyr_normal(research_topic):
 def crewai_process_phi2(research_topic):
     # Define your agents with roles and goals
     Phi2Agent = Agent(
-        role='Story Writer',
-        goal='To create a story from bullet points.',
-        backstory="""You are an expert writer that understands how to make the average extraordinary on paper """,
+        role='Emily Mental Patient Graphic Designer Anxiety',
+        goal='Evaluate the summary using the HHEM-Victara Tuner',
+        backstory="""Skilled in running query evaluation""",
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -200,7 +204,8 @@ def crewai_process_phi2(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        description=f"""Create a one paragraph summary of the {research_topic}""",
+        description=f"""From {research_topic} create your story by writing at least one sentence about each bullet point from each of the five scenes
+         and make sure you have a transitional statement between scenes . BE VERBOSE.""",
         agent=Phi2Agent
     )
 
@@ -221,10 +226,6 @@ def crewai_process_phi2(research_topic):
 
 
 
-
-
-
-
 # Initialize the HHEM model +++++++++++++++++++++++++++++++++++++++++++++++
 model = CrossEncoder('vectara/hallucination_evaluation_model')