Update pages/dashboard.py

pages/dashboard.py

@@ -1,114 +1,166 @@
-"""
-Dashboard page for the MONA application
-Fixed version with proper imports and error handling
-"""
-
 import streamlit as st
 import pandas as pd
+import numpy as np
+from datetime import datetime, timedelta
 import plotly.express as px
-from datetime import datetime
-from typing import Dict, List, Any
-
-from utils.storage import load_data, save_data, get_cached_data, set_cached_data
-from utils.error_handling import handle_data_exceptions, ValidationError, DataError
-from utils.logging import get_logger, log_info, log_error, log_warning
-
-# Initialize logger
-logger = get_logger(__name__)
-
-@handle_data_exceptions
-def load_dashboard_data() -> Dict[str, Any]:
-    cached_data = get_cached_data("dashboard_data")
-    if cached_data:
-        log_info("Using cached dashboard data")
-        return cached_data
-
+import plotly.graph_objects as go
+from utils.storage import load_data, save_data, get_session_data, set_session_data
+from utils.error_handling import handle_ui_exceptions
+from utils.logging import log_info, log_error
+
+@handle_ui_exceptions
+def create_dashboard_page():
+    """Create the main dashboard page with enhanced features"""
+    
+    st.title("🏠 Dashboard")
+    st.markdown("---")
+    
+    # Key metrics section
+    create_metrics_section()
+    
+    # Charts section
+    create_charts_section()
+    
+    # Recent activity
+    create_activity_section()
+    
+    # Quick actions
+    create_quick_actions()
+
+@handle_ui_exceptions
+def create_metrics_section():
+    """Create the key metrics display"""
+    st.subheader("📊 Key Metrics")
+    
+    # Sample metrics - in a real app, these would come from your data
+    col1, col2, col3, col4 = st.columns(4)
+    
+    with col1:
+        st.metric(
+            label="Total Users",
+            value="2,847",
+            delta="12%",
+            help="Total registered users"
+        )
+    
+    with col2:
+        st.metric(
+            label="Active Sessions",
+            value="456",
+            delta="8%",
+            help="Currently active user sessions"
+        )
+    
+    with col3:
+        st.metric(
+            label="Messages Today",
+            value="1,234",
+            delta="-3%",
+            help="Messages sent today"
+        )
+    
+    with col4:
+        st.metric(
+            label="System Uptime",
+            value="99.9%",
+            delta="0.1%",
+            help="System availability"
+        )
+
+@handle_ui_exceptions
+def create_charts_section():
+    """Create charts and visualizations"""
+    st.subheader("📈 Analytics")
+    
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        # Line chart for usage over time
+        st.write("**Usage Trend (Last 30 Days)**")
+        
+        # Generate sample data
+        dates = pd.date_range(start=datetime.now() - timedelta(days=30), 
+                             end=datetime.now(), 
+                             freq='D')
+        
+        usage_data = pd.DataFrame({
+            'Date': dates,
+            'Users': np.random.randint(50, 200, len(dates)),
+            'Messages': np.random.randint(100, 500, len(dates))
+        })
+        
+        fig = px.line(usage_data, x='Date', y=['Users', 'Messages'],
+                     title="Daily Usage Statistics")
+        st.plotly_chart(fig, use_container_width=True)
+    
+    with col2:
+        # Pie chart for user distribution
+        st.write("**User Distribution**")
+        
+        user_types = ['Free Users', 'Premium Users', 'Enterprise']
+        user_counts = [1200, 800, 400]
+        
+        fig = px.pie(values=user_counts, names=user_types,
+                    title="User Type Distribution")
+        st.plotly_chart(fig, use_container_width=True)
+
+@handle_ui_exceptions
+def create_activity_section():
+    """Create recent activity feed"""
+    st.subheader("🔔 Recent Activity")
+    
+    # Sample activity data
+    activities = [
+        {"time": "2 minutes ago", "action": "New user registered", "user": "[email protected]"},
+        {"time": "5 minutes ago", "action": "Message sent", "user": "[email protected]"},
+        {"time": "10 minutes ago", "action": "File uploaded", "user": "[email protected]"},
+        {"time": "15 minutes ago", "action": "Settings updated", "user": "[email protected]"},
+        {"time": "20 minutes ago", "action": "Login", "user": "[email protected]"}
+    ]
+    
+    for activity in activities:
+        with st.container():
+            col1, col2, col3 = st.columns([2, 4, 2])
+            with col1:
+                st.write(activity["time"])
+            with col2:
+                st.write(f"**{activity['action']}** - {activity['user']}")
+            with col3:
+                st.write("✓")
+        st.markdown("---")
+
+@handle_ui_exceptions
+def create_quick_actions():
+    """Create quick action buttons"""
+    st.subheader("⚡ Quick Actions")
+    
+    col1, col2, col3, col4 = st.columns(4)
+    
+    with col1:
+        if st.button("💬 Start Chat", use_container_width=True):
+            st.switch_page("pages/chat.py")
+    
+    with col2:
+        if st.button("📊 View Analytics", use_container_width=True):
+            st.switch_page("pages/analytics.py")
+    
+    with col3:
+        if st.button("⚙️ Settings", use_container_width=True):
+            st.switch_page("pages/settings.py")
+    
+    with col4:
+        if st.button("📁 File Manager", use_container_width=True):
+            st.switch_page("pages/files.py")
+
+# Export the main function
+def main():
+    """Main function for the dashboard page"""
     try:
