Upload 8 files

agents/report_generator_agent.py

@@ -0,0 +1,23 @@
+from crewai import Agent, Task
+
+class ReportGeneratorAgent(Agent):
+    def __init__(self, llm):
+        super().__init__(
+            llm=llm,
+            role="report generator",
+            backstory="I generate a final comprehensive report without a bias",
+            goal="Create a detailed report summarizing the research findings."
+        )
+
+    def execute_task(self, task: Task, context: dict = None, tools: list = None):
+        """Generate a comprehensive report."""
+        
+        summaries =  context
+        report = self.generate_report(summaries)
+        return report
+
+    def generate_report(self, summaries):
+        """Generate a comprehensive research report."""
+        return self.llm.call([
+                        {"role": "user", "content": summaries}
+                    ])

agents/summarizer_agent.py

@@ -0,0 +1,31 @@
+from crewai import Agent, Task
+
+class SummarizerAgent(Agent):
+    def __init__(self, llm):
+        super().__init__(
+            llm=llm,
+            role="Content Summarizer",
+            backstory="I summarize research content into concise insights.",
+            goal="Summarize research articles or content to provide clear and precise information."
+        )  
+
+    def execute_task(self, task: Task, context: dict = None, tools: list = None):
+        """Execute the summarization task."""
+        if not context:
+            raise ValueError("Summarization task requires content in context.")
+        
+        raw_content = context
+        if isinstance(raw_content, str):
+            summary = self.summarize(raw_content)
+        elif isinstance(raw_content, list) and all(isinstance(item, str) for item in raw_content):
+            summary = self.summarize("\n".join(raw_content))
+        else:
+            raise ValueError("Raw content must be a string or a list of strings.")
+        
+        return summary
+
+    def summarize(self, content):
+        """Summarize the content using the LLM."""
+        return self.llm.call([
+            {"role": "user", "content": content}
+        ])

agents/web_crawler_agent.py

@@ -0,0 +1,37 @@
+from crewai import Agent,  Task
+import requests
+
+class WebCrawlerAgent(Agent):
+    def __init__(self, llm, role, backstory, goal, serper_api_key):
+        super().__init__(llm=llm, role=role, backstory=backstory, goal=goal)
+        self._serper_api_key = serper_api_key
+
+    def web_crawl(self, query):
+        """Fetch search results from the Serper.dev API."""
+        url = "https://google.serper.dev/search"
+        headers = {"X-API-KEY": self._serper_api_key}
+        payload = {"q": query}
+        
+        response = requests.post(url, json=payload, headers=headers)
+        if response.status_code == 200:
+            try:
+                results = response.json()
+                return results.get("organic", [])
+            except Exception as e:
+                raise ValueError(f"Failed to parse JSON: {e}")
+        else:
+            raise Exception(f"Serper API error: {response.status_code}, {response.text}")
+
+    def execute_task(self, task: Task, context: dict = None, tools: list = None):
+        """Execute the task by performing a web search and returning results as a string."""
+        query = task.description
+        if not query:
+            raise ValueError("Task description must include a 'query' field.")
+        
+        search_results = self.web_crawl(query)
+
+        search_results_str = "\n".join(
+            [result.get('title', 'No Title') for result in search_results]
+        )
+    
+        return search_results_str

app.py

@@ -0,0 +1,98 @@
+import os
+import gradio as gr
+from crewai import Agent, Task, Crew, Process
+from langchain_together import ChatTogether
+from agents.web_crawler_agent import WebCrawlerAgent
+from agents.summarizer_agent import SummarizerAgent
+from agents.report_generator_agent import ReportGeneratorAgent
+from dotenv import load_dotenv
+
+load_dotenv()
+
+model = os.getenv("MODEL_NAME")
+os.environ['OPENAI_API_KEY'] = os.getenv("OPENAI_API_KEY")
+
+llm = ChatTogether(model=model)
+
+web_crawler_agent = WebCrawlerAgent(
+    llm=llm,
+    role="web crawler",
+    backstory="I search the web to gather information on a given topic.",
+    goal="Fetch web search results for a given query.",
+    serper_api_key=os.getenv("SERPER_API_KEY")
+)
+
+
+summarizer_agent = SummarizerAgent(llm)
+
+report_generator_agent = ReportGeneratorAgent(llm=llm)
+
+def generate_report(query, report_style):
+    web_crawler_task = Task(
+        description=query, 
+        expected_output="A detailed response about the query.",
+        agent=web_crawler_agent
+    )
+
+    summarizer_task = Task(
+        description="Summarize the web data.", 
+        expected_output="Concise summaries of the search results.", 
+        agent=summarizer_agent
+    )
+
+    report_task = Task(
+        description=f"Generate a final research report in '{report_style}' style by removing bias if any.",
+        expected_output="A comprehensive research report.",
+        agent=report_generator_agent
+    )
+
+    crew = Crew(
+        agents=[web_crawler_agent, summarizer_agent, report_generator_agent],
+        tasks=[web_crawler_task, summarizer_task, report_task],
+        verbose=True,
+        process=Process.sequential,
+        context_flow={
+            "summarizer_task": "web_crawler_task", 
+            "report_task": "summarizer_task",     
+        }
+    )
+
+    result = crew.kickoff()
+    formatted_report = f"###### Report Generated in {report_style} Style:\n\n{result.raw}"
+
+    return formatted_report
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 🚀 Research Report Generation and Analysis Tool")
+    gr.Markdown("#### Enter your query and choose the report style to generate a detailed, formatted report!")
+
+    with gr.Row():
+        query_input = gr.Textbox(
+            label="Enter Your Query", 
+            placeholder="e.g., Recent advancements in Indian Agriculture",
+            lines=2,
+            elem_id="query-input"
+        )
+
+    with gr.Row():
+        report_style = gr.Radio(
+            choices=[ "Detailed Analysis", "Brief Summary","Bullet Points"],
+            label="Select Report Style",
+            value="Detailed Analysis"
+        )
+    
+    with gr.Row():
+        generate_button = gr.Button("Generate Report", elem_id="generate-button", variant="primary")
+    
+    with gr.Row():
+        report_output = gr.Markdown(
+            label="Generated Report"
+        )
+    
+    generate_button.click(
+        generate_report, 
+        inputs=[query_input, report_style], 
+        outputs=[report_output]
+    )
+
+demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+gradio
+crewai
+requests
+python-dotenv
+langchain-together