Update utils.py

utils.py

@@ -1,8 +1,12 @@
-from langchain_openai import ChatOpenAI
-import streamlit as st
+import re
+import json
 import pandas as pd
 import plotly.express as px
 from datetime import datetime
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import letter, landscape
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, PageBreak
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 
 def update_progress(container, percentage, message=""):
     if container:
@@ -221,82 +225,174 @@ def display_presentation_slide(slide, slide_num, total_slides):
         if slide_num < total_slides - 1:
             st.button("Next ➡️", key=f"next_{slide_num}")
             
+# utils.py
+import re
+import json
+import pandas as pd
+import plotly.express as px
+from datetime import datetime
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import letter, landscape
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, PageBreak
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+
 def display_report(report_data):
     try:
-        # Display metrics dashboard
-        if 'metrics' in report_data and report_data['metrics']:
-            st.subheader("📊 Market Metrics")
-            cols = st.columns(3)
-            metrics = report_data['metrics']
-            
-            cols[0].metric("Market Size", metrics.get('market_size', 'N/A'))
-            cols[0].metric("CAGR", metrics.get('cagr', 'N/A'))
-            cols[1].metric("Leader Share", metrics.get('leader_share', 'N/A'))
-            cols[1].metric("Key Players", metrics.get('key_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'))
+        # Display key metrics in tiles
+        st.write("### 📊 Key Market Insights")
+        metrics = report_data.get('metrics', {})
         
-        # Display enhanced report content
-        if 'content' in report_data and report_data['content']:
-            st.markdown(report_data['content'])
+        # Create metric tiles in a grid
+        col1, col2, col3 = st.columns(3)
         
-        # Add download button
-        if 'raw' in report_data:
-            st.download_button(
-                "📥 Download Full Report",
-                report_data['raw'],
-                "market_research_report.md",
-                "text/markdown"
-            )
-            
+        with col1:
+            st.markdown("""
+            <div style='background-color: #f0f7ff; padding: 20px; border-radius: 10px; height: 150px;'>
+                <h4 style='color: #1e88e5;'>Market Size</h4>
+                <h2>{}</h2>
+                <p>CAGR: {}</p>
+            </div>
+            """.format(
+                metrics.get('market_size', 'N/A'),
+                metrics.get('cagr', 'N/A')
+            ), unsafe_allow_html=True)
+
+        with col2:
+            st.markdown("""
+            <div style='background-color: #fff8e1; padding: 20px; border-radius: 10px; height: 150px;'>
+                <h4 style='color: #ffa000;'>Market Leadership</h4>
+                <h2>{}</h2>
+                <p>Key Players: {}</p>
+            </div>
+            """.format(
+                metrics.get('leader_share', 'N/A'),
+                metrics.get('key_players', 'N/A')
+            ), unsafe_allow_html=True)
+
+        with col3:
+            st.markdown("""
+            <div style='background-color: #e8f5e9; padding: 20px; border-radius: 10px; height: 150px;'>
+                <h4 style='color: #43a047;'>Regional Focus</h4>
+                <h2>{}</h2>
+                <p>Dominant Segment: {}</p>
+            </div>
+            """.format(
+                metrics.get('key_regions', 'N/A'),
+                metrics.get('dominant_segment', 'N/A')
+            ), unsafe_allow_html=True)
+
+        # Create tabs for different sections of the report
+        report_tabs = st.tabs([
+            "Executive Summary", 
+            "Market Analysis", 
+            "Competitive Landscape",
+            "Regional Analysis",
+            "Future Outlook"
+        ])
+
+        # Split content into sections
+        content = report_data.get('content', '')
+        sections = content.split('#')
+
+        with report_tabs[0]:
+            st.markdown("""
+            <div style='background-color: white; padding: 20px; border-radius: 10px; border-left: 5px solid #1e88e5;'>
+            """, unsafe_allow_html=True)
+            st.markdown(sections[1] if len(sections) > 1 else "Executive Summary not available")
+            st.markdown("</div>", unsafe_allow_html=True)
+
+        with report_tabs[1]:
+            st.markdown("""
+            <div style='background-color: white; padding: 20px; border-radius: 10px; border-left: 5px solid #43a047;'>
+            """, unsafe_allow_html=True)
+            st.markdown(sections[2] if len(sections) > 2 else "Market Analysis not available")
+            st.markdown("</div>", unsafe_allow_html=True)
+
+            # Add market visualizations if available
+            if 'market_data' in report_data:
+                display_market_visualizations(report_data['market_data'])
+
+        with report_tabs[2]:
+            st.markdown("""
+            <div style='background-color: white; padding: 20px; border-radius: 10px; border-left: 5px solid #ffa000;'>
+            """, unsafe_allow_html=True)
+            st.markdown(sections[3] if len(sections) > 3 else "Competitive Landscape not available")
+            st.markdown("</div>", unsafe_allow_html=True)
+
+            # Add competitor table if available
+            if 'market_data' in report_data and 'keyPlayers' in report_data['market_data']:
+                display_competitor_table(report_data['market_data']['keyPlayers'])
+
+        with report_tabs[3]:
+            st.markdown("""
+            <div style='background-color: white; padding: 20px; border-radius: 10px; border-left: 5px solid #e91e63;'>
+            """, unsafe_allow_html=True)
+            st.markdown(sections[4] if len(sections) > 4 else "Regional Analysis not available")
+            st.markdown("</div>", unsafe_allow_html=True)
+
+        with report_tabs[4]:
+            st.markdown("""
+            <div style='background-color: white; padding: 20px; border-radius: 10px; border-left: 5px solid #9c27b0;'>
+            """, unsafe_allow_html=True)
+            st.markdown(sections[5] if len(sections) > 5 else "Future Outlook not available")
+            st.markdown("</div>", unsafe_allow_html=True)
+
     except Exception as e:
         st.error(f"Error displaying report: {str(e)}")
 
