Update app.py

app.py

@@ -1,45 +1,63 @@
-import os
 import gradio as gr
+import os
+from groq import Groq
 from crewai import Agent, Task, Crew, Process
-from langchain_groq import ChatGroq
-from langchain.tools import DuckDuckGoSearchRun
+from langchain_community.tools import DuckDuckGoSearchRun
 
-# Initialize the GROQ language model
-groq_llm = ChatGroq(
-    groq_api_key=os.environ["GROQ_API_KEY"],
-    model_name="mixtral-8x7b-32768"
+# Initialize the GROQ client
+client = Groq(
+    api_key=os.environ["GROQ_API_KEY"],
 )
 
 # Initialize the search tool
 search_tool = DuckDuckGoSearchRun()
 
-# Create agents
-researcher = Agent(
-    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.',
-    verbose=True,
-    allow_delegation=False,
-    llm=groq_llm,
-    tools=[search_tool]
+# Create a function to generate text using GROQ
+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,
+    )
+    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
+            }
+        }
+    )
+
+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 = Agent(
-    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.',
-    verbose=True,
-    allow_delegation=False,
-    llm=groq_llm
+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 = Agent(
-    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.',
-    verbose=True,
-    allow_delegation=False,
-    llm=groq_llm
+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):
@@ -69,18 +87,21 @@ def create_crew(query):
 
     return crew
 
-def process_query(query):
+def process_query(query, max_tokens=500):
     crew = create_crew(query)
     result = crew.kickoff()
     return result
 
-# Create Gradio interface
+# Create the Gradio interface
 iface = gr.Interface(
     fn=process_query,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your query here..."),
+    inputs=[
+        gr.Textbox(lines=5, label="Enter your query"),
+        gr.Slider(minimum=100, maximum=1000, value=500, step=50, label="Max Tokens")
+    ],
     outputs=gr.Textbox(lines=10, label="AI Agent Response"),
-    title="AI Agent Chatbot",
-    description="Ask a question or provide a topic, and our AI agents will research, write, and edit a response for you."
+    title="CrewAI-powered Chatbot",
+    description="Enter a query and get a researched, written, and edited response using CrewAI and GROQ API."
 )
 
 iface.launch()