Update app.py

logic improvements

app.py

@@ -1339,18 +1339,47 @@ def create_app():
                     department_df, subscription_df, compliance_df, 
                     initial_dev_cost, num_ai_personnel, avg_annual_salary,
                     annual_maintenance, security_compliance, step):
-                dept_roi_results = calculate_department_roi(department_df)
-                sub_savings = calculate_subscription_savings(subscription_df)
-                comp_savings = calculate_compliance_savings(compliance_df)
-                report, comparison = generate_recommendations(
-                    industry, org_size, monthly_budget, has_ai_app,
-                    dept_roi_results, sub_savings, comp_savings, api_calls,
-                    initial_dev_cost, num_ai_personnel, avg_annual_salary,
-                    annual_maintenance, security_compliance)
-                indicators = update_progress_indicator(4)
-                return (gr.update(visible=False), gr.update(visible=False),
-                        gr.update(visible=False), gr.update(visible=False),
-                        gr.update(visible=True), report, *indicators)
+                try:
+                    # Convert inputs to appropriate types with error handling
+                    org_size_val = int(org_size) if org_size is not None else 90
+                    monthly_budget_val = float(monthly_budget) if monthly_budget is not None else 30000
+                    api_calls_val = int(api_calls) if api_calls is not None else 10000
+                    initial_dev_cost_val = float(initial_dev_cost) if initial_dev_cost is not None else 1000000
+                    num_ai_personnel_val = int(num_ai_personnel) if num_ai_personnel is not None else 1
+                    avg_annual_salary_val = float(avg_annual_salary) if avg_annual_salary is not None else 200000
+                    annual_maintenance_val = float(annual_maintenance) if annual_maintenance is not None else 500000
+                    security_compliance_val = float(security_compliance) if security_compliance is not None else 250000
+
+                    # Validate dataframes and convert to default if invalid
+                    if department_df is None or department_df.empty:
+                        department_df = default_departments
+                    if subscription_df is None or subscription_df.empty:
+                        subscription_df = default_subscriptions
+                    if compliance_df is None or compliance_df.empty:
+                        compliance_df = industry_compliance_data.get(industry, industry_compliance_data["Other"])
+
+                    # Calculate ROIs and savings
+                    dept_roi_results = calculate_department_roi(department_df)
+                    sub_savings = calculate_subscription_savings(subscription_df)
+                    comp_savings = calculate_compliance_savings(compliance_df)
+
+                    report, comparison = generate_recommendations(
+                        industry, org_size_val, monthly_budget_val, has_ai_app,
+                        dept_roi_results, sub_savings, comp_savings, api_calls_val,
+                        initial_dev_cost_val, num_ai_personnel_val, avg_annual_salary_val,
+                        annual_maintenance_val, security_compliance_val)
+
+                    indicators = update_progress_indicator(4)
+                    return (gr.update(visible=False), gr.update(visible=False),
+                            gr.update(visible=False), gr.update(visible=False),
+                            gr.update(visible=True), report, *indicators)
+                except Exception as e:
+                    # Provide a graceful error message instead of crashing
+                    error_message = f"An error occurred while generating recommendations: {str(e)}"
+                    indicators = update_progress_indicator(4)
+                    return (gr.update(visible=False), gr.update(visible=False),
+                            gr.update(visible=False), gr.update(visible=False),
+                            gr.update(visible=True), error_message, *indicators)
 
 
             next_button1.click(fn=go_to_step2,