-# Add to utils.py
-def display_market_visualizations(market_data):
-    """Display market visualizations using Streamlit"""
-    st.subheader("📊 Market Analysis Visualizations")
+def create_presentation_slides(report_data):
+    """Convert report data into presentation-style slides"""
+    slides = []
     
-    # Market Share Pie Chart
-    if 'marketShares' in market_data:
-        st.write("### Market Share Distribution")
-        share_data = pd.DataFrame(market_data['marketShares'])
-        fig_pie = px.pie(share_data, 
-                        values='share', 
-                        names='company',
-                        title='Market Share Distribution')
-        st.plotly_chart(fig_pie, use_container_width=True)
+    # Slide 1: Title
+    slides.append({
+        'title': "Executive Summary",
+        'type': 'title',
+        'content': report_data.get('title', 'Market Research Report')
+    })
     
-    # Growth Trend Line Chart
-    if 'growthTrend' in market_data:
-        st.write("### Growth Trend Analysis")
-        growth_data = pd.DataFrame(market_data['growthTrend'])
-        fig_line = px.line(growth_data, 
-                          x='year', 
-                          y='growth',
-                          title='Market Growth Trend')
-        st.plotly_chart(fig_line, use_container_width=True)
+    # Slide 2: Key Metrics
+    if 'metrics' in report_data:
+        slides.append({
+            'title': "Key Market Metrics",
+            'type': 'metrics',
+            'content': report_data['metrics']
+        })
     
-    # Regional Distribution Bar Chart
-    if 'regionalDistribution' in market_data:
-        st.write("### Regional Market Distribution")
-        regional_data = pd.DataFrame(market_data['regionalDistribution'])
-        fig_bar = px.bar(regional_data, 
-                        x='region', 
-                        y='share',
-                        title='Regional Market Distribution')
-        st.plotly_chart(fig_bar, use_container_width=True)
-
-def display_market_tables(market_data):
-    """Display market data tables using Streamlit"""
-    st.subheader("📋 Detailed Market Analysis")
+    # Extract sections from content
+    content = report_data.get('content', '')
+    sections = content.split('#')
+    
+    # Create slides for each major section
+    for section in sections[1:]:  # Skip the first empty section
+        if section.strip():
+            title = section.split('\n')[0].strip()
+            content = '\n'.join(section.split('\n')[1:]).strip()
+            slides.append({
+                'title': title,
+                'type': 'text',
+                'content': content
+            })
     
