Update pages/dashboard-page.py

pages/dashboard-page.py

@@ -1,48 +1,11 @@
+import streamlit as st
+import plotly.express as px
+import pandas as pd
+import google.generativeai as genai
+
 def render_financial_dashboard(startup_data, cash_flow_df):
-    """
-    Render the AI-powered financial dashboard page.
-    
-    This dashboard uses AI to analyze financial data and provide actionable insights
-    to startup founders, helping them make better decisions about their runway,
-    spending, and financial health.
-    """
-    st.markdown("<h1 class='main-header'>Financial Dashboard</h1>", unsafe_allow_html=True)
-    st.markdown("<p class='sub-header'>AI-powered financial insights at a glance</p>", unsafe_allow_html=True)
-    
-    # How AI helps with financial dashboards
-    with st.expander("ℹ️ How AI enhances your financial dashboard"):
-        st.markdown("""
-        ### How AI Powers Your Financial Dashboard
-        
-        The financial dashboard uses AI to transform raw financial data into actionable intelligence:
-        
-        - **Automated Analysis**: Instead of manually calculating runway and burn rates, our AI model analyzes your data and highlights critical trends
-        - **Predictive Forecasting**: AI forecasts your runway using pattern recognition and predictive analytics to account for varying growth rates
-        - **Anomaly Detection**: The system identifies unusual spending patterns or concerning financial trends that human analysis might miss
-        - **Strategic Recommendations**: Based on your specific financial situation, the AI provides tailored recommendations to optimize your runway
-        - **Benchmark Comparison**: Your metrics are automatically compared against industry standards for startups at your funding stage
-        
-        This helps founders save time, catch financial issues early, and make data-driven decisions without needing financial expertise.
-        """)
-    
-    # AI Insights Summary
-    insights_key = f"dashboard_{date.today().isoformat()}"
-    if insights_key not in st.session_state.insights_cache:
-        insights = generate_ai_response(f"""
-        You are a financial advisor for startups. Based on this startup's data:
-        - Current cash: ${startup_data['cash']}
-        - Monthly burn rate: ${startup_data['burn_rate']}
-        - Monthly revenue: ${startup_data['revenue']}
-        - Monthly growth rate: {startup_data['growth_rate'] * 100}%
-        
-        Provide the top 3 most important financial insights that the founder should know today.
-        Format each insight as a brief, action-oriented bullet point.
-        """, simulate=True)
-        st.session_state.insights_cache[insights_key] = insights
-    
-    with st.expander("📊 AI Financial Insights", expanded=True):
-        st.markdown("<span class='ai-badge'>AI-Generated Insights</span>", unsafe_allow_html=True)
-        st.markdown(st.session_state.insights_cache[insights_key])
+    """Render financial dashboard page."""
+    st.title("Financial Dashboard")
     
     # Key metrics
     col1, col2, col3, col4 = st.columns(4)
@@ -55,42 +18,14 @@ def render_financial_dashboard(startup_data, cash_flow_df):
         startup_data['growth_rate']
     )
     
-    # Determine status colors based on financial health indicators
-    runway_status = "danger-metric" if runway_months < 6 else ("warning-metric" if runway_months < 9 else "good-metric")
-    burn_status = "danger-metric" if startup_data['burn_rate'] > 100000 else ("warning-metric" if startup_data['burn_rate'] > 80000 else "good-metric")
-    revenue_status = "good-metric" if startup_data['revenue'] > 20000 else ("warning-metric" if startup_data['revenue'] > 10000 else "danger-metric")
-    
     with col1:
-        st.markdown(f"""
-        <div class='metric-card'>
-            <p class='metric-label'>Current Cash</p>
-            <p class='metric-value'>${startup_data['cash']:,}</p>
-        </div>
-        """, unsafe_allow_html=True)
-    
+        st.metric("Current Cash", f"${startup_data['cash']:,}")
     with col2:
-        st.markdown(f"""
-        <div class='metric-card'>
-            <p class='metric-label'>Monthly Burn</p>
-            <p class='metric-value {burn_status}'>${startup_data['burn_rate']:,}</p>
-        </div>
-        """, unsafe_allow_html=True)
-    
+        st.metric("Monthly Burn", f"${startup_data['burn_rate']:,}")
     with col3:
