Update app.py

app.py

@@ -2,8 +2,7 @@
 Pharmaceutical Analytics Agentic AI Demo - Streamlit App
 
 This is the main Streamlit application for the pharmaceutical analytics demo.
-It includes initialization steps that would normally be in main.py to support
-running directly on Hugging Face Spaces.
+It initializes the environment and delegates page rendering to the page manager.
 
 Usage (on Hugging Face):
     The app runs automatically
@@ -13,34 +12,17 @@ Usage (locally):
 """
 
 import streamlit as st
-import pandas as pd
-import numpy as np
-import matplotlib.pyplot as plt
-import seaborn as sns
 import os
-import json
-import time
 import sys
+import time
 from datetime import datetime
-import sqlite3
-import plotly.express as px
-import plotly.graph_objects as go
-import networkx as nx
-from pyvis.network import Network
-import tempfile
-import base64
-from PIL import Image
-from io import BytesIO
-import uuid
-import threading
-import queue
-
-# Import diagnostics module
-from updated_diagnostics import render_diagnostics_tab
 
 # Add current directory to path to ensure module imports work
 sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
 
+# Import page manager for modular UI
+from page_manager import render_page
+
 # Initialize environment
 def setup_environment():
     """Setup the demo environment"""
@@ -121,72 +103,6 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Initialize session state for workflow
-if "workflow_state" not in st.session_state:
-    st.session_state.workflow_state = None
-
-if "workflow_thread" not in st.session_state:
-    st.session_state.workflow_thread = None
-
-if "status_queue" not in st.session_state:
-    st.session_state.status_queue = queue.Queue()
-
-if "logs" not in st.session_state:
-    st.session_state.logs = []
-
-if "current_step" not in st.session_state:
-    st.session_state.current_step = None
-
-if "alert_submitted" not in st.session_state:
-    st.session_state.alert_submitted = False
-
-if "show_sql" not in st.session_state:
-    st.session_state.show_sql = False
-
-if "show_code" not in st.session_state:
-    st.session_state.show_code = False
-
-if "environment_ready" not in st.session_state:
-    st.session_state.environment_ready = False
-
-if "agents_loaded" not in st.session_state:
-    st.session_state.agents_loaded = False
-
-if "initialization_attempted" not in st.session_state:
-    st.session_state.initialization_attempted = False
-
-if "active_tab" not in st.session_state:
-    st.session_state.active_tab = "Main"
-
-# Run workflow in separate thread
-def run_workflow_thread(workflow, alert):
-    try:
-        # Update status
-        st.session_state.status_queue.put(("info", "Planning analysis approach..."))
-        st.session_state.current_step = "planning"
-        
-        # Run the workflow with more detailed logging
-        print(f"Starting workflow with alert: {alert}")
-        result = workflow.run_workflow(alert)
-        print("Workflow completed successfully")
-        
-        # Store the result
-        st.session_state.workflow_state = result
-        
-        # Update status
-        st.session_state.status_queue.put(("success", "Analysis complete!"))
-        st.session_state.current_step = "complete"
-    except Exception as e:
-        import traceback
-        error_details = f"Error in workflow: {str(e)}\n{traceback.format_exc()}"
-        print(error_details)
-        st.session_state.status_queue.put(("error", f"Error: {str(e)}"))
-        st.session_state.current_step = "error"
-        # Add to error details for debugging
-        if "error_details" not in st.session_state:
-            st.session_state.error_details = []
-        st.session_state.error_details.append(error_details)
-
 # Main application header
 st.title("🔮 Agentic Pharmaceutical Analytics Platform")
 st.markdown("### Automated Analytics Workflow for Pharmaceutical Brand Leaders")
@@ -210,7 +126,7 @@ with st.sidebar:
         st.success("✓ Claude API key found")
     
     # Setup environment if not already done
-    if not st.session_state.environment_ready:
+    if not st.session_state.get("environment_ready", False):
         st.session_state.environment_ready = setup_environment()
         if not st.session_state.environment_ready:
             st.error("Environment setup failed. Check error messages above.")
@@ -218,7 +134,7 @@ with st.sidebar:
             st.stop()
     
     # Initialize agents if environment is ready
-    if st.session_state.environment_ready and not st.session_state.agents_loaded:
+    if st.session_state.environment_ready and not st.session_state.get("agents_loaded", False):
         st.markdown("### Initializing agents...")
         agents_loaded, agents_or_error = initialize_agents()
         st.session_state.agents_loaded = agents_loaded
@@ -232,13 +148,13 @@ with st.sidebar:
             st.info("Try refreshing the page or check logs.")
             # Don't stop here, allow diagnostics tab to run
     
-    # Debug options (only show if environment and agents are ready)
-    if st.session_state.environment_ready:
+    # Debug options (only show if environment is ready)
+    if st.session_state.get("environment_ready", False):
         st.subheader("Debug Options")
-        show_sql = st.checkbox("Show Generated SQL", value=st.session_state.show_sql)
+        show_sql = st.checkbox("Show Generated SQL", value=st.session_state.get("show_sql", False))
         st.session_state.show_sql = show_sql
         
-        show_code = st.checkbox("Show Generated Python", value=st.session_state.show_code)
+        show_code = st.checkbox("Show Generated Python", value=st.session_state.get("show_code", False))
         st.session_state.show_code = show_code
     
     # Explanation
@@ -257,238 +173,23 @@ with st.sidebar:
     """)
     
     # Restart button
