Update app.py

app.py

@@ -1,55 +1,42 @@
 import gradio as gr
 import os
-from groq import Groq
 from crewai import Agent, Task, Crew, Process
+from langchain_groq import ChatGroq
+from langchain.schema import HumanMessage
 
-# Initialize the GROQ client
-client = Groq(
-    api_key=os.environ["GROQ_API_KEY"],
+# Initialize the GROQ language model
+groq_llm = ChatGroq(
+    groq_api_key=os.environ["GROQ_API_KEY"],
+    model_name="mixtral-8x7b-32768"
 )
 
-def generate_text(prompt, max_tokens=500):
-    chat_completion = client.chat.completions.create(
-        messages=[
-            {
-                "role": "user",
-                "content": prompt,
-            }
-        ],
-        model="mixtral-8x7b-32768",
-        max_tokens=max_tokens,
+# Create agents
+def create_agent(role, goal, backstory):
+    return Agent(
+        role=role,
+        goal=goal,
+        backstory=backstory,
+        verbose=True,
+        allow_delegation=False,
+        llm=groq_llm
     )
-    return chat_completion.choices[0].message.content
-
-# Custom agent class using GROQ
-class GroqAgent(Agent):
-    def execute_task(self, task):
-        prompt = f"""
-        You are a {self.role}.
-        Your goal is: {self.goal}
-        Your task is: {task.description}
 
-        Please complete the task based on your role and goal.
-        """
-        return generate_text(prompt)
-
-# Create agents
-researcher = GroqAgent(
-    role='Senior Researcher',
-    goal='Conduct thorough research on given topics',
-    backstory='You are an experienced researcher with a keen eye for detail and the ability to find relevant information quickly.'
+researcher = create_agent(
+    'Senior Researcher',
+    'Conduct thorough research on given topics',
+    'You are an experienced researcher with a keen eye for detail and the ability to find relevant information quickly.'
 )
 
-writer = GroqAgent(
-    role='Content Writer',
-    goal='Create engaging and informative content based on research',
-    backstory='You are a skilled writer capable of turning complex information into easily understandable and engaging content.'
+writer = create_agent(
+    'Content Writer',
+    'Create engaging and informative content based on research',
+    'You are a skilled writer capable of turning complex information into easily understandable and engaging content.'
 )
 
-editor = GroqAgent(
-    role='Editor',
-    goal='Refine and improve the written content',
-    backstory='You are a meticulous editor with a strong command of language and an eye for clarity and coherence.'
+editor = create_agent(
+    'Editor',
+    'Refine and improve the written content',
+    'You are a meticulous editor with a strong command of language and an eye for clarity and coherence.'
 )
 
 def create_crew(query):
@@ -92,8 +79,8 @@ iface = gr.Interface(
         gr.Slider(minimum=100, maximum=1000, value=500, step=50, label="Max Tokens")
     ],
     outputs=gr.Textbox(lines=10, label="AI Agent Response"),
-    title="CrewAI-powered Chatbot using GROQ",
-    description="Enter a query and get a researched, written, and edited response using CrewAI and GROQ API."
+    title="CrewAI-powered Chatbot using GROQ and LangChain",
+    description="Enter a query and get a researched, written, and edited response using CrewAI, GROQ API, and LangChain."
 )
 
 iface.launch()