-        st.markdown(f"""
-        <div class='metric-card'>
-            <p class='metric-label'>Monthly Revenue</p>
-            <p class='metric-value {revenue_status}'>${startup_data['revenue']:,}</p>
-        </div>
-        """, unsafe_allow_html=True)
-    
+        st.metric("Monthly Revenue", f"${startup_data['revenue']:,}")
     with col4:
-        st.markdown(f"""
-        <div class='metric-card'>
-            <p class='metric-label'>Runway</p>
-            <p class='metric-value {runway_status}'>{runway_months} months</p>
-        </div>
-        """, unsafe_allow_html=True)
+        st.metric("Runway", f"{runway_months} months")
     
     # Financial charts
     st.subheader("Financial Overview")
@@ -101,29 +36,19 @@ def render_financial_dashboard(startup_data, cash_flow_df):
         # Runway projection chart
         fig = px.line(runway_df.reset_index(), x='index', y='Cumulative_Cash', 
                      title="Cash Runway Projection",
-                     labels={'index': 'Date', 'Cumulative_Cash': 'Remaining Cash ($)'},
-                     color_discrete_sequence=['#0066cc'])
-        fig.add_hline(y=0, line_dash="dash", line_color="red", annotation_text="Out of Cash")
-        fig.update_layout(
-            height=400,
-            plot_bgcolor='rgba(240,247,255,0.8)',
-            xaxis_title="Date",
-            yaxis_title="Cash Balance ($)",
-            font=dict(family="Arial, sans-serif", size=12),
-            margin=dict(l=20, r=20, t=40, b=20),
-        )
+                     labels={'index': 'Date', 'Cumulative_Cash': 'Remaining Cash'})
+        fig.add_hline(y=0, line_dash="dash", line_color="red")
+        fig.update_layout(height=400)
         st.plotly_chart(fig, use_container_width=True)
         
         # Get analysis from Gemini
-        with st.expander("🔍 AI Financial Analysis", expanded=True):
-            # Use cache to avoid repeated API calls
-            analysis_key = f"runway_{date.today().isoformat()}"
-            if analysis_key not in st.session_state.insights_cache:
-                analysis = get_runway_analysis(startup_data)
-                st.session_state.insights_cache[analysis_key] = analysis
-                
-            st.markdown("<span class='ai-badge'>AI Financial Analysis</span>", unsafe_allow_html=True)
-            st.markdown(st.session_state.insights_cache[analysis_key])
+        try:
+            if setup_genai():
+                with st.expander("AI Financial Analysis"):
+                    analysis = get_runway_analysis(startup_data)
+                    st.write(analysis)
+        except Exception as e:
+            st.error(f"Error generating AI analysis: {e}")
     
     with tab2:
         # Revenue vs Expenses chart
@@ -131,71 +56,17 @@ def render_financial_dashboard(startup_data, cash_flow_df):
         fig = px.bar(rev_exp_df, x='Month', y=['Revenue', 'Total_Expenses'],
                     title="Revenue vs. Expenses",
                     barmode='group',
-                    labels={'value': 'Amount ($)', 'variable': 'Category'},
-                    color_discrete_sequence=['#28a745', '#dc3545'])
-        fig.update_layout(
-            height=400,
-            plot_bgcolor='rgba(240,247,255,0.8)',
-            xaxis_title="Month",
-            yaxis_title="Amount ($)",
-            font=dict(family="Arial, sans-serif", size=12),
-            legend_title="",
-            margin=dict(l=20, r=20, t=40, b=20),
-        )
+                    labels={'value': 'Amount ($)', 'variable': 'Category'})
+        fig.update_layout(height=400)
         st.plotly_chart(fig, use_container_width=True)