-        data = {
-            "metrics": load_data("dashboard_metrics.json", default={}),
-            "user_activity": load_data("user_activity.json", default=[]),
-            "system_stats": load_data("system_stats.json", default={}),
-            "recent_events": load_data("recent_events.json", default=[]),
-            "performance_data": load_data("performance_data.json", default=[])
-        }
-        set_cached_data("dashboard_data", data)
-        log_info("Dashboard data loaded and cached successfully")
-        return data
+        create_dashboard_page()
+        log_info("Dashboard page loaded successfully")
     except Exception as e:
-        log_error("Failed to load dashboard data", error=e)
-        return {
-            "metrics": {},
-            "user_activity": [],
-            "system_stats": {},
-            "recent_events": [],
-            "performance_data": []
-        }
-
-@handle_data_exceptions
-def create_metrics_cards(metrics: Dict[str, Any]) -> None:
-    if not metrics:
-        st.warning("No metrics data available")
-        return
-
-    cols = st.columns(4)
-    with cols[0]:
-        st.metric("Total Users", f"{metrics.get('total_users', 0):,}", delta=metrics.get("users_change", 0))
-    with cols[1]:
-        st.metric("Active Sessions", f"{metrics.get('active_sessions', 0):,}", delta=metrics.get("sessions_change", 0))
-    with cols[2]:
-        st.metric("System Health", f"{metrics.get('system_health', 0):.1f}%", delta=f"{metrics.get('health_change', 0):.1f}%")
-    with cols[3]:
-        st.metric("Avg Response Time", f"{metrics.get('avg_response_time', 0):.0f}ms", delta=f"{metrics.get('response_change', 0):.0f}ms", delta_color="inverse")
-
-@handle_data_exceptions
-def create_activity_chart(activity_data: List[Dict]) -> None:
-    if not activity_data:
-        st.warning("No activity data available")
-        return
-
-    df = pd.DataFrame(activity_data)
-    if df.empty:
-        st.warning("Activity data is empty")
-        return
-
-    if 'timestamp' not in df.columns:
-        df['timestamp'] = pd.date_range(start='2024-01-01', periods=len(df), freq='H')
-    if 'users' not in df.columns:
-        df['users'] = [0] * len(df)
-
-    df['timestamp'] = pd.to_datetime(df['timestamp'])
-
-    fig = px.line(
-        df,
-        x='timestamp',
-        y='users',
-        title='User Activity Over Time',
-        labels={'users': 'Active Users', 'timestamp': 'Time'}
-    )
-
-    fig.update_layout(
-        xaxis_title="Time",
-        yaxis_title="Active Users",
-    )
-
-    st.plotly_chart(fig, use_container_width=True)
-
-@handle_data_exceptions
-def render_dashboard():
-    """Main function to render the dashboard page."""
-    st.title("📊 MONA Dashboard")
-
-    # Load all data
-    data = load_dashboard_data()
-
-    # Metrics cards
-    create_metrics_cards(data.get("metrics", {}))
-    st.markdown("---")
-
-    # Activity chart
-    st.subheader("User Activity")
-    create_activity_chart(data.get("user_activity", []))
+        log_error("Error loading dashboard page", error=e)
+        st.error("Failed to load dashboard. Please try refreshing the page.")
 
-# Run page
-if __name__ == "__main__" or __name__.endswith(".dashboard"):
-    render_dashboard()
+if __name__ == "__main__":
+    main()