Update app.py

app.py

@@ -2,17 +2,12 @@ import gradio as gr
 import os
 from groq import Groq
 from crewai import Agent, Task, Crew, Process
-from langchain_community.tools import DuckDuckGoSearchRun
 
 # Initialize the GROQ client
 client = Groq(
     api_key=os.environ["GROQ_API_KEY"],
 )
 
-# Initialize the search tool
-search_tool = DuckDuckGoSearchRun()
-
-# Create a function to generate text using GROQ
 def generate_text(prompt, max_tokens=500):
     chat_completion = client.chat.completions.create(
         messages=[
@@ -26,38 +21,35 @@ def generate_text(prompt, max_tokens=500):
     )
     return chat_completion.choices[0].message.content
 
-# Create agents using the GROQ text generation function
-def create_agent(role, goal, backstory):
-    return Agent(
-        role=role,
-        goal=goal,
-        backstory=backstory,
-        verbose=True,
-        allow_delegation=False,
-        llm_config={
-            "function": generate_text,
-            "config": {
-                "max_tokens": 500
-            }
-        }
-    )
+# 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)
 
-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.'
+# 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.'
 )
 
-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.'
+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.'
 )
 
-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.'
+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.'
 )
 
 def create_crew(query):
@@ -100,7 +92,7 @@ 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",
+    title="CrewAI-powered Chatbot using GROQ",
     description="Enter a query and get a researched, written, and edited response using CrewAI and GROQ API."
 )