-        
-        # Calculate revenue growth
-        revenue_growth = [(cash_flow_df['Revenue'].iloc[i] / cash_flow_df['Revenue'].iloc[i-1] - 1) * 100 if i > 0 else 0 
-                          for i in range(len(cash_flow_df))]
-        avg_growth = sum(revenue_growth[1:]) / len(revenue_growth[1:])
-        
-        col1, col2 = st.columns(2)
-        with col1:
-            st.metric("Average Monthly Revenue Growth", f"{avg_growth:.1f}%")
-        with col2:
-            expense_growth = (cash_flow_df['Total_Expenses'].iloc[-1] / cash_flow_df['Total_Expenses'].iloc[0] - 1) * 100
-            st.metric("Total Expense Growth", f"{expense_growth:.1f}%", delta=f"{expense_growth - avg_growth:.1f}%", delta_color="inverse")
     
     with tab3:
         # Burn rate trend
         fig = px.line(cash_flow_df, x='Month', y='Net_Burn',
                      title="Monthly Net Burn Trend",
-                     labels={'Net_Burn': 'Net Burn ($)'},
-                     color_discrete_sequence=['#dc3545'])
-        fig.update_layout(
-            height=400,
-            plot_bgcolor='rgba(240,247,255,0.8)',
-            xaxis_title="Month",
-            yaxis_title="Net Burn ($)",
-            font=dict(family="Arial, sans-serif", size=12),
-            margin=dict(l=20, r=20, t=40, b=20),
-        )
-        
-        # Add efficiency ratio as a second y-axis
-        efficiency_ratio = [cash_flow_df['Revenue'].iloc[i] / cash_flow_df['Total_Expenses'].iloc[i] * 100 
-                            for i in range(len(cash_flow_df))]
-        
-        fig.add_trace(go.Scatter(
-            x=cash_flow_df['Month'], 
-            y=efficiency_ratio,
-            name='Efficiency Ratio (%)',
-            yaxis='y2',
-            line=dict(color='#0066cc', width=2, dash='dot')
-        ))
-        
-        fig.update_layout(
-            yaxis2=dict(
-                title='Efficiency Ratio (%)',
-                overlaying='y',
-                side='right',
-                range=[0, max(efficiency_ratio) * 1.2]
-            )
-        )
-        
+                     labels={'Net_Burn': 'Net Burn ($)'})
+        fig.update_layout(height=400)
         st.plotly_chart(fig, use_container_width=True)
-        
-        with st.expander("🔎 Understanding Efficiency Ratio"):
-            st.info("The efficiency ratio measures how efficiently your startup is generating revenue relative to expenses. A higher percentage means you're getting more revenue per dollar spent. Venture-backed startups typically aim for at least 40% before Series B funding.")
     
     # Expense breakdown
     st.subheader("Expense Breakdown")
@@ -205,138 +76,111 @@ def render_financial_dashboard(startup_data, cash_flow_df):
     expense_categories = ['Payroll', 'Marketing', 'Office', 'Software', 'Travel', 'Legal', 'Misc']
     expense_values = [last_month[cat] for cat in expense_categories]
     
