Update app.py

app.py

@@ -7,6 +7,232 @@ from sentence_transformers import CrossEncoder
 import numpy as np
 import re
 
+from textwrap import dedent
+import google.generativeai as genai
+
+
+# Tool import
+from crewai.tools.gemini_tools import GeminiSearchTools
+from crewai.tools.mixtral_tools import MixtralSearchTools
+from crewai.tools.zephyr_tools import ZephyrSearchTools
+from crewai.tools.phi2_tools import Phi2SearchTools
+
+
+# Google Langchain
+from langchain_google_genai import GoogleGenerativeAI
+
+#Crew imports
+from crewai import Agent, Task, Crew, Process
+
+# Retrieve API Key from Environment Variable
+GOOGLE_AI_STUDIO = os.environ.get('GOOGLE_API_KEY')
+
+# Ensure the API key is available
+if not GOOGLE_AI_STUDIO:
+    raise ValueError("API key not found. Please set the GOOGLE_AI_STUDIO2 environment variable.")
+
+# Set gemini_llm
+gemini_llm = GoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_AI_STUDIO)
+
+# CrewAI +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+def crewai_process_gemini(research_topic):
+    # Define your agents with roles and goals
+    GeminiAgent = 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.""",
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                GeminiSearchTools.gemini_search
+                   
+      ]
+
+    )
+
+
+    # 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.
+        """,
+        agent=GeminiAgent
+    )
+
+    # Instantiate your crew with a sequential process
+    crew = Crew(
+        agents=[GeminiAgent],
+        tasks=[task1],
+        verbose=2,
+        process=Process.sequential
+    )
+
+    # Get your crew to work!
+    result = crew.kickoff()
+    
+    return result
+
+
+
+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.""",
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                MixtralSearchTools.mixtral_crazy      
+      ]
+
+    )
+
+
+    # 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.
+        """,
+        agent=MixtralCrazyAgent
+    )
+
+    # Instantiate your crew with a sequential process
+    crew = Crew(
+        agents=[MixtralCrazyAgent],
+        tasks=[task1],
+        verbose=2,
+        process=Process.sequential
+    )
+
+    # Get your crew to work!
+    result = crew.kickoff()
+    
+    return result
+
+
+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.""",
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                MixtralSearchTools.mixtral_normal      
+      ]
+
+    )
+
+
+    # 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.
+        """,
+        agent=MixtralNormalAgent
+    )
+
+    # Instantiate your crew with a sequential process
+    crew = Crew(
+        agents=[MixtralNormalAgent],
+        tasks=[task1],
+        verbose=2,
+        process=Process.sequential
+    )
+
+    # Get your crew to work!
+    result = crew.kickoff()
+    
+    return result
+
+
+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.""",
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                ZephyrSearchTools.zephyr_normal     
+      ]
+
+    )
+
+
+    # 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.
+        """,
+        agent=ZephrNormalAgent
+    )
+
+    # Instantiate your crew with a sequential process
+    crew = Crew(
+        agents=[ZephrNormalAgent],
+        tasks=[task1],
+        verbose=2,
+        process=Process.sequential
+    )
+
+    # Get your crew to work!
+    result = crew.kickoff()
+    
+    return result
+
+
+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.""",
+        verbose=True,
+        allow_delegation=False,
+        llm = gemini_llm,
+        tools=[
+                Phi2SearchTools.phi2_search     
+      ]
+
+    )
+
+
+    # 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.
+        """,
+        agent=Phi2Agent
+    )
+
+    # Instantiate your crew with a sequential process
+    crew = Crew(
+        agents=[Phi2Agent],
+        tasks=[task1],
+        verbose=2,
+        process=Process.sequential
+    )
+
+    # Get your crew to work!
+    result = crew.kickoff()
+    
+    return result
+
+
 
 # Credentials ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++