Create app.py

app.py

@@ -0,0 +1,164 @@
+# app.py
+import os
+import streamlit as st
+from dotenv import load_dotenv
+from crewai import Agent, Task, Crew, Process
+from langchain.tools import DuckDuckGoSearchRun
+
+# Load environment variables
+load_dotenv()
+
+# Set page config
+st.set_page_config(
+    page_title="Market Research Generator",
+    page_icon="📊",
+    layout="wide"
+)
+
+# Add custom CSS
+st.markdown("""
+    <style>
+        .stButton>button {
+            width: 100%;
+            margin-top: 20px;
+        }
+        .report-container {
+            background-color: #f0f2f6;
+            padding: 20px;
+            border-radius: 10px;
+            margin-top: 20px;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+def run_market_research(topic):
+    with st.status("🤖 AI Crew working on your report...", expanded=True) as status:
+        try:
+            # Initialize the search tool
+            search_tool = DuckDuckGoSearchRun()
+            
+            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="🚀 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
+            )
+            
+            # Execute the crew's tasks
+            result = market_research_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")
+            return None
+
+def main():
+    st.title("🤖 AI Market Research Generator")
+    st.markdown("""
+    This tool uses a crew of AI agents to generate comprehensive market research reports. 
+    Each report is created through a collaborative process involving:
+    - 🔍 A Research Analyst who gathers market data
+    - 📊 A Data Analyst who processes the findings
+    - ✍️ 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",
+        key="research_topic"
+    )
+    
+    if st.button("Generate Report", type="primary"):
+        if not topic:
+            st.error("Please enter a research topic")
+            return
+            
+        report = run_market_research(topic)
+        
+        if report:
+            st.subheader("📑 Generated Report")
+            with st.container():
+                st.markdown('<div class="report-container">', unsafe_allow_html=True)
+                st.markdown(report)
+                st.markdown('</div>', unsafe_allow_html=True)
+                
+                # Add download button
+                st.download_button(
+                    label="Download Report",
+                    data=report,
+                    file_name=f"market_research_{topic.lower().replace(' ', '_')}.md",
+                    mime="text/markdown"
+                )
+
+if __name__ == "__main__":
+    main()