Update app.py

app.py

@@ -1,282 +1,280 @@
-"""
-Main application file for MONA
-Fixed version with proper imports and error handling
-"""
-
 import streamlit as st
+import logging
 import sys
-from pathlib import Path
-
-# Add the current directory to Python path for imports
-current_dir = Path(__file__).parent
-sys.path.append(str(current_dir))
-
-try:
-    from pages.dashboard import create_dashboard_page
-    from utils.logging import setup_logging, get_logger, log_info, log_error
-    from utils.error_handling import ErrorHandler
-    from utils.storage import initialize_sample_data
-except ImportError as e:
-    st.error(f"Import Error: {str(e)}")
-    st.error("Please ensure all required files are present and properly configured.")
-    st.stop()
+import traceback
+from datetime import datetime
+import os
 
-
-def setup_streamlit_config():
-    """Configure Streamlit page settings"""
-    st.set_page_config(
-        page_title="MONA - Dashboard",
-        page_icon="🎯",
-        layout="wide",
-        initial_sidebar_state="expanded"
+# Configure logging at the module level
+def setup_logging():
+    """Setup logging configuration"""
+    logging.basicConfig(
+        level=logging.INFO,
+        format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        handlers=[
+            logging.StreamHandler(sys.stdout),
+            logging.FileHandler('app.log', mode='a')
+        ]
     )
+    return logging.getLogger(__name__)
 
+def log_error(message, error=None):
+    """Log error with traceback if available"""
+    logger = logging.getLogger(__name__)
+    if error:
+        logger.error(f"{message}: {str(error)}")
+        logger.error(f"Traceback: {traceback.format_exc()}")
+    else:
+        logger.error(message)
 
-def create_sidebar():
-    """Create application sidebar"""
-    with st.sidebar:
-        st.title("🎯 MONA")
-        st.markdown("---")
-        
-        # Navigation
-        st.subheader("Navigation")
-        page = st.selectbox(
-            "Select Page",
-            ["Dashboard", "Analytics", "Settings", "Help"],
-            index=0
-        )
-        
-        st.markdown("---")
-        
-        # Quick Stats
-        st.subheader("Quick Stats")
-        try:
-            from utils.storage import load_data
-            metrics = load_data("dashboard_metrics.json", default={})
-            
-            if metrics:
-                st.metric("Total Users", f"{metrics.get('total_users', 0):,}")
-                st.metric("Active Sessions", f"{metrics.get('active_sessions', 0):,}")
-                st.metric("System Health", f"{metrics.get('system_health', 0):.1f}%")
-        except Exception as e:
-            st.warning("Could not load quick stats")
-        
-        st.markdown("---")
-        
-        # System Status
-        st.subheader("System Status")
-        st.success("🟢 All Systems Operational")
-        
-        return page
+def log_info(message):
+    """Log info message"""
+    logger = logging.getLogger(__name__)
+    logger.info(message)
 
+# Initialize logging
+logger = setup_logging()
 
-def create_header():
-    """Create application header"""
-    col1, col2, col3 = st.columns([2, 6, 2])
-    
-    with col1:
-        st.image("https://via.placeholder.com/100x50/4CAF50/FFFFFF?text=MONA", width=100)
-    
-    with col2:
-        st.markdown("""
-        <h1 style='text-align: center; color: #1f77b4;'>
-            MONA Monitoring & Analytics Platform
-        </h1>
-        """, unsafe_allow_html=True)
-    
-    with col3:
-        if st.button("🔄 Refresh All"):
-            from utils.storage import clear_cache
-            clear_cache()
-            st.rerun()
+# Page configuration
+st.set_page_config(
+    page_title="Mona - AI Assistant",
+    page_icon="🤖",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
 
+# Custom CSS
+st.markdown("""
+<style>
+    .main-header {
+        font-size: 3rem;
+        font-weight: bold;
+        text-align: center;
+        background: linear-gradient(90deg, #667eea 0%, #764ba2 100%);
+        -webkit-background-clip: text;
+        -webkit-text-fill-color: transparent;
+        margin-bottom: 2rem;
+    }
+    .feature-card {
+        background: white;
+        padding: 1.5rem;
+        border-radius: 10px;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        border-left: 4px solid #667eea;
+        margin: 1rem 0;
+    }
+    .sidebar-content {
+        background: #f8f9fa;
+        padding: 1rem;
+        border-radius: 10px;
+        margin: 1rem 0;
+    }
+</style>
+""", unsafe_allow_html=True)
 
-def create_footer():
-    """Create application footer"""
-    st.markdown("---")
-    col1, col2, col3 = st.columns([1, 2, 1])
-    
-    with col2:
+def create_dashboard_page():
+    """Create the main dashboard page"""
+    try:
+        st.markdown('<h1 class="main-header">🤖 Mona AI Assistant</h1>', unsafe_allow_html=True)
+        
+        # Welcome message
         st.markdown("""
-        <div style='text-align: center; color: #666;'>
-            <p>MONA Platform v2.1.0 | Built with Streamlit</p>
-            <p>© 2024 MONA Systems. All rights reserved.</p>
+        <div class="feature-card">
+            <h3>Welcome to Mona!</h3>
+            <p>Your intelligent AI assistant ready to help with various tasks.</p>
         </div>
         """, unsafe_allow_html=True)
-
-
-def create_analytics_page():
-    """Create analytics page (placeholder)"""
-    st.title("📈 Analytics")
-    st.info("Analytics page coming soon...")
-    
-    # Placeholder content
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        st.subheader("Data Overview")
-        st.write("• Total Records: 10,000+")
-        st.write("• Data Sources: 5")
-        st.write("• Last Update: Just now")
-    
-    with col2:
+        
+        # Feature cards
+        col1, col2, col3 = st.columns(3)
+        
+        with col1:
+            st.markdown("""
+            <div class="feature-card">
+                <h4>💬 Chat</h4>
+                <p>Have conversations with AI</p>
+            </div>
+            """, unsafe_allow_html=True)
+        
+        with col2:
+            st.markdown("""
+            <div class="feature-card">
+                <h4>📊 Analytics</h4>
+                <p>Data analysis and insights</p>
+            </div>
+            """, unsafe_allow_html=True)
+        
+        with col3:
+            st.markdown("""
+            <div class="feature-card">
+                <h4>🔧 Tools</h4>
+                <p>Various utility functions</p>
+            </div>
+            """, unsafe_allow_html=True)
+        
+        # Main content area
         st.subheader("Quick Actions")
-        if st.button("Generate Report"):
-            st.success("Report generation initiated")
-        if st.button("Export Data"):
-            st.success("Data export started")
-
-
-def create_settings_page():
-    """Create settings page (placeholder)"""
-    st.title("⚙️ Settings")
-    
-    tab1, tab2, tab3 = st.tabs(["General", "Security", "Advanced"])
-    
-    with tab1:
-        st.subheader("General Settings")
         
         col1, col2 = st.columns(2)
+        
         with col1:
-            st.text_input("Application Name", value="MONA")
-            st.selectbox("Theme", ["Light", "Dark", "Auto"])
-            st.selectbox("Language", ["English", "Spanish", "French"])
+            if st.button("Start Chat", type="primary"):
+                st.switch_page("pages/chat.py")
         
         with col2:
-            st.number_input("Refresh Interval (seconds)", min_value=1, max_value=300, value=30)
-            st.checkbox("Enable Notifications", value=True)
-            st.checkbox("Auto-save Settings", value=True)
-    
-    with tab2:
-        st.subheader("Security Settings")
-        st.text_input("Admin Password", type="password")
-        st.checkbox("Enable Two-Factor Authentication")
-        st.checkbox("Enable Audit Logging", value=True)
-        st.selectbox("Session Timeout", ["15 minutes", "30 minutes", "1 hour", "4 hours"])
-    
-    with tab3:
-        st.subheader("Advanced Settings")
-        st.code("""
-        {
-            "debug_mode": false,
-            "cache_size": 100,
-            "max_connections": 20,
-            "log_level": "INFO"
-        }
-        """)
-        
-        if st.button("Reset to Defaults"):
-            st.warning("Settings reset to default values")
-
+            if st.button("View Analytics"):
+                st.switch_page("pages/analytics.py")
+        
+        # Recent activity
+        st.subheader("Recent Activity")
+        st.info("No recent activity to display.")
+        
+        log_info("Dashboard page loaded successfully")
+        
+    except Exception as e:
+        log_error("Error in dashboard page", error=e)
+        st.error("An error occurred while loading the dashboard. Please try again.")
 
-def create_help_page():
-    """Create help page"""
-    st.title("❓ Help & Documentation")
-    
-    col1, col2 = st.columns([1, 2])
-    
-    with col1:
-        st.subheader("Quick Links")
-        st.markdown("""
-        - [Getting Started](#getting-started)
-        - [Dashboard Guide](#dashboard)
-        - [Analytics](#analytics)
-        - [Troubleshooting](#troubleshooting)
-        - [API Documentation](#api)
-        """)
-    
-    with col2:
-        st.subheader("Getting Started")
-        st.markdown("""
-        Welcome to MONA! This platform provides comprehensive monitoring 
-        and analytics capabilities for your systems.
-        
-        **Key Features:**
-        - Real-time dashboard with key metrics
-        - System performance monitoring
-        - User activity tracking
-        - Event logging and analysis
-        - Customizable alerts and notifications
-        
-        **Navigation:**
-        Use the sidebar to navigate between different sections of the application.
-        The dashboard provides an overview of system health and key metrics.
-        """)
-        
-        st.subheader("Dashboard Guide")
-        st.markdown("""
-        The dashboard displays:
-        - **Key Metrics:** Total users, active sessions, system health
-        - **Activity Charts:** User activity over time
-        - **Performance Graphs:** CPU, memory, and disk usage
-        - **Recent Events:** Latest system events and alerts
-        - **System Status:** Current status of all services
-        """)
-        
-        st.subheader("Troubleshooting")
-        with st.expander("Common Issues"):
-            st.markdown("""
-            **Data not loading:**
-            - Check your internet connection
-            - Try refreshing the page
-            - Contact support if issue persists
+def create_chat_page():
+    """Create the chat page"""
+    try:
+        st.title("💬 Chat with Mona")
+        
+        # Initialize session state
+        if "messages" not in st.session_state:
+            st.session_state.messages = []
+        
+        # Display chat messages
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.markdown(message["content"])
+        
+        # Chat input
+        if prompt := st.chat_input("What would you like to know?"):
+            # Add user message
+            st.session_state.messages.append({"role": "user", "content": prompt})
+            with st.chat_message("user"):
+                st.markdown(prompt)
             
-            **Charts not displaying:**
-            - Ensure JavaScript is enabled
-            - Try a different browser
-            - Clear browser cache
-            """)
-
+            # Add assistant response
+            with st.chat_message("assistant"):
+                response = f"I received your message: '{prompt}'. This is a demo response. In a real implementation, this would connect to an AI service."
+                st.markdown(response)
+                st.session_state.messages.append({"role": "assistant", "content": response})
+        
+        log_info("Chat page loaded successfully")
+        
+    except Exception as e:
+        log_error("Error in chat page", error=e)
+        st.error("An error occurred in the chat. Please try again.")
 
-def main():
-    """Main application function"""
+def create_analytics_page():
+    """Create the analytics page"""
     try:
-        # Setup logging
-        setup_logging()
-        logger = get_logger(__name__)
-        log_info("Starting MONA application")
+        st.title("📊 Analytics Dashboard")
+        
+        # Sample analytics content
+        col1, col2, col3 = st.columns(3)
         
-        # Configure Streamlit
-        setup_streamlit_config()
+        with col1:
+            st.metric("Total Users", "1,234", "+5%")
+        
+        with col2:
+            st.metric("Active Sessions", "456", "+12%")
         
-        with ErrorHandler("main_application", raise_on_error=False):
-            # Create header
-            create_header()
+        with col3:
+            st.metric("Messages Sent", "7,890", "+8%")
+        
+        # Sample chart
+        import pandas as pd
+        import numpy as np
+        
+        chart_data = pd.DataFrame(
+            np.random.randn(20, 3),
+            columns=['Series A', 'Series B', 'Series C']
+        )
+        
+        st.line_chart(chart_data)
+        
+        log_info("Analytics page loaded successfully")
+        
+    except Exception as e:
+        log_error("Error in analytics page", error=e)
+        st.error("An error occurred while loading analytics. Please try again.")
+
+def create_sidebar():
+    """Create the sidebar navigation"""
+    try:
+        with st.sidebar:
+            st.markdown('<div class="sidebar-content">', unsafe_allow_html=True)
+            st.title("Navigation")
+            
+            # Navigation buttons
+            if st.button("🏠 Dashboard", use_container_width=True):
+                st.switch_page("app.py")
             
-            # Create sidebar and get selected page
-            selected_page = create_sidebar()
+            if st.button("💬 Chat", use_container_width=True):
+                st.switch_page("pages/chat.py")
+            
+            if st.button("📊 Analytics", use_container_width=True):
+                st.switch_page("pages/analytics.py")
             
-            # Main content area
             st.markdown("---")
             
-            # Route to appropriate page
-            if selected_page == "Dashboard":
-                create_dashboard_page()
-            elif selected_page == "Analytics":
-                create_analytics_page()
-            elif selected_page == "Settings":
-                create_settings_page()
-            elif selected_page == "Help":
-                create_help_page()
-            else:
-                st.error(f"Unknown page: {selected_page}")
+            # Settings section
+            st.subheader("Settings")
+            theme = st.selectbox("Theme", ["Light", "Dark"])
+            language = st.selectbox("Language", ["English", "Spanish", "French"])
+            
+            st.markdown("---")
             
-            # Create footer
-            create_footer()
+            # Info section
+            st.subheader("Info")
+            st.markdown(f"**Version:** 1.0.0")
+            st.markdown(f"**Last Updated:** {datetime.now().strftime('%Y-%m-%d')}")
+            st.markdown('</div>', unsafe_allow_html=True)
         
-        log_info("MONA application loaded successfully")
+        log_info("Sidebar created successfully")
         
     except Exception as e:
-        log_error("Critical error in main application", error=e)
-        st.error("⚠️ Application Error")
-        st.error("A critical error occurred. Please check the logs and try again.")
-        
-        # Show error details for debugging
-        with st.expander("Error Details"):
-            st.code(str(e))
-            import traceback
-            st.code(traceback.format_exc())
+        log_error("Error creating sidebar", error=e)
 
+def main():
+    """Main application function"""
+    try:
+        # Setup logging first
+        setup_logging()
+        log_info("Application starting up")
+        
+        # Create sidebar
+        create_sidebar()
+        
+        # Get current page
+        current_page = st.session_state.get('page', 'dashboard')
+        
+        # Route to appropriate page
+        if current_page == 'chat':
+            create_chat_page()
+        elif current_page == 'analytics':
+            create_analytics_page()
+        else:
+            create_dashboard_page()
+        
+        log_info("Application loaded successfully")
+        
+    except Exception as e:
+        log_error("Critical error in main application", error=e)
+        st.error("A critical error occurred. Please refresh the page or contact support.")
+        st.exception(e)
 
 if __name__ == "__main__":
-    main()
+    try:
+        main()
+    except Exception as e:
+        # Final fallback error handling
+        print(f"Critical application error: {str(e)}")
+        print(f"Traceback: {traceback.format_exc()}")
+        
+        # Still try to show something in Streamlit
+        st.error("Application failed to start. Please check the logs.")
+        st.exception(e)