Update app.py

app.py

@@ -162,15 +162,72 @@ def calculate_runway(cash, burn_rate, revenue, growth_rate, months=24):
 
 def simulate_decision(cash, burn_rate, revenue, growth_rate, 
                      additional_expenses, new_hires, marketing_increase, growth_impact):
-    """Simulate the financial impact of a business decision"""
+    """
+    Simulate the financial impact of a business decision with AI-powered insights
+    
+    Args:
+    - cash: Current cash balance
+    - burn_rate: Current monthly burn rate
+    - revenue: Current monthly revenue
+    - growth_rate: Current monthly growth rate
+    - additional_expenses: Proposed additional monthly expenses
+    - new_hires: Number of new hires
+    - marketing_increase: Proposed marketing budget increase
+    - growth_impact: Expected growth rate impact
+    
+    Returns:
+    - current_runway: Current financial runway in months
+    - new_runway: Projected runway after proposed changes
+    - current_df: DataFrame with current financial projection
+    - new_df: DataFrame with projected financial scenario
+    - ai_analysis: AI-generated insights about the decision
+    """
+    # Calculate current runway
     current_runway, current_df = calculate_runway(cash, burn_rate, revenue, growth_rate)
     
+    # Calculate new financial parameters
     new_burn_rate = burn_rate + additional_expenses + (new_hires * ENGINEER_SALARY) + marketing_increase
     new_growth_rate = growth_rate + growth_impact
     
+    # Calculate new runway
     new_runway, new_df = calculate_runway(cash, new_burn_rate, revenue, new_growth_rate)
     
-    return current_runway, new_runway, current_df, new_df
+    # Generate AI analysis of the decision
+    try:
+        ai_analysis = generate_ai_response(f"""
+        You are a strategic financial advisor for startups. Analyze this potential business decision:
+
+        Current Financial Situation:
+        - Cash Balance: ${cash:,}
+        - Monthly Burn Rate: ${burn_rate:,}
+        - Monthly Revenue: ${revenue:,}
+        - Current Growth Rate: {growth_rate * 100:.1f}%
+        - Current Runway: {current_runway} months
+
+        Proposed Changes:
+        - Additional Expenses: ${additional_expenses:,}/month
+        - New Hires: {new_hires} engineers (${new_hires * ENGINEER_SALARY:,}/month)
+        - Marketing Budget Increase: ${marketing_increase:,}/month
+        - Expected Growth Impact: +{growth_impact * 100:.1f}%
+
+        Projected Outcome:
+        - New Burn Rate: ${new_burn_rate:,}/month
+        - New Growth Rate: {new_growth_rate * 100:.1f}%
+        - Projected Runway: {new_runway} months
+
+        Provide a comprehensive analysis addressing:
+        1. Financial feasibility of the proposed changes
+        2. Risk assessment
+        3. Potential strategic benefits
+        4. Recommendations for optimization
+        5. Key metrics to monitor
+
+        Be direct, specific, and provide actionable insights.
+        """, simulate=False)
+    except Exception as e:
+        ai_analysis = f"AI analysis unavailable. Error: {str(e)}"
+    
+    return current_runway, new_runway, current_df, new_df, ai_analysis
 
 def detect_suspicious_transactions(transactions_df):
     """AI-enhanced suspicious transaction detection"""
@@ -480,15 +537,11 @@ def render_financial_dashboard():
     else:
         st.info("Upload cash flow data to see detailed financial charts")
 
-def render_decision_simulator():
+
+
+        
+def render_decision_simulator(startup_data):
     """Render the decision simulator page"""
-    if not st.session_state.current_startup or st.session_state.current_startup not in st.session_state.startups:
-        st.warning("No startup selected. Please upload data first.")
-        render_upload_page()
-        return
-    
-    startup_data = st.session_state.startups[st.session_state.current_startup]['profile']
-    
     st.markdown("<h1 class='main-header'>Decision Simulator</h1>", unsafe_allow_html=True)
     st.markdown("<p class='sub-header'>Test the financial impact of business decisions</p>", unsafe_allow_html=True)
     
@@ -528,8 +581,8 @@ def render_decision_simulator():
         submitted = st.form_submit_button("Simulate Decision")
     
     if submitted:
-        # Calculate current and new runway
-        current_runway, new_runway, current_df, new_df = simulate_decision(
+        # Calculate current and new runway with AI analysis
+        current_runway, new_runway, current_df, new_df, ai_analysis = simulate_decision(
             startup_data['cash'],
             startup_data['burn_rate'],
             startup_data['revenue'],
@@ -543,7 +596,7 @@ def render_decision_simulator():
         # Display results
         st.markdown("<h3>Decision Impact Analysis</h3>", unsafe_allow_html=True)
         
-        # Summary metrics
+        # Summary metrics (existing code remains the same)
         col1, col2, col3 = st.columns(3)
         
         with col1:
@@ -561,7 +614,7 @@ def render_decision_simulator():
                      delta=f"${burn_change:,} ({burn_percentage:.1f}%)", 
                      delta_color="inverse")
         
-        # Cash projection comparison
+        # Cash projection comparison (existing code remains the same)
         st.subheader("Cash Projection Comparison")
         
         # Combine dataframes for comparison
@@ -583,40 +636,12 @@ def render_decision_simulator():
         
         st.plotly_chart(fig, use_container_width=True)
         
-        # Get AI analysis
-        if question:
-            analysis_key = f"decision_analysis_{new_hires}_{new_marketing}_{other_expenses}_{growth_impact}"
-            if analysis_key not in st.session_state.insights_cache:
-                analysis = generate_ai_response(f"""
-                You are a financial advisor for startups. A founder asks:
-                "{question}"
-
-                Here's their current financial situation:
-                - 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}%
-                
-                They're considering these changes:
-                - Adding {new_hires} new engineers (${ENGINEER_SALARY}/month each)
-                - Increasing marketing budget by ${new_marketing}/month
-                - Adding ${other_expenses}/month in other expenses
-                - Expecting {growth_impact * 100}% additional monthly growth
-
-                Analyze this decision thoroughly:
-                1. Quantify the impact on runway
-                2. Assess the risk level (low, medium, high)
-                3. Compare the ROI potential
-                4. Provide recommendations
-                
-                Be direct and specific with numbers and timeframes.
-                """)
-                st.session_state.insights_cache[analysis_key] = analysis
-            
-            st.markdown("<div class='advisor-card'>", unsafe_allow_html=True)
-            st.markdown("<span class='ai-badge'>AI Decision Analysis</span>", unsafe_allow_html=True)
-            st.markdown(f"<p class='advice-text'>{st.session_state.insights_cache[analysis_key]}</p>", unsafe_allow_html=True)
-            st.markdown("</div>", unsafe_allow_html=True)
+        # Display AI Analysis (New section)
+        st.markdown("<div class='advisor-card'>", unsafe_allow_html=True)
+        st.markdown("<span class='ai-badge'>AI Decision Analysis</span>", unsafe_allow_html=True)
+        st.markdown(f"<p class='advice-text'>{ai_analysis}</p>", unsafe_allow_html=True)
+        st.markdown("</div>", unsafe_allow_html=True)
+        
 
 def render_fund_monitoring():
     """Render the fund monitoring page"""