Update report_generator.py

report_generator.py

@@ -8,125 +8,91 @@ from reportlab.lib.units import inch
 import io
 import streamlit as st
 
-def generate_direct_report(topic, questions=None):
+def generate_direct_report(topic, questions):
     llm = ChatOpenAI(temperature=0.7, model="gpt-4")
+    serper = SerperDevTool()
     
-    # First prompt for research with more detailed requirements
-    research_prompt = f"""Conduct comprehensive market research on: {topic}
-
-    Create an extremely detailed analysis covering:
-    1. Current Market Overview
-        - Detailed market size with regional breakdowns
-        - Historical growth patterns (last 5 years)
-        - Market segmentation analysis
-        - Value chain analysis
+    # First, gather research data
+    research_prompt = f"""Research the current market status of {topic}.
+    
+    Specifically focus on:
+    1. Market size and recent growth
+    2. Key competitors and their market shares
+    3. Major trends and developments
+    4. Regional analysis
+    5. Future projections
+    
+    Also address these specific questions:
+    {questions}
+    
+    Use real market data and statistics."""
+    
+    try:
+        # Use SerperDev to get real market data
+        search_results = serper.search(f"{topic} market size statistics trends analysis")
+        
+        # Research using the search results
+        research_data = llm.invoke(research_prompt + f"\n\nUse this market data: {search_results}")
+        
+        # Generate comprehensive report
+        report_prompt = f"""Create a detailed market research report for {topic} using this research data:
+        {research_data.content}
+        
+        Address these specific questions:
+        {questions}
         
-    2. Competitive Landscape
-        - Top 10 key players with market shares
-        - Detailed company profiles
+        Format the report with:
+        # Executive Summary
+        [Provide 2-3 paragraphs summary with key points]
+        
+        # Market Overview
+        - Current market size
+        - Growth rate
+        - Market segments
+        - Geographic distribution
+        
+        # Competitive Analysis
+        - Key players
+        - Market shares
         - Competitive strategies
-        - Recent developments and innovations
+        - SWOT analysis
         
-    3. Market Dynamics
-        - Comprehensive growth drivers analysis
-        - Detailed market challenges
-        - Opportunities in the market
-        - Regulatory landscape
+        # Market Trends
+        - Current trends
+        - Technology impacts
+        - Consumer behavior
+        - Regulatory factors
         
-    4. Regional Analysis
-        - Market size by region
-        - Growth rates by region
-        - Regional market characteristics
-        - Key regional players
+        # Regional Analysis
+        - Key regions
+        - Growth by region
+        - Market characteristics
         
-    5. Technology Trends
-        - Current technology landscape
-        - Emerging technologies
-        - Innovation patterns
-        - Technology adoption rates
+        # Future Outlook
+        - Growth projections
+        - Opportunities
+        - Challenges
         
-    6. Consumer Analysis
-        - Buying patterns
-        - Consumer preferences
-        - Demographics
-        - Regional variations
-
-    7. Future Outlook
-        - 5-year projections
+        # Strategic Recommendations
+        - Key strategies
         - Growth opportunities
-        - Potential challenges
-        - Market evolution scenarios
-
-    Provide extensive data-backed insights with specific numbers, statistics, and multiple examples for each section.
-    Include real market data and examples where possible."""
-
-    if questions:
-        research_prompt += f"\nAdditional questions to address:\n{questions}" 
-
-    research_data = llm.invoke(research_prompt).content
-    
-    # Second prompt for comprehensive report
-    report_prompt = f"""Create an extensive professional market research report using this research data:
-    {research_data}
-    
-    Structure the report with these detailed sections:
-
-    # Executive Summary
-    - Comprehensive overview (2-3 paragraphs)
-    - Key findings (at least 6 bullet points)
-    - Critical market highlights (4-5 paragraphs)
-    
-    # Market Overview
-    - Detailed current market size analysis (2-3 paragraphs)
-    - Historical growth patterns (2 paragraphs)
-    - Segmentation analysis (5-6 bullet points per segment)
-    - Value chain breakdown (detailed list with explanations)
-    
-    # Competitive Landscape
-    - Key players profiles (minimum 8-10 companies)
-    - Market share analysis (detailed breakdown)
-    - Competitive strategies (5-6 strategies per player)
-    - SWOT analysis for top 3 players
-    
-    # Market Dynamics
-    - Growth drivers (6-8 detailed points)
-    - Market challenges (6-8 detailed points)
-    - Opportunities (6-8 detailed points)
-    - Entry barriers (4-5 detailed points)
-    
-    # Regional Analysis
-    - North America (2-3 paragraphs)
-    - Europe (2-3 paragraphs)
-    - Asia Pacific (2-3 paragraphs)
-    - Rest of World (2-3 paragraphs)
-    
-    # Industry Trends
-    - Current technology trends (5-6 points)
-    - Consumer behavior patterns (5-6 points)
-    - Regulatory landscape (3-4 paragraphs)
-    - Innovation trends (5-6 points)
-    
-    # Future Outlook
-    - Market projections (2-3 paragraphs)
-    - Growth scenarios (3 scenarios with details)
-    - Emerging opportunities (6-8 points)
-    - Risk factors (6-8 points)
-    
-    # Strategic Recommendations
-    - Short-term strategies (5-6 detailed points)
-    - Long-term opportunities (5-6 detailed points)
-    - Risk mitigation strategies (5-6 points)
-    
-    Format in clean markdown with clear headers and bullet points.
-    Each section should be highly detailed and comprehensive.
-    Use specific data points, numbers, and statistics throughout the report."""
-    
-    final_report = llm.invoke(report_prompt).content
-    
-    return {
-        'research_data': research_data,
-        'final_report': final_report
-    }
+        - Risk mitigation
+        
+        Use specific data points and statistics throughout the report."""
+        
+        final_report = llm.invoke(report_prompt)
+        
+        return {
+            'research_data': research_data.content,
+            'final_report': final_report.content
+        }
+        
+    except Exception as e:
+        st.error(f"Error generating direct report: {str(e)}")
+        return {
+            'research_data': "Error collecting research data",
+            'final_report': "Error generating report"
+        }
 
 def create_pdf_report(report_content, topic):
     # Create an in-memory bytes buffer