Update app.py

app.py

@@ -1,11 +1,101 @@
 # app.py
 import os
-import time
-import re
+from utils import ReportManager
 import streamlit as st
 from dotenv import load_dotenv
 from crewai import Agent, Crew, Process, Task
 
+# Initialize report manager
+report_manager = ReportManager()
+
+def create_research_crew(topic: str):
+    researcher = Agent(
+        role='Research Analyst',
+        goal=f'Conduct thorough market research about {topic} with verifiable sources',
+        backstory='You are an experienced market research analyst with expertise in data analysis and trend identification. You always provide specific numbers and data points with sources.',
+        verbose=True
+    )
+    
+    analyst = Agent(
+        role='Data Analyst',
+        goal='Create data-driven insights with specific metrics and visualizable data points',
+        backstory='You are a skilled data analyst who specializes in turning research into actionable insights. You focus on quantifiable metrics and clear data points that can be visualized.',
+        verbose=True
+    )
+    
+    writer = Agent(
+        role='Report Writer',
+        goal='Create both executive summary and detailed reports with proper citations',
+        backstory='You are a professional writer who creates clear, structured reports with both high-level summaries and detailed analysis. You always include sources and references.',
+        verbose=True
+    )
+
+    # Define tasks with improved prompts
+    research_task = Task(
+        description=f"""
+        Research {topic} with a focus on:
+        1. Market size (provide specific numbers)
+        2. Growth rates and projections
+        3. Key players and market share percentages
+        4. Competitive analysis
+        5. Find and verify credible sources for all data points
+        
+        Format data in a way that can be easily visualized.
+        Include URLs and references for all sources.
+        """,
+        agent=researcher,
+        expected_output="Detailed research data with sources and numerical data points"
+    )
+
+    analysis_task = Task(
+        description=f"""
+        Analyze the research findings and:
+        1. Create growth projections with specific percentages
+        2. Break down market share data
+        3. Identify key trends with supporting data
+        4. Analyze competitive landscape with market positions
+        
+        Present all data in a format suitable for charts and graphs.
+        """,
+        agent=analyst,
+        expected_output="Analysis with specific metrics and visualization-ready data",
+        context=[research_task]
+    )
+
+    report_task = Task(
+        description=f"""
+        Create two report versions:
+        1. Executive Summary (2-3 paragraphs)
+        - Key findings and recommendations
+        - Major metrics and insights
+        - Strategic implications
+        
+        2. Detailed Report
+        - Comprehensive market analysis
+        - Detailed data presentation
+        - Competitive analysis
+        - Future projections
+        - Source citations and references
+        
+        Format the output as JSON with the following structure:
+        {{
+            "exec_summary": "executive summary text",
+            "detailed_report": "detailed report text",
+            "sources": ["source1", "source2", ...]
+        }}
+        """,
+        agent=writer,
+        expected_output="JSON containing exec summary, detailed report, and sources",
+        context=[research_task, analysis_task]
+    )
+
+    return Crew(
+        agents=[researcher, analyst, writer],
+        tasks=[research_task, analysis_task, report_task],
+        verbose=True,
+        process=Process.sequential
+    )
+
 # Load environment variables
 load_dotenv()
 
@@ -305,71 +395,102 @@ def run_market_research(topic: str, progress_bar, chat_container):
 def main():
     st.title("🤖 AI Market Research Generator")
     
-    if 'current_tab' not in st.session_state:
-        st.session_state.current_tab = "generate"
-    
     # Create tabs
-    tab1, tab2 = st.tabs(["Generate Report", "View Report"])
+    tab1, tab2, tab3 = st.tabs(["Generate Report", "View Reports", "About"])
     
     with tab1:
-        if st.session_state.current_tab == "generate":
-            col1, col2 = st.columns([2, 3])
+        col1, col2 = st.columns([2, 3])
+        
+        with col1:
+            st.subheader("Enter Research Topic")
+            topic = st.text_input(
+                "What market would you like to research?",
+                placeholder="e.g., Electric Vehicles Market"
+            )
             
