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='Summary Evaluator',
-        goal='To learn how to manage her anxiety in social situations through group therapy.',
-        backstory="""Skilled in running query evaluation""",
+        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 """,
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -55,8 +55,7 @@ def crewai_process_gemini(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        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.""",
+        description=f"""Create a one paragraph summary of the {research_topic}""",
         agent=GeminiAgent
     )
 
@@ -78,9 +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='Summary Evaluator',
-        goal='Evaluate the summary using the HHEM-Victara Tuner',
-        backstory="""Skilled in running query evaluation""",
+        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 """,
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -93,8 +92,7 @@ def crewai_process_mixtral_crazy(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        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.""",
+        description=f"""Create a one paragraph summary of the {research_topic}""",
         agent=MixtralCrazyAgent
     )
 
@@ -115,9 +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='Summary Evaluator',
-        goal='Evaluate the summary using the HHEM-Victara Tuner',
-        backstory="""Skilled in running query evaluation""",
+        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 """,
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -130,8 +128,7 @@ def crewai_process_mixtral_normal(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        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.""",
+        description=f"""Create a one paragraph summary of the {research_topic}""",
         agent=MixtralNormalAgent
     )
 
@@ -152,9 +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='Summary Evaluator',
-        goal='Evaluate the summary using the HHEM-Victara Tuner',
-        backstory="""Skilled in running query evaluation""",
+        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 """,
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -167,8 +164,7 @@ def crewai_process_zephyr_normal(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        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.""",
+        description=f"""Create a one paragraph summary of the {research_topic}""",
         agent=ZephrNormalAgent
     )
 
@@ -189,9 +185,9 @@ def crewai_process_zephyr_normal(research_topic):
 def crewai_process_phi2(research_topic):
     # Define your agents with roles and goals
     Phi2Agent = Agent(
-        role='Emily Mental Patient Graphic Designer Anxiety',
-        goal='Evaluate the summary using the HHEM-Victara Tuner',
-        backstory="""Skilled in running query evaluation""",
+        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 """,
         verbose=True,
         allow_delegation=False,
         llm = gemini_llm,
@@ -204,8 +200,7 @@ def crewai_process_phi2(research_topic):
 
     # Create tasks for your agents
     task1 = Task(
-        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.""",
+        description=f"""Create a one paragraph summary of the {research_topic}""",
         agent=Phi2Agent
     )
 
@@ -226,6 +221,10 @@ def crewai_process_phi2(research_topic):
 
 
 
+
+
+
+
 # Initialize the HHEM model +++++++++++++++++++++++++++++++++++++++++++++++
 model = CrossEncoder('vectara/hallucination_evaluation_model')
 
@@ -422,9 +421,8 @@ with tab4:
     st.header("Model Evaluation")
 
     # User input for the research topic
-    #research_topic = st.text_area('Enter your research topic:', '', height=100)
+    research_topic = st.text_area('Enter your research topic:', '', height=100)
 
-    research_topic = text1
 
     # Selection box for the function to execute
     process_selection = st.selectbox(