Create app.py

app.py

@@ -0,0 +1,109 @@
+import os
+import gradio as gr
+from crewai import Agent, Task, Crew
+from langchain_groq import ChatGroq
+from langchain_community.tools import DuckDuckGoSearchRun, DuckDuckGoSearchResults
+from crewai_tools import tool
+
+# Define the DuckDuckGoSearch tool using the decorator for tool registration
+@tool('DuckDuckGoSearch')
+def search(search_query: str):
+    """Search the web for information on a given topic."""
+    return DuckDuckGoSearchRun().run(search_query)
+
+# Define the DuckDuckGoSearchResults tool
+@tool('DuckDuckGoSearchResults')
+def search_results(search_query: str):
+    """Search the web for information and sources on a given topic with links and titles."""
+    return DuckDuckGoSearchResults().run(search_query)
+
+def kickoff_crew(topic: str) -> dict:
+    """Kickoff the research process for a given topic using CrewAI components."""
+    # Retrieve the API key from the environment variables
+    groq_api_key = os.environ.get("GROQ_API_KEY")
+    if not groq_api_key:
+        raise ValueError("API Key for Groq is not set in environment variables")
+
+    # Initialize the Groq large language model
+    groq_llm = ChatGroq(temperature=0, groq_api_key=groq_api_key, model_name="llama3-70b-8192")
+
+    # Define Agents
+    researcher = Agent(
+        role='Researcher',
+        goal=f'Collect detailed information on {topic}',
+        tools=[search, search_results],
+        llm=groq_llm,
+        backstory=(
+            "As a diligent researcher, you explore the depths of the internet to "
+            "unearth crucial information and insights on the assigned topics. "
+            "Attention to detail and accuracy is critical and requires that you keep a record of all the sources you use."
+        ),
+        allow_delegation=False,
+        max_iter=5,
+        verbose=True
+    )
+
+    editor = Agent(
+        role='Editor',
+        goal='Compile and refine the information into a comprehensive report on {topic}',
+        llm=groq_llm,
+        backstory=(
+            "With a keen eye for detail and a strong command of language, you transform "
+            "raw data into polished, insightful reports that are both informative and engaging."
+        ),
+        allow_delegation=False,
+        max_iter=3,
+        verbose=True
+    )
+
+    # Define Tasks
+    research_task = Task(
+        description=(
+            "Use DuckDuckGoSearch and DuckDuckGoSearchResults to gather information about {topic}. "
+            "Compile your findings into an initial draft. "
+            "Gather all the sources with title and link relevant to the topic. "
+            "Be sure you don't make up or invent any information."
+        ),
+        expected_output='A draft report containing all relevant information about the topic and sources used',
+        agent=researcher
+    )
+
+    edit_task = Task(
+        description=(
+            "Review and refine the draft report. Organize the content, check for accuracy, "
+            "and enhance readability. "
+            "Include a section with all the sources used in the report from research_task as bullets: (title)[link]"
+        ),
+        expected_output='A finalized comprehensive report on ## {topic} ##',
+        agent=editor,
+        context=[research_task]
+    )
+
+    # Forming the Crew
+    crew = Crew(
+        agents=[researcher, editor],
+        tasks=[research_task, edit_task]
+    )
+
+    # Kick-off the research process
+    result = crew.kickoff(inputs={'topic': topic})
+    return result
+
+def main():
+    """Set up the Gradio interface for the CrewAI Research Tool."""
+    with gr.Blocks() as demo:
+        gr.Markdown("## CrewAI Research Tool")
+        topic_input = gr.Textbox(label="Enter Topic", placeholder="Type here...")
+        submit_button = gr.Button("Start Research")
+        output = gr.Textbox(label="Result")
+
+        submit_button.click(
+            fn=kickoff_crew,
+            inputs=topic_input,
+            outputs=output
+        )
+
+    demo.launch()
+
+if __name__ == "__main__":
+    main()