-    if st.session_state.workflow_state is not None:
+    if st.session_state.get("workflow_state") is not None:
         if st.button("Restart Demo"):
             st.session_state.workflow_state = None
             st.session_state.workflow_thread = None
             st.session_state.logs = []
             st.session_state.current_step = None
             st.session_state.alert_submitted = False
+            st.session_state.active_page = "main"
             st.rerun()
 
-# Tab selection
-tab1, tab2 = st.tabs(["Main App", "Diagnostics"])
-
-with tab1:
-    # Only show main app if environment is ready
-    if not st.session_state.environment_ready:
-        st.info("Setting up environment. Please wait...")
-    else:
-        # Main screen
-        if st.session_state.workflow_state is None:
-            # Alert input section
-            st.header("📱 Incoming Alert")
-            
-            col1, col2 = st.columns([3, 1])
-            
-            with col1:
-                # Dropdown for alert selection
-                alert_option = st.selectbox(
-                    "Select Alert Type:",
-                    [
-                        "Sales Decline",
-                        "Market Share Loss",
-                        "Inventory Issue",
-                        "Competitor Launch",
-                        "Custom Alert"
-                    ]
-                )
-                
-                # Alert templates
-                alert_templates = {
-                    "Sales Decline": "Sales of DrugX down 15% in Northeast region over past 30 days compared to forecast.",
-                    "Market Share Loss": "Market share of DrugX decreased by 8 percentage points in the Southeast region over the last quarter.",
-                    "Inventory Issue": "Multiple stockouts of DrugX reported in Midwest distribution centers affecting 25% of pharmacies.",
-                    "Competitor Launch": "Competitor MedCorp launched similar product at 20% lower price point in Western territories.",
-                    "Custom Alert": ""
-                }
-                
-                # Alert description
-                alert_text = st.text_area(
-                    "Alert Description:",
-                    value=alert_templates[alert_option],
-                    height=100
-                )
-                
-               
-               
-                # Submit button
-                submit_button = st.button("Analyze Alert")
-                
-                if submit_button and alert_text:
-                    st.info("Starting analysis workflow...")
-                    
-                    try:
-                        # Create the workflow
-                        SalesAnalysisWorkflow = st.session_state.agents["SalesAnalysisWorkflow"]
-                        workflow = SalesAnalysisWorkflow(db_path="data/pharma_db.sqlite")
-                        
-                        # Initialize workflow state tracking
-                        if "workflow_progress" not in st.session_state:
-                            st.session_state.workflow_progress = {
-                                "started": True,
-                                "step": "planning",
-                                "completed_steps": [],
-                                "error": None
-                            }
-                        
-                        # Show current progress
-                        progress_status = st.empty()
-                        progress_status.info(f"Step: {st.session_state.workflow_progress['step']}")
-                        
-                        # Execute workflow in steps for better visibility
-                        if st.session_state.workflow_progress["step"] == "planning":
-                            try:
-                                # Execute planning step
-                                progress_status.info("Planning analysis approach...")
-                                result = workflow.run_workflow(alert_text)
-                                
-                                # Store the result
-                                st.session_state.workflow_state = result
-                                st.session_state.workflow_progress["completed_steps"].append("planning")
-                                st.session_state.workflow_progress["step"] = "complete"
-                                
-                                # Show success
-                                progress_status.success("Analysis complete!")
-                                st.rerun()
-                            except Exception as e:
-                                import traceback
-                                error_msg = f"Error in workflow: {str(e)}"
-                                st.error(error_msg)
-                                st.code(traceback.format_exc())
-                                st.session_state.workflow_progress["error"] = error_msg
-                        
-                        # If we have a completed workflow, display it
-                        if "workflow_state" in st.session_state and st.session_state.workflow_state:
-                            workflow_state = st.session_state.workflow_state
-                            if workflow_state["status"] == "complete":
-                                st.success("Analysis completed successfully!")
-                    except Exception as e:
-                        import traceback
-                        st.error(f"Error initializing workflow: {str(e)}")
-                        st.code(traceback.format_exc()) 
-            
-            with col2:
-                st.image("https://img.icons8.com/fluency/240/000000/notification-center.png", width=150)
-                st.markdown("**Source:** Mobile Alert System")
-                st.markdown("**Priority:** High")
-                st.markdown("**Time:** " + datetime.now().strftime("%Y-%m-%d %H:%M"))
-            
-            # Show example dashboard
-            st.markdown("---")
-            st.header("📊 Example Dashboard Preview")
-            
-            # Create sample visualization
-            fig = px.line(
-                pd.DataFrame({
-                    'Month': pd.date_range(start='2023-01-01', periods=12, freq='M'),
-                    'Sales': [1200, 1250, 1300, 1400, 1500, 1600, 1550, 1500, 1450, 1300, 1200, 1150],
-                    'Target': [1200, 1250, 1300, 1350, 1400, 1450, 1500, 1550, 1600, 1650, 1700, 1750]
-                }),
-                x='Month',
-                y=['Sales', 'Target'],
-                title="DrugX Performance (Sample Data)",
-                labels={"value": "Sales ($K)", "variable": "Metric"}
-            )
-            fig.update_layout(height=400)
-            st.plotly_chart(fig, use_container_width=True)
-        
-        else:
-            # Process status updates from thread
-            while not st.session_state.status_queue.empty():
-                status_type, status_message = st.session_state.status_queue.get()
-                if status_type == "info":
-                    st.info(status_message)
-                elif status_type == "success":
-                    st.success(status_message)
-                elif status_type == "error":
-                    st.error(status_message)
-                
-                # Add to logs
-                st.session_state.logs.append({
-                    "timestamp": datetime.now().isoformat(),
-                    "type": status_type,
-                    "message": status_message
-                })
-            
-            # Get current workflow state
-            workflow_state = st.session_state.workflow_state
-            
-            if workflow_state is None or workflow_state["status"] == "error":
-                # Show error state
-                st.error(f"Analysis failed: {workflow_state.get('error', 'Unknown error')}")
-                if st.button("Start Over"):
-                    st.session_state.workflow_state = None
-                    st.session_state.workflow_thread = None
-                    st.session_state.logs = []
-                    st.session_state.current_step = None
-                    st.session_state.alert_submitted = False
-                    st.rerun()
-            
-            elif workflow_state["status"] != "complete":
-                # Show progress
-                current_step = st.session_state.current_step
-                
-                # Progress indicators
-                st.markdown("### 🔄 Analysis in Progress")
-                
-                # Progress bar
-                steps = ["planning", "data_collection", "analysis", "validation", "insights", "complete"]
-                step_idx = steps.index(current_step) if current_step in steps else 0
-                progress = (step_idx + 1) / len(steps)
-                
-                progress_bar = st.progress(progress)
-                
-                # Step indicators
-                col1, col2, col3, col4, col5 = st.columns(5)
-                
-                with col1:
-                    check = "✅" if step_idx >= 0 else "🔄"
-                    st.markdown(f"{check} **Planning**")
-                
-                with col2:
-                    check = "✅" if step_idx >= 1 else "⏳"
-                    check = "🔄" if step_idx == 1 else check
-                    st.markdown(f"{check} **Data Collection**")
-                
-                with col3:
-                    check = "✅" if step_idx >= 2 else "⏳"
-                    check = "🔄" if step_idx == 2 else check
-                    st.markdown(f"{check} **Analysis**")
-                
-                with col4:
-                    check = "✅" if step_idx >= 3 else "⏳"
-                    check = "🔄" if step_idx == 3 else check
-                    st.markdown(f"{check} **Validation**")
-                
-                with col5:
-                    check = "✅" if step_idx >= 4 else "⏳"
-                    check = "🔄" if step_idx == 4 else check
-                    st.markdown(f"{check} **Insights**")
-                
-                # Show agent activity visualization based on current step
-                st.markdown("---")
-                
-                # ... (rest of the agent visualization code remains unchanged)
-                
-                # Placeholder for log console
-                with st.expander("View Process Log"):
-                    for log in st.session_state.logs:
-                        st.text(f"{log['timestamp']} - {log['message']}")
-                
-                # Auto refresh
-                time.sleep(0.5)
-                st.rerun()
-            
-            else:
-                # Show completed analysis
-                st.markdown("### ✅ Analysis Complete")
-                
-                # ... (rest of the analysis display code remains unchanged)
+# Only show main app if environment is ready
+if not st.session_state.get("environment_ready", False):
+    st.info("Setting up environment and initializing agents. Please wait...")
+    st.stop()
 
-with tab2:
-    # Render diagnostics tab
-    render_diagnostics_tab()
+# Render the appropriate page based on session state
+render_page()
 
 # Run the app
 if __name__ == "__main__":