-            with col1:
-                st.subheader("Enter Research Topic")
-                topic = st.text_input(
-                    "What market would you like 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
                 
-                if st.button("Generate Report", type="primary"):
-                    if not topic:
-                        st.error("Please enter a research topic")
-                        return
-                    
-                    # Create progress bar and containers
-                    progress = st.progress(0)
-                    chat_container = st.container()
+                with st.spinner("Generating report..."):
+                    # Create and run crew
+                    crew = create_research_crew(topic)
+                    result = crew.kickoff()
                     
-                    # Show agent flow diagram
-                    create_agent_flow(col2)
+                    # Process and save report
+                    try:
+                        report_data = json.loads(result)
+                        # Generate visualizations
+                        charts = report_manager.generate_visualizations(report_data['detailed_report'])
+                        report_data['charts'] = charts
+                        
+                        # Save report
+                        report_id = report_manager.save_report(topic, report_data)
+                        st.success("Report generated successfully!")
+                        
+                        # Switch to the view tab
+                        st.session_state.current_report = report_id
+                        st.rerun()
                     
-                    # Run the research
-                    report = run_market_research(topic, progress, chat_container)
-                    
-                    if report:
-                        # Store results in session state
-                        st.session_state.report = report
-                        st.session_state.current_tab = "report"
-                        st.rerun()  # Using st.rerun() instead of experimental_rerun
+                    except Exception as e:
+                        st.error(f"Error processing report: {str(e)}")
     
     with tab2:
-        if st.session_state.current_tab == "report" and hasattr(st.session_state, 'report'):
-            st.subheader("📊 Market Research Report")
+        st.subheader("Saved Reports")
+        reports = report_manager.get_all_reports()
+        
+        if not reports:
+            st.info("No reports generated yet. Create your first report in the Generate tab!")
+            return
+        
+        # Report selection
+        report_options = {f"{data['topic']} ({data['timestamp']})": rid 
+                         for rid, data in reports.items()}
+        selected_report = st.selectbox(
+            "Select a report to view",
+            options=list(report_options.keys())
+        )
+        
+        if selected_report:
+            report_id = report_options[selected_report]
+            report = report_manager.get_report(report_id)
             
-            # Display report in sections
-            sections = st.session_state.report.split('\n\n')
-            for section in sections:
-                if section.strip():
-                    st.markdown(f"""
-                        <div class="report-section">
-                            {section}
-                        </div>
-                    """, unsafe_allow_html=True)
+            # Display report
+            col1, col2 = st.columns([2, 1])
             
-            # Add 'Generate New Report' button
-            if st.button("Generate New Report"):
-                st.session_state.current_tab = "generate"
-                st.rerun()
+            with col1:
+                st.subheader("Executive Summary")
+                st.markdown(report['exec_summary'])
+                
+                st.subheader("Detailed Report")
+                st.markdown(report['detailed_report'])
+                
+                st.subheader("Sources")
+                for source in report['sources']:
+                    st.markdown(f"- {source}")
             
-            # Download button
-            st.download_button(
-                label="Download Report",
-                data=st.session_state.report,
-                file_name=f"market_research_{st.session_state.get('research_topic', 'report').lower().replace(' ', '_')}.md",
-                mime="text/markdown"
-            )
+            with col2:
+                st.subheader("Visualizations")
+                for chart in report['charts']:
+                    st.plotly_chart(chart, use_container_width=True)
+            
+            # Download buttons
+            col1, col2 = st.columns(2)
+            with col1:
+                st.download_button(
+                    "Download as PDF",
+                    data=report_manager.generate_pdf(report_id),
+                    file_name=f"{report['topic']}_report.pdf",
+                    mime="application/pdf"
+                )
+            with col2:
+                st.download_button(
+                    "Download as Markdown",
+                    data=report['detailed_report'],
+                    file_name=f"{report['topic']}_report.md",
+                    mime="text/markdown"
+                )
 
 if __name__ == "__main__":
     main()