Update response_formatter.py

response_formatter.py

@@ -117,6 +117,9 @@ class ResponseFormatter:
         """Format facility distribution analysis results"""
         response = ""
         
+        if "total_facilities" in dist_results:
+            response += f"**Total Facilities**: {dist_results['total_facilities']:,}\n\n"
+        
         if "geographic_distribution" in dist_results:
             response += "### Geographic Distribution\n\n"
             response += "| Region | Facility Count |\n"
@@ -142,6 +145,22 @@ class ResponseFormatter:
                 diversity = dist_results["facility_diversity"]
                 response += f"**Facility Diversity Index**: {diversity:.2f} (higher values indicate more diversity)\n\n"
         
+        if "top_cities" in dist_results and "city_breakdown" in dist_results:
+            response += "### Top Cities by Facility Count\n\n"
+            response += "| City | Hospitals | Nursing/Residential | Ambulatory | Total |\n"
+            response += "|------|-----------|-------------------|------------|-------|\n"
+            
+            for city in dist_results["top_cities"]:
+                if city in dist_results["city_breakdown"]:
+                    breakdown = dist_results["city_breakdown"][city]
+                    hospitals = breakdown.get("Hospitals", 0)
+                    nursing = breakdown.get("Nursing and residential care facilities", 0)
+                    ambulatory = breakdown.get("Ambulatory health care services", 0)
+                    total = hospitals + nursing + ambulatory
+                    response += f"| {city} | {hospitals} | {nursing} | {ambulatory} | {total} |\n"
+            
+            response += "\n"
+        
         return response if response else "No facility distribution data available"
     
     @staticmethod
@@ -183,6 +202,89 @@ class ResponseFormatter:
                 growth = capacity_results["capacity_growth_rate"]
                 response += f"**Overall Growth Rate**: {growth:.1f}%\n\n"
         
+        if "zone_summary" in capacity_results:
+            response += "### Bed Capacity by Zone\n\n"
+            response += "| Zone | Current Beds | Previous Beds | Absolute Change | Percentage Change |\n"
+            response += "|------|--------------|---------------|-----------------|-------------------|\n"
+            
+            for zone_data in capacity_results["zone_summary"]:
+                # Extract zone name - try different possible keys
+                zone_name = "Unknown"
+                for key in ["zone", "Zone", "ZONE", "region", "Region", "REGION"]:
+                    if key in zone_data:
+                        zone_name = zone_data[key]
+                        break
+                
+                # Extract bed counts
+                current_beds = zone_data.get("beds_current", zone_data.get("current", "N/A"))
+                prev_beds = zone_data.get("beds_prev", zone_data.get("previous", "N/A"))
+                abs_change = zone_data.get("bed_change", "N/A")
+                pct_change = zone_data.get("percent_change", "N/A")
+                
+                response += f"| {zone_name} | {current_beds} | {prev_beds} | {abs_change} | {pct_change:.1f}% |\n"
+            
+            response += "\n"
+        
+        if "max_absolute_decrease" in capacity_results and isinstance(capacity_results["max_absolute_decrease"], dict):
+            response += "### Zone with Largest Absolute Decrease\n\n"
+            max_abs = capacity_results["max_absolute_decrease"]
+            
+            # Extract zone name
+            zone_name = "Unknown"
+            for key in ["zone", "Zone", "ZONE", "region", "Region", "REGION"]:
+                if key in max_abs:
+                    zone_name = max_abs[key]
+                    break
+            
+            # Extract values
+            abs_change = max_abs.get("bed_change", "N/A")
+            pct_change = max_abs.get("percent_change", "N/A")
+            
+            response += f"**Zone**: {zone_name}\n"
+            response += f"**Absolute Decrease**: {abs_change} beds\n"
+            response += f"**Percentage Decrease**: {pct_change:.1f}%\n\n"
+        
+        if "max_percentage_decrease" in capacity_results and isinstance(capacity_results["max_percentage_decrease"], dict):
+            response += "### Zone with Largest Percentage Decrease\n\n"
+            max_pct = capacity_results["max_percentage_decrease"]
+            
+            # Extract zone name
+            zone_name = "Unknown"
+            for key in ["zone", "Zone", "ZONE", "region", "Region", "REGION"]:
+                if key in max_pct:
+                    zone_name = max_pct[key]
+                    break
+            
+            # Extract values
+            abs_change = max_pct.get("bed_change", "N/A")
+            pct_change = max_pct.get("percent_change", "N/A")
+            
+            response += f"**Zone**: {zone_name}\n"
+            response += f"**Absolute Decrease**: {abs_change} beds\n"
+            response += f"**Percentage Decrease**: {pct_change:.1f}%\n\n"
+        
+        if "facilities_with_largest_declines" in capacity_results:
+            response += "### Facilities with Largest Bed Declines\n\n"
+            response += "| Facility | Zone | Teaching Status | Beds Lost |\n"
+            response += "|----------|------|----------------|-----------|\n"
+            
+            for facility in capacity_results["facilities_with_largest_declines"]:
+                name = facility.get("facility_name", "N/A")
+                
+                # Extract zone name
+                zone_name = "Unknown"
+                for key in ["zone", "Zone", "ZONE", "region", "Region", "REGION"]:
+                    if key in facility:
+                        zone_name = facility[key]
+                        break
+                
+                teaching = facility.get("teaching_status", "N/A")
+                change = facility.get("bed_change", "N/A")
+                
+                response += f"| {name} | {zone_name} | {teaching} | {change} |\n"
+            
+            response += "\n"
+        
         return response if response else "No capacity data available"
     
     @staticmethod