Update app.py

app.py

@@ -1,10 +1,15 @@
 
+
 # app.py
 import os
+import yaml
 import streamlit as st
+from typing import List
 from dotenv import load_dotenv
-from crewai import Agent, Task, Crew, Process
-from crewai.tools import WebsiteSearchTool
+from crewai import Agent, Crew, Process, Task
+from crewai.project import CrewBase, agent, crew, task
+from crewai_tools import SerperDevTool, ScrapeWebsiteTool
+from pydantic import BaseModel, Field
 
 # Load environment variables
 load_dotenv()
@@ -32,88 +37,98 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-def run_market_research(topic):
+# Define output models
+class MarketResearchReport(BaseModel):
+    """Market research report model"""
+    title: str = Field(..., description="Title of the report")
+    executive_summary: str = Field(..., description="Executive summary of the report")
+    market_overview: str = Field(..., description="Market overview section")
+    key_findings: List[str] = Field(..., description="List of key findings")
+    recommendations: List[str] = Field(..., description="List of strategic recommendations")
+
+@CrewBase
+class MarketResearchCrew:
+    """Market Research crew"""
+    agents_config = 'config/agents.yaml'
+    tasks_config = 'config/tasks.yaml'
+
+    @agent
+    def research_analyst(self) -> Agent:
+        return Agent(
+            config=self.agents_config['research_analyst'],
+            tools=[SerperDevTool(), ScrapeWebsiteTool()],
+            verbose=True,
+            memory=False,
+        )
+
+    @agent
+    def data_analyst(self) -> Agent:
+        return Agent(
+            config=self.agents_config['data_analyst'],
+            tools=[SerperDevTool()],
+            verbose=True,
+            memory=False,
+        )
+
+    @agent
+    def report_writer(self) -> Agent:
+        return Agent(
+            config=self.agents_config['report_writer'],
+            verbose=True,
+            memory=False,
+        )
+
+    @task
+    def research_task(self) -> Task:
+        return Task(
+            config=self.tasks_config['research_task'],
+            agent=self.research_analyst()
+        )
+
+    @task
+    def analysis_task(self) -> Task:
+        return Task(
+            config=self.tasks_config['analysis_task'],
+            agent=self.data_analyst(),
+            context=[self.research_task()]
+        )
+
+    @task
+    def report_task(self) -> Task:
+        return Task(
+            config=self.tasks_config['report_task'],
+            agent=self.report_writer(),
+            context=[self.research_task(), self.analysis_task()],
+            output_json=MarketResearchReport
+        )
+
+    @crew
+    def crew(self) -> Crew:
+        """Creates the Market Research crew"""
+        return Crew(
+            agents=self.agents,
+            tasks=self.tasks,
+            process=Process.sequential,
+            verbose=2,
+        )
+
+def run_market_research(topic: str):
     with st.status("🤖 AI Crew working on your report...", expanded=True) as status:
         try:
-            # Use CrewAI's built-in tool
-            search_tool = WebsiteSearchTool()
-            
-            status.update(label="🔍 Creating AI agents...")
-            # Create agents
-            researcher = Agent(
-                role='Research Analyst',
-                goal=f'Conduct thorough market research about {topic}',
-                backstory='You are an experienced market research analyst with expertise in data analysis and trend identification.',
-                verbose=True,
-                tools=[search_tool],
-                allow_delegation=True
-            )
-            
-            analyst = Agent(
-                role='Data Analyst',
-                goal='Analyze research findings and identify key insights',
-                backstory='You are a skilled data analyst with expertise in interpreting market research and creating actionable insights.',
-                verbose=True,
-                allow_delegation=True
-            )
-            
-            writer = Agent(
-                role='Report Writer',
-                goal='Create comprehensive and engaging market research reports',
-                backstory='You are a professional writer specializing in creating clear and compelling business reports.',
-                verbose=True,
-                allow_delegation=True
-            )
-            
-            status.update(label="📋 Defining research tasks...")
-            # Define tasks
-            research_task = Task(
-                description=f"""
-                Conduct comprehensive market research on {topic}.
-                Focus on:
-                1. Current market size and growth projections
-                2. Key players and their market share
-                3. Consumer adoption trends
-                4. Regulatory environment
-                """,
-                agent=researcher
-            )
-            
-            analysis_task = Task(
-                description=f"""
-                Analyze the research findings for {topic} and identify:
-                1. Key market opportunities
-                2. Potential challenges
-                3. Growth drivers
-                4. Competitive dynamics
-                Use the research provided by the Research Analyst.
-                """,
-                agent=analyst
-            )
-            
-            report_task = Task(
-                description=f"""
-                Create a detailed market research report that includes:
-                1. Executive summary
-                2. Market overview
-                3. Key findings
-                4. Strategic recommendations
-                Base the report on the research and analysis provided by the other agents.
-                """,
-                agent=writer
-            )
+            status.update(label="🔍 Initializing research crew...")
+            crew_instance = MarketResearchCrew()
             
-            status.update(label="🚀 Assembling the crew and starting research...")
-            # Create the crew
-            market_research_crew = Crew(
-                agents=[researcher, analyst, writer],
-                tasks=[research_task, analysis_task, report_task],
-                verbose=2,
-                process=Process.sequential
-            )
+            status.update(label="🚀 Starting research process...")
+            inputs = {
+                'topic': topic,
+                'project_description': f"""
+                Conduct comprehensive market research for {topic}.
+                The research should cover market size, growth projections, key players,
+                consumer trends, and regulatory environment.
+                """
+            }
             
-            # Execute the crew's tasks
-            result = market_research_crew.kickoff()
+            result = crew_instance.crew().kickoff(inputs=inputs)
             status.update(label="✅ Report generated successfully!", state="complete")
             return result
 
@@ -131,8 +146,6 @@ def main():
     - ✍️ A Report Writer who creates the final document
     """)
     
-    # Input section
-    st.subheader("Research Topic")
     topic = st.text_input(
         "Enter the market or product you want to research",
         placeholder="e.g., Electric Vehicles Market",
@@ -153,7 +166,6 @@ def main():
                 st.markdown(report)
                 st.markdown('</div>', unsafe_allow_html=True)
                 
-                # Add download button
                 st.download_button(
                     label="Download Report",
                     data=report,