-    col1, col2 = st.columns([2, 1])
-    
-    with col1:
-        fig = px.pie(values=expense_values, names=expense_categories, 
-                    title="Current Month Expense Breakdown",
-                    color_discrete_sequence=px.colors.sequential.Blues_r)
-        fig.update_layout(
-            height=400,
-            font=dict(family="Arial, sans-serif", size=12),
-            margin=dict(l=20, r=20, t=40, b=20),
-        )
-        fig.update_traces(textposition='inside', textinfo='percent+label')
-        st.plotly_chart(fig, use_container_width=True)
-    
-    with col2:
-        # Expense analysis
-        st.markdown("<h4>Expense Analysis</h4>", unsafe_allow_html=True)
-        
-        # Calculate industry benchmarks (simulated)
-        benchmarks = {
-            "Payroll": "70-80%",
-            "Marketing": "10-15%",
-            "Office": "5-8%",
-            "Software": "3-5%"
-        }
-        
-        # Create a table with expense categories, amounts, and % of total
-        expense_df = pd.DataFrame({
-            "Category": expense_categories,
-            "Amount": expense_values,
-            "% of Total": [v / sum(expense_values) * 100 for v in expense_values]
-        })
-        
-        # Add benchmark column
-        expense_df["Industry Benchmark"] = expense_df["Category"].map(
-            lambda x: benchmarks.get(x, "N/A")
-        )
-        
-        # Format the dataframe for display
-        formatted_df = expense_df.copy()
-        formatted_df["Amount"] = formatted_df["Amount"].apply(lambda x: f"${x:,.0f}")
-        formatted_df["% of Total"] = formatted_df["% of Total"].apply(lambda x: f"{x:.1f}%")
-        
-        st.table(formatted_df)
-        
-        # AI-powered spending optimization
-        with st.expander("💡 AI Spending Optimization"):
-            st.markdown("<span class='ai-badge'>AI Recommendation</span>", unsafe_allow_html=True)
-            
-            # Use cache to avoid repeated API calls
-            spending_key = f"spending_{date.today().isoformat()}"
-            if spending_key not in st.session_state.insights_cache:
-                spending_recommendation = generate_ai_response("""
-                Based on your expense breakdown, recommend 2-3 specific ways to optimize spending to extend runway.
-                Focus on industry best practices for seed-stage startups.
-                """, simulate=True)
-                st.session_state.insights_cache[spending_key] = spending_recommendation
-                
-            st.markdown(st.session_state.insights_cache[spending_key])
-            
-    # Fundraising Readiness Assessment
-    st.subheader("Fundraising Readiness")
-    
-    # Get AI analysis of fundraising readiness
-    fundraising_key = f"fundraising_{date.today().isoformat()}"
-    if fundraising_key not in st.session_state.insights_cache:
-        fundraising_analysis = get_fundraising_readiness_analysis(startup_data, cash_flow_df)
-        st.session_state.insights_cache[fundraising_key] = fundraising_analysis
-    
-    st.markdown("<div class='advisor-card'>", unsafe_allow_html=True)
-    st.markdown("<span class='ai-badge'>AI Fundraising Assessment</span>", unsafe_allow_html=True)
-    st.markdown(f"<p class='advice-text'>{st.session_state.insights_cache[fundraising_key]}</p>", unsafe_allow_html=True)
-    st.markdown("</div>", unsafe_allow_html=True)
-    
-    # Call-to-action for advisor
-    st.info("📅 Need personalized guidance on fundraising? [Book a session](#book-a-session) with our AI financial advisor.")
+    fig = px.pie(values=expense_values, names=expense_categories, 
+                title="Current Month Expense Breakdown")
+    fig.update_layout(height=400)
+    st.plotly_chart(fig, use_container_width=True)
+
+def calculate_runway(initial_cash, monthly_burn, monthly_revenue, growth_rate, months=24):
+    """Calculate runway based on current burn rate and revenue growth."""
+    from datetime import datetime, timedelta
+    import pandas as pd
+    
+    dates = [datetime.now() + timedelta(days=30*i) for i in range(months)]
+    df = pd.DataFrame(index=dates, columns=['Cash', 'Revenue', 'Expenses', 'Net_Burn', 'Cumulative_Cash'])
+    
+    current_cash = initial_cash
+    current_revenue = monthly_revenue
+    df.iloc[0, df.columns.get_loc('Cash')] = current_cash
+    df.iloc[0, df.columns.get_loc('Revenue')] = current_revenue
+    df.iloc[0, df.columns.get_loc('Expenses')] = monthly_burn
+    df.iloc[0, df.columns.get_loc('Net_Burn')] = monthly_burn - current_revenue
+    df.iloc[0, df.columns.get_loc('Cumulative_Cash')] = current_cash
+    
+    runway_months = months
+    for i in range(1, months):
+        current_revenue = current_revenue * (1 + growth_rate)
+        net_burn = monthly_burn - current_revenue
+        current_cash = current_cash - net_burn
+        
+        df.iloc[i, df.columns.get_loc('Cash')] = current_cash
+        df.iloc[i, df.columns.get_loc('Revenue')] = current_revenue
+        df.iloc[i, df.columns.get_loc('Expenses')] = monthly_burn
+        df.iloc[i, df.columns.get_loc('Net_Burn')] = net_burn
+        df.iloc[i, df.columns.get_loc('Cumulative_Cash')] = current_cash
+        
+        if current_cash <= 0:
+            runway_months = i
+            break
+    
+    return runway_months, df
 
 def get_runway_analysis(financial_data):
     """Get runway analysis using Gemini."""
