Update utils.py

utils.py

@@ -10,11 +10,18 @@ def update_progress(container, percentage, message=""):
 
 def extract_data_points(text):
     llm = ChatOpenAI(temperature=0, model="gpt-4")
-    prompt = """Extract market metrics from this text as JSON:
+    prompt = """Extract key market metrics as JSON:
     {text}
     
-    Return format:
-    {{"market_size": "value", "cagr": "value", "market_share_leader": "value", "total_players": "value"}}"""
+    Include:
+    - Market size (with currency)
+    - CAGR
+    - Market leader share
+    - Number of major players
+    - Key regions
+    - Dominant segment
+    
+    Format as JSON with these exact keys."""
     
     response = llm.invoke(prompt.format(text=text))
     try:
@@ -23,48 +30,106 @@ def extract_data_points(text):
     except:
         return {}
 
-def expand_content(base_content, topic):
+def generate_executive_summary(base_content, topic):
+    llm = ChatOpenAI(temperature=0.7, model="gpt-4")
+    prompt = f"""Create an executive summary for {topic} market research:
+    {base_content}
+    
+    Include:
+    1. Brief market overview (2-3 sentences)
+    2. Key findings (3-4 bullet points)
+    3. Growth drivers
+    4. Market challenges
+    5. Strategic recommendations
+    
+    Format in markdown with clear headers."""
+    
+    return llm.invoke(prompt).content
+
+def generate_detailed_analysis(base_content, topic):
     llm = ChatOpenAI(temperature=0.7, model="gpt-4")
-    prompt = f"""Expand this market research content for {topic}:
+    prompt = f"""Create detailed market analysis for {topic}:
     {base_content}
     
-    Provide detailed analysis covering:
-    1. Current market state
-    2. Growth drivers and barriers
-    3. Competitive dynamics
-    4. Future outlook
+    Sections to include:
+    1. Market Overview
+       - Current state
+       - Size and growth
+       - Geographic distribution
+    
+    2. Competitive Analysis
+       - Key players
+       - Market shares
+       - Competitive strategies
+    
+    3. Industry Trends
+       - Technology trends
+       - Consumer trends
+       - Regulatory trends
+    
+    4. Future Outlook
+       - Growth projections
+       - Emerging opportunities
+       - Potential challenges
     
-    Use markdown formatting."""
+    5. Strategic Recommendations
+       - Short-term actions
+       - Long-term strategy
+       - Risk mitigation
+    
+    Format using markdown with clear sections and subsections.
+    Include sources where available."""
     
     return llm.invoke(prompt).content
 
 def process_crew_output(crew_result, topic):
     try:
-        # Directly use the crew result object
         report_text = str(crew_result)
         metrics = extract_data_points(report_text)
-        expanded_content = expand_content(report_text, topic)
+        exec_summary = generate_executive_summary(report_text, topic)
+        detailed_analysis = generate_detailed_analysis(report_text, topic)
         
         return {
             'metrics': metrics,
-            'content': expanded_content
+            'executive_summary': exec_summary,
+            'detailed_analysis': detailed_analysis,
+            'raw_content': report_text
         }
     except Exception as e:
         st.error(f"Error processing report: {str(e)}")
-        return {'metrics': {}, 'content': ''}
+        return {'metrics': {}, 'executive_summary': '', 'detailed_analysis': ''}
 
 def display_enhanced_report(report):
     try:
+        # Metrics Dashboard
         if 'metrics' in report and report['metrics']:
-            st.subheader("Key Market Metrics")
-            cols = st.columns(4)
+            st.subheader("📊 Key Market Metrics")
+            cols = st.columns(3)
             metrics = report['metrics']
             cols[0].metric("Market Size", metrics.get('market_size', 'N/A'))
-            cols[1].metric("CAGR", metrics.get('cagr', 'N/A'))
-            cols[2].metric("Leader Share", metrics.get('market_share_leader', 'N/A'))
-            cols[3].metric("Key Players", metrics.get('total_players', 'N/A'))
+            cols[0].metric("CAGR", metrics.get('cagr', 'N/A'))
+            cols[1].metric("Market Leader Share", metrics.get('market_leader_share', 'N/A'))
+            cols[1].metric("Major Players", metrics.get('major_players', 'N/A'))
+            cols[2].metric("Key Region", metrics.get('key_regions', 'N/A'))
+            cols[2].metric("Dominant Segment", metrics.get('dominant_segment', 'N/A'))
+        
+        # Executive Summary
+        if 'executive_summary' in report and report['executive_summary']:
+            with st.expander("📑 Executive Summary", expanded=True):
+                st.markdown(report['executive_summary'])
+        
+        # Detailed Analysis
+        if 'detailed_analysis' in report and report['detailed_analysis']:
+            st.header("📈 Detailed Market Analysis")
+            st.markdown(report['detailed_analysis'])
         
-        if 'content' in report and report['content']:
-            st.markdown(report['content'])
+        # Download Options
+        if 'raw_content' in report:
+            st.download_button(
+                "📥 Download Full Report",
+                report['raw_content'],
+                "market_research_report.md",
+                "text/markdown"
+            )
     except Exception as e:
         st.error(f"Error displaying report: {str(e)}")