Update app.py

app.py

@@ -1,15 +1,10 @@
-
-
 # app.py
 import os
-import yaml
 import streamlit as st
 from typing import List
 from dotenv import load_dotenv
 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()
@@ -37,103 +32,105 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# 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:
-            status.update(label="🔍 Initializing research crew...")
-            crew_instance = MarketResearchCrew()
+            status.update(label="🔍 Creating AI agents...")
+            
+            # Initialize tools
+            search_tool = SerperDevTool()
+            scrape_tool = ScrapeWebsiteTool()
+            
+            # 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.',
+                tools=[search_tool, scrape_tool],
+                verbose=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.',
+                tools=[search_tool],
+                verbose=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
+            )
+            
+            status.update(label="📋 Defining research tasks...")
+            
+            # Define tasks with expected outputs
+            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
+                
+                Output should be a detailed research findings document with data and sources.
+                """,
+                agent=researcher,
+                expected_output="A comprehensive research document containing market data, trends, and competitor analysis with clear citations and sources."
+            )
+            
+            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 to create an analytical report.
+                """,
+                agent=analyst,
+                expected_output="An analytical report with key insights, opportunities, challenges, and strategic implications based on the research data.",
+                context=[research_task]
+            )
+            
+            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.
+                The report should be well-structured, professional, and actionable.
+                """,
+                agent=writer,
+                expected_output="A complete market research report in markdown format with executive summary, findings, and recommendations.",
+                context=[research_task, analysis_task]
+            )
+            
+            status.update(label="🚀 Assembling the crew and starting research...")
             
-            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.
-                """
-            }
+            # Create and run the crew
+            crew = Crew(
+                agents=[researcher, analyst, writer],
+                tasks=[research_task, analysis_task, report_task],
+                verbose=2,
+                process=Process.sequential
+            )
             
-            result = crew_instance.crew().kickoff(inputs=inputs)
+            result = crew.kickoff()
             status.update(label="✅ Report generated successfully!", state="complete")
             return result
 
         except Exception as e:
             status.update(label=f"❌ Error: {str(e)}", state="error")
+            st.error(f"Error details: {str(e)}")
             return None
 
 def main():
@@ -146,6 +143,8 @@ 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",
@@ -166,6 +165,7 @@ def main():
                 st.markdown(report)
                 st.markdown('</div>', unsafe_allow_html=True)
                 
+                # Add download button
                 st.download_button(
                     label="Download Report",
                     data=report,