Update app.py

app.py

@@ -6,14 +6,66 @@ from datetime import datetime, timedelta, date
 import time
 import io
 import base64
+import plotly.express as px
+import plotly.graph_objects as go
+from datetime import datetime, timedelta, date
+import time
+import json
+import google.generativeai as genai
+from google.generativeai.types import HarmCategory, HarmBlockThreshold
+
+
+# Setup AI Services
+def setup_genai():
+    """Initialize and configure Google's Generative AI and list available models"""
+    try:
+        # Try getting API key from Streamlit secrets first
+        if 'GOOGLE_API_KEY' in st.secrets:
+            api_key = st.secrets['GOOGLE_API_KEY']
+        # Fall back to environment variable
+        elif 'GOOGLE_API_KEY' in os.environ:
+            api_key = os.environ['GOOGLE_API_KEY']
+        else:
+            st.warning("Google API key not found. Using simulated AI responses.")
+            st.session_state.gemini_model = "gemini-1.5-pro"
+            return False
+            
+        genai.configure(api_key=api_key)
+
+# Import pages
+from pages.dashboard_page import render_financial_dashboard, get_runway_analysis, get_fundraising_readiness_analysis
+from pages.decision_simulator import render_decision_simulator, get_decision_analysis
+from pages.fund_monitoring import render_fund_monitoring, get_fraud_analysis
+from pages.financial_advisor import render_ai_financial_advisor, get_advisory_guidance, generate_voice_response
+from pages.book_session import render_book_session
+
+# Initialize page configuration
+st.set_page_config(
+    page_title="StartupFinancePilot",
+    page_icon="💰",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
 
-# Import your existing pages (make sure they're in the same directory or in a 'pages' directory)
-# If you have these in separate files, you'll need to adjust the imports
-# For example:
-# from pages.dashboard import render_financial_dashboard
-# from pages.decision_simulator import render_decision_simulator
-# from pages.fund_monitoring import render_fund_monitoring
-# from pages.advisor import render_ai_financial_advisor
+
+# Initialize session state variables
+if 'booked_sessions' not in st.session_state:
+    st.session_state.booked_sessions = []
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []
+if 'audio_response' not in st.session_state:
+    st.session_state.audio_response = None
+if 'insights_cache' not in st.session_state:
+    st.session_state.insights_cache = {}
+if 'gemini_model' not in st.session_state:
+    st.session_state.gemini_model = None
+if 'current_page' not in st.session_state:
+    st.session_state.current_page = "Financial Dashboard"
+    
+from pages.dashboard import render_financial_dashboard
+from pages.decision_simulator import render_decision_simulator
+from pages.fund_monitoring import render_fund_monitoring
+from pages.advisor import render_ai_financial_advisor
 
 # Constants
 DEFAULT_GROWTH_RATE = 0.08  # 8% monthly growth
@@ -464,45 +516,50 @@ def create_sidebar():
         if st.button("🤖 AI Financial Advisor", use_container_width=True):
             switch_page('advisor')
 
-# Main function
+
 def main():
-    # Create sidebar navigation
-    create_sidebar()
+    # Load sample data
+    startup_data, cash_flow_df, transactions_df = load_sample_data()
     
-    # Check if we have any data to work with
-    if not st.session_state.startups and st.session_state.current_page != 'upload':
-        st.warning("No startup data found. Please upload your data or use a sample dataset.")
-        render_upload_page()
-        return
+    # Create sidebar
+    st.sidebar.title("StartupFinancePilot")
+    st.sidebar.image("https://img.freepik.com/premium-vector/business-finance-analytics-logo-design-vector-template_67715-552.jpg", width=150)
     
-    # Render the correct page based on session state
-    if st.session_state.current_page == 'upload':
-        render_upload_page()
-    elif st.session_state.current_page == 'dashboard':
-        # Replace with your dashboard page function
-        from dashboard import render_financial_dashboard
-        render_financial_dashboard(
-            st.session_state.startups[st.session_state.current_startup]['profile'],
-            st.session_state.startups[st.session_state.current_startup]['cash_flow']
-        )
-    elif st.session_state.current_page == 'simulator':
-        # Replace with your simulator page function
-        from decision_simulator import render_decision_simulator
-        render_decision_simulator(
-            st.session_state.startups[st.session_state.current_startup]['profile']
-        )
-    elif st.session_state.current_page == 'monitoring':
-        # Replace with your monitoring page function
-        from fund_monitoring import render_fund_monitoring
-        render_fund_monitoring(
-            st.session_state.startups[st.session_state.current_startup]['transactions']
-        )
-    elif st.session_state.current_page == 'advisor':
-        # Replace with your advisor page function
-        from advisor import render_ai_financial_advisor
-        render_ai_financial_advisor(
-            st.session_state.startups[st.session_state.current_startup]['profile']
-        )
+    # Company profile
+    st.sidebar.header("Company Profile")
+    st.sidebar.write(f"**{startup_data['name']}**")
+    st.sidebar.write(f"Stage: {startup_data['stage']}")
+    st.sidebar.write(f"Founded: {startup_data['founded']}")
+    st.sidebar.write(f"Employees: {startup_data['employees']}")
+    st.sidebar.write(f"Last Funding: {startup_data['last_funding']}")
+    
+    
+    # Navigation
+    st.sidebar.header("Navigation")
+    pages = {
+        "Financial Dashboard": render_financial_dashboard,
+        "Decision Simulator": render_decision_simulator,
+        "Fund Monitoring": render_fund_monitoring,
+        "AI Financial Advisor": render_ai_financial_advisor,
+        "Book a Session": render_book_session
+    }
+    
+    # Page selection
+    selected_page = st.sidebar.radio("Go to", list(pages.keys()))
+    st.session_state.current_page = selected_page
+    
+    # Render selected page
+    if selected_page == "Financial Dashboard":
+        pages[selected_page](startup_data, cash_flow_df)
+    elif selected_page == "Decision Simulator":
+        pages[selected_page](startup_data)
+    elif selected_page == "Fund Monitoring":
+        pages[selected_page](transactions_df)
+    elif selected_page == "AI Financial Advisor":
+        pages[selected_page](startup_data)
+    else:  # Book a Session
+        pages[selected_page]()
 
 if __name__ == "__main__":
     main()
+