-    # Key Players Table
-    if 'keyPlayers' in market_data:
-        st.write("### Key Players Analysis")
-        players_df = pd.DataFrame(market_data['keyPlayers'])
+    return slides
+
+def display_market_visualizations(market_data):
+    """Display market visualizations using Plotly"""
+    if 'marketShares' in market_data:
+        fig = px.pie(
+            market_data['marketShares'], 
+            values='share', 
+            names='company',
+            title='Market Share Distribution'
+        )
+        st.plotly_chart(fig, use_container_width=True)
+
+def display_competitor_table(competitors_data):
+    """Display competitor information in a styled table"""
+    if competitors_data:
+        df = pd.DataFrame(competitors_data)
         st.dataframe(
-            players_df,
+            df,
             column_config={
                 "company": "Company",
                 "marketShare": st.column_config.NumberColumn(
@@ -306,46 +402,12 @@ def display_market_tables(market_data):
                 "strengths": "Key Strengths",
                 "developments": "Recent Developments"
             },
-            hide_index=True
-        )
-    
-    # Regional Analysis Table
-    if 'regionalAnalysis' in market_data:
-        st.write("### Regional Market Analysis")
-        regional_df = pd.DataFrame(market_data['regionalAnalysis'])
-        st.dataframe(
-            regional_df,
-            column_config={
-                "name": "Region",
-                "marketSize": "Market Size",
-                "growthRate": st.column_config.NumberColumn(
-                    "Growth Rate (%)",
-                    format="%.1f%%"
-                ),
-                "trends": "Key Trends"
-            },
-            hide_index=True
-        )
-    
-    # Technology Adoption Table
-    if 'techAdoption' in market_data:
-        st.write("### Technology Adoption Analysis")
-        tech_df = pd.DataFrame(market_data['techAdoption'])
-        st.dataframe(
-            tech_df,
-            column_config={
-                "name": "Technology",
-                "adoptionRate": st.column_config.NumberColumn(
-                    "Adoption Rate (%)",
-                    format="%.1f%%"
-                ),
-                "impact": "Impact",
-                "outlook": "Future Outlook"
-            },
+            use_container_width=True,
             hide_index=True
         )
 
 
+
 def apply_report_styling():
     return """
     <style>
@@ -456,71 +518,6 @@ def apply_report_styling():
     </style>
     """
 
-def display_report(report_data):
-    try:
-        # Apply custom styling
-        st.markdown(apply_report_styling(), unsafe_allow_html=True)
-        
-        # Display metrics dashboard
-        if 'metrics' in report_data and report_data['metrics']:
-            st.subheader("📊 Market Metrics Dashboard")
-            metrics = report_data['metrics']
-            cols = st.columns(3)
-            
-            with cols[0]:
-                st.metric("Market Size", metrics.get('market_size', 'N/A'))
-                st.metric("CAGR", metrics.get('cagr', 'N/A'))
-            with cols[1]:
-                key_players = metrics.get('key_players', [])
-                key_players_count = len(key_players) if isinstance(key_players, list) else 'N/A'
-                st.metric("Market Leader Share", metrics.get('leader_share', 'N/A'))
-                st.metric("Number of Key Players", key_players_count)
-            with cols[2]:
-                st.metric("Key Region", metrics.get('key_regions', 'N/A'))
-                st.metric("Dominant Segment", metrics.get('dominant_segment', 'N/A'))
-        
-        # Display visualizations
-        if 'market_data' in report_data:
-            display_market_visualizations(report_data['market_data'])
-        
-        # Display data tables
-        if 'market_data' in report_data:
-            display_market_tables(report_data['market_data'])
-        
-        # Display main report content
-        if 'content' in report_data:
-            st.markdown("## 📑 Comprehensive Report")
-            st.markdown(report_data['content'])
-        
-        # Display agent outputs
-        if 'agent_outputs' in report_data:
-            st.markdown("## 🤖 Agent Analysis - Raw Feed")
-            tabs = st.tabs(["Researcher", "Analyst", "Writer"])
-            
-            with tabs[0]:
-                researcher_data = report_data['agent_outputs'].get('researcher', {})
-                st.markdown(f"**Timestamp:** {researcher_data.get('timestamp', 'N/A')}")
-                st.markdown(f"**Analysis Type:** {researcher_data.get('analysis_type', 'N/A')}")
-                st.markdown("### Raw Output")
-                st.markdown(researcher_data.get('raw_output', 'No output available'))
-            
-            with tabs[1]:
-                analyst_data = report_data['agent_outputs'].get('analyst', {})
-                st.markdown(f"**Timestamp:** {analyst_data.get('timestamp', 'N/A')}")
-                st.markdown(f"**Analysis Type:** {analyst_data.get('analysis_type', 'N/A')}")
-                st.markdown("### Raw Output")
-                st.markdown(analyst_data.get('raw_output', 'No output available'))
-            
-            with tabs[2]:
-                writer_data = report_data['agent_outputs'].get('writer', {})
-                st.markdown(f"**Timestamp:** {writer_data.get('timestamp', 'N/A')}")
-                st.markdown(f"**Analysis Type:** {writer_data.get('analysis_type', 'N/A')}")
-                st.markdown("### Raw Output")
-                st.markdown(writer_data.get('raw_output', 'No output available'))
-        
-    except Exception as e:
-        st.error(f"Error displaying report: {str(e)}")
-
     
 def extract_sources(text):
     pattern = r'(?:Source|Reference):\s*(.*?)(?:\n|$)'