Add 899999999999999999999

enhanced_websocket_handler.py

@@ -441,42 +441,61 @@ async def get_hybrid_response(user_message: str, context: str, config: dict, kno
         from rag_service import search_documents_async
         docs = await search_documents_async(user_message, limit=3)
         if docs:
-            from scenario_analysis_service import run_scenario_analysis
-            # Detect scenario analysis intent (simple keyword match)
-            scenario_keywords = ["impact", "cost", "scenario", "multiplier", "da", "dr"]
-            if any(kw in user_message.lower() for kw in scenario_keywords):
-                # Example params extraction (can be improved)
-                params = {
-                    'base_pension': 30000,
-                    'multiplier': 1.1 if "multiplier" in user_message.lower() else 1.0,
-                    'da_percent': 0.06 if "da" in user_message.lower() else 0.0,
-                    'num_beneficiaries': 1000,
-                    'years': 3,
-                    'inflation': 0.05
-                }
-                scenario_result = run_scenario_analysis(params)
-                # Generate charts for scenario_result
-                chart_gen = PolicyChartGenerator()
-                charts = []
-                # Example: line chart for yearly results
-                if "yearly_results" in scenario_result:
-                    years = [r['year'] for r in scenario_result['yearly_results']]
-                    base_costs = [r['base_cost'] for r in scenario_result['yearly_results']]
-                    scenario_costs = [r['scenario_cost'] for r in scenario_result['yearly_results']]
-                    # Generate chart and append to charts list
-                    fig, ax = plt.subplots()
-                    ax.plot(years, base_costs, label='Base Cost')
-                    ax.plot(years, scenario_costs, label='Scenario Cost')
-                    ax.legend()
-                    ax.set_title('Scenario Analysis: Cost Over Years')
-                    buf = io.BytesIO()
-                    fig.savefig(buf, format='png')
-                    buf.seek(0)
-                    chart_base64 = base64.b64encode(buf.read()).decode('utf-8')
-                    plt.close(fig)
-                    charts.append({"type": "line_chart", "data": chart_base64})
-                scenario_result["charts"] = charts
-            else:
+            try:
+                from scenario_analysis_service import run_scenario_analysis
+                # Detect scenario analysis intent (simple keyword match)
+                scenario_keywords = ["impact", "cost", "scenario", "multiplier", "da", "dr"]
+                if any(kw in user_message.lower() for kw in scenario_keywords):
+                    logger.info("🔍 Running scenario analysis")
+                    # Example params extraction (can be improved)
+                    params = {
+                        'base_pension': 30000,
+                        'multiplier': 1.1 if "multiplier" in user_message.lower() else 1.0,
+                        'da_percent': 0.06 if "da" in user_message.lower() else 0.0,
+                        'num_beneficiaries': 1000,
+                        'years': 3,
+                        'inflation': 0.05
+                    }
+                    scenario_result = run_scenario_analysis(params)
+                    
+                    # Generate charts for scenario_result
+                    try:
+                        chart_gen = PolicyChartGenerator()
+                        charts = []
+                        # Example: line chart for yearly results
+                        if "yearly_results" in scenario_result:
+                            years = [r['year'] for r in scenario_result['yearly_results']]
+                            base_costs = [r['base_cost'] for r in scenario_result['yearly_results']]
+                            scenario_costs = [r['scenario_cost'] for r in scenario_result['yearly_results']]
+                            # Generate chart and append to charts list
+                            fig, ax = plt.subplots(figsize=(10, 6))
+                            ax.plot(years, base_costs, label='Base Cost', marker='o')
+                            ax.plot(years, scenario_costs, label='Scenario Cost', marker='s')
+                            ax.legend()
+                            ax.set_title('Scenario Analysis: Cost Over Years')
+                            ax.set_xlabel('Year')
+                            ax.set_ylabel('Cost (₹)')
+                            ax.grid(True, alpha=0.3)
+                            
+                            # Format y-axis to show values in lakhs
+                            ax.yaxis.set_major_formatter(plt.FuncFormatter(lambda x, p: f'₹{x/100000:.1f}L'))
+                            
+                            buf = io.BytesIO()
+                            fig.savefig(buf, format='png', dpi=150, bbox_inches='tight')
+                            buf.seek(0)
+                            chart_base64 = base64.b64encode(buf.read()).decode('utf-8')
+                            plt.close(fig)
+                            charts.append({"type": "line_chart", "data": chart_base64})
+                            logger.info(f"✅ Generated {len(charts)} charts for scenario analysis")
+                        scenario_result["charts"] = charts
+                    except Exception as chart_error:
+                        logger.error(f"❌ Failed to generate charts: {chart_error}")
+                        scenario_result["charts"] = []
+                        scenario_result["chart_error"] = str(chart_error)
+                else:
+                    scenario_result = None
+            except Exception as scenario_error:
+                logger.error(f"❌ Scenario analysis failed: {scenario_error}")
                 scenario_result = None
             for doc in docs:
                 response_obj = {