-    prompt = f"""
-    You are a financial advisor for startups. Analyze this startup's financial data:
-    - Current cash: ${financial_data['cash']}
-    - Monthly burn rate: ${financial_data['burn_rate']}
-    - Monthly revenue: ${financial_data['revenue']}
-    - Monthly growth rate: {financial_data['growth_rate'] * 100}%
+    try:
+        prompt = f"""
+        You are a financial advisor for startups. Analyze this startup's financial data:
+        - Current cash: ${financial_data['cash']}
+        - Monthly burn rate: ${financial_data['burn_rate']}
+        - Monthly revenue: ${financial_data['revenue']}
+        - Monthly growth rate: {financial_data['growth_rate'] * 100}%
+
+        Calculate and explain their runway, financial health, and recommendations in a concise paragraph.
+        """
+        
+        model = genai.GenerativeModel('gemini-pro')
+        response = model.generate_content(prompt)
+        
+        return response.text
+    except Exception as e:
+        return f"Error generating runway analysis: {e}"
 
-    Provide a detailed analysis of their runway and financial health. Include:
-    1. Exact runway calculation in months
-    2. Assessment of financial health (critical, concerning, stable, or healthy)
-    3. Benchmarks compared to similar seed-stage startups
-    4. Three specific, actionable recommendations to improve runway
-    5. Key metrics they should focus on
+def get_fundraising_readiness_analysis(financial_data):
+    """Analyze startup's readiness for fundraising."""
+    try:
+        prompt = f"""
+        You are a fundraising advisor for startups. Evaluate this startup's fundraising readiness:
+        - Current cash: ${financial_data['cash']}
+        - Monthly burn rate: ${financial_data['burn_rate']}
+        - Monthly revenue: ${financial_data['revenue']}
+        - Monthly growth rate: {financial_data['growth_rate'] * 100}%
+        - Last funding: {financial_data.get('last_funding', 'Not specified')}
+        - Company stage: {financial_data.get('stage', 'Not specified')}
 
-    Format your response in a structured, easy-to-read format with clear sections and bullet points.
-    """
-    
-    return generate_ai_response(prompt)
+        Provide insights on:
+        1. Current runway and funding needs
+        2. Attractiveness to potential investors
+        3. Recommended fundraising strategy
+        4. Key metrics to improve before fundraising
 
-def get_fundraising_readiness_analysis(startup_data, cash_flow_df):
-    """Get AI analysis of fundraising readiness."""
-    metrics = {
-        "MRR Growth": f"{(cash_flow_df['Revenue'].iloc[-1] / cash_flow_df['Revenue'].iloc[-2] - 1) * 100:.1f}%",
-        "Gross Margin": f"{(cash_flow_df['Revenue'].iloc[-1] - cash_flow_df['Total_Expenses'].iloc[-1] / 2) / cash_flow_df['Revenue'].iloc[-1] * 100:.1f}%",
-        "CAC": "$950",  # Example value
-        "LTV": "$4,500",  # Example value
-        "Churn": "3.2%",  # Example value
-    }
-    
-    metrics_text = "\n".join([f"- {k}: {v}" for k, v in metrics.items()])
-    
-    prompt = f"""
-    You are a startup fundraising advisor. Analyze this startup's readiness for their next funding round:
-    
-    Company Profile:
-    - Stage: {startup_data['stage']}
-    - Last Funding: {startup_data['last_funding']}
-    - Current Cash: ${startup_data['cash']}
-    - Monthly Burn: ${startup_data['burn_rate']}
-    - Runway: {startup_data['cash'] / (startup_data['burn_rate'] - startup_data['revenue']):.1f} months
-    
-    Key Metrics:
-    {metrics_text}
-    
-    Provide a comprehensive fundraising readiness assessment:
-    1. Overall fundraising readiness score (0-10)
-    2. Assessment of current metrics compared to investor expectations for next round
-    3. Identify the 3 most critical metrics to improve before fundraising
-    4. Recommend specific targets for each key metric
-    5. Suggest timeline and specific milestones for fundraising preparation
-    6. Estimate reasonable valuation range based on metrics and market conditions
-    
-    Be specific with numbers, timelines, and actionable targets.
-    """
-    
-    return generate_ai_response(prompt)
+        Respond in a concise, actionable paragraph.
+        """
+        
+        model = genai.GenerativeModel('gemini-pro')
+        response = model.generate_content(prompt)
+        
+        return response.text
+    except Exception as e:
+        return f"Error generating fundraising readiness analysis: {e}"
+
+def setup_genai():
+    """Setup Google Generative AI"""
+    try:
+        import os
+        import streamlit as st
+        
+        # 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.")
+            return False
+            
+        genai.configure(api_key=api_key)
+        return True
+    except Exception as e:
+        st.error(f"Error setting up Generative AI: {e}")
+        return False