Update utils.py

utils.py

@@ -1,7 +1,8 @@
 from langchain_openai import ChatOpenAI
 import streamlit as st
-import json
-import re
+import pandas as pd
+import plotly.express as px
+from datetime import datetime
 
 def update_progress(container, percentage, message=""):
     if container:
@@ -153,6 +154,98 @@ def display_report(report_data):
         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")
+    
+    # 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)
+    
+    # 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)
+    
+    # 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")
+    
+    # Key Players Table
+    if 'keyPlayers' in market_data:
+        st.write("### Key Players Analysis")
+        players_df = pd.DataFrame(market_data['keyPlayers'])
+        st.dataframe(
+            players_df,
+            column_config={
+                "company": "Company",
+                "marketShare": st.column_config.NumberColumn(
+                    "Market Share (%)",
+                    format="%.1f%%"
+                ),
+                "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"
+            },
+            hide_index=True
+        )
+
 
 def apply_report_styling():
     return """
@@ -271,84 +364,9 @@ def display_report(report_data):
         
         # Display metrics dashboard
         if 'metrics' in report_data and report_data['metrics']:
-            st.markdown('<div class="metrics-container">', unsafe_allow_html=True)
             st.subheader("📊 Market Metrics Dashboard")
-            cols = st.columns(3)
             metrics = report_data['metrics']
-            
-            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'))
-            st.markdown('</div>', unsafe_allow_html=True)
-        
-        # Display key players if available
-        if isinstance(key_players, list) and key_players:
-            st.markdown('<div class="key-players">', unsafe_allow_html=True)
-            st.subheader("🏢 Key Market Players")
-            cols = st.columns(2)
-            for i, player in enumerate(key_players):
-                with cols[i % 2]:
-                    st.markdown(f'<div class="player-card">{player}</div>', unsafe_allow_html=True)
-            st.markdown('</div>', unsafe_allow_html=True)
-        
-        # Display the main report content with enhanced styling
-        if 'content' in report_data and report_data['content']:
-            sections = report_data['content'].split('#')
-            for section in sections:
-                if section.strip():
-                    # Split into title and content
-                    parts = section.strip().split('\n', 1)
-                    if len(parts) > 1:
-                        title, content = parts
-                        st.markdown(f'<div class="section-box">', unsafe_allow_html=True)
-                        st.markdown(f'## {title.strip()}')
-                        st.markdown(content)
-                        st.markdown('</div>', unsafe_allow_html=True)
-        
-        # Display agent outputs in styled expandable sections
-        if 'agent_outputs' in report_data:
-            st.markdown('<div class="section-box">', unsafe_allow_html=True)
-            st.markdown("## 🤖 Detailed Agent Analysis")
-            
-            with st.expander("Researcher's Findings"):
-                st.markdown('<div class="bullet-points">', unsafe_allow_html=True)
-                st.markdown(report_data['agent_outputs'].get('researcher', 'No researcher output available'))
-                st.markdown('</div>', unsafe_allow_html=True)
-                
-            with st.expander("Analyst's Insights"):
-                st.markdown('<div class="bullet-points">', unsafe_allow_html=True)
-                st.markdown(report_data['agent_outputs'].get('analyst', 'No analyst output available'))
-                st.markdown('</div>', unsafe_allow_html=True)
-                
-            with st.expander("Writer's Draft"):
-                st.markdown('<div class="bullet-points">', unsafe_allow_html=True)
-                st.markdown(report_data['agent_outputs'].get('writer', 'No writer output available'))
-                st.markdown('</div>', unsafe_allow_html=True)
-            
-            st.markdown('</div>', unsafe_allow_html=True)
-            
-    except Exception as e:
-        st.error(f"Error displaying report: {str(e)}")
-
-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.markdown('<div class="metrics-container">', unsafe_allow_html=True)
-            st.subheader("📊 Market Metrics Dashboard")
             cols = st.columns(3)
-            metrics = report_data['metrics']
             
             with cols[0]:
                 st.metric("Market Size", metrics.get('market_size', 'N/A'))
@@ -361,56 +379,49 @@ def display_report(report_data):
             with cols[2]:
                 st.metric("Key Region", metrics.get('key_regions', 'N/A'))
                 st.metric("Dominant Segment", metrics.get('dominant_segment', 'N/A'))
-            st.markdown('</div>', unsafe_allow_html=True)
         
-        # Display key players if available
-        if isinstance(key_players, list) and key_players:
-            st.markdown('<div class="key-players">', unsafe_allow_html=True)
-            st.subheader("🏢 Key Market Players")
-            cols = st.columns(2)
-            for i, player in enumerate(key_players):
-                with cols[i % 2]:
-                    st.markdown(f'<div class="player-card">{player}</div>', unsafe_allow_html=True)
-            st.markdown('</div>', unsafe_allow_html=True)
+        # Display visualizations
+        if 'market_data' in report_data:
+            display_market_visualizations(report_data['market_data'])
         
-        # Display the main report content with enhanced styling
-        if 'content' in report_data and report_data['content']:
-            sections = report_data['content'].split('#')
-            for section in sections:
-                if section.strip():
-                    # Split into title and content
-                    parts = section.strip().split('\n', 1)
-                    if len(parts) > 1:
-                        title, content = parts
-                        st.markdown(f'<div class="section-box">', unsafe_allow_html=True)
-                        st.markdown(f'## {title.strip()}')
-                        st.markdown(content)
-                        st.markdown('</div>', unsafe_allow_html=True)
+        # 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 in styled expandable sections
+        # Display agent outputs
         if 'agent_outputs' in report_data:
-            st.markdown('<div class="section-box">', unsafe_allow_html=True)
-            st.markdown("## 🤖 Detailed Agent Analysis")
+            st.markdown("## 🤖 Agent Analysis - Raw Feed")
+            tabs = st.tabs(["Researcher", "Analyst", "Writer"])
             
-            with st.expander("Researcher's Findings"):
-                st.markdown('<div class="bullet-points">', unsafe_allow_html=True)
-                st.markdown(report_data['agent_outputs'].get('researcher', 'No researcher output available'))
-                st.markdown('</div>', unsafe_allow_html=True)
-                
-            with st.expander("Analyst's Insights"):
-                st.markdown('<div class="bullet-points">', unsafe_allow_html=True)
-                st.markdown(report_data['agent_outputs'].get('analyst', 'No analyst output available'))
-                st.markdown('</div>', unsafe_allow_html=True)
-                
-            with st.expander("Writer's Draft"):
-                st.markdown('<div class="bullet-points">', unsafe_allow_html=True)
-                st.markdown(report_data['agent_outputs'].get('writer', 'No writer output available'))
-                st.markdown('</div>', unsafe_allow_html=True)
+            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'))
             
-            st.markdown('</div>', unsafe_allow_html=True)
+            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|$)'