Upload 2 files

app/main.py

@@ -526,6 +526,20 @@ class HVACCalculator:
             if not summary:
                 return False, "Cooling load summary calculation failed. Check input data.", {}
             
+            # Ensure summary has all required keys
+            if 'total' not in summary:
+                # Calculate total if missing
+                if 'total_sensible' in summary and 'total_latent' in summary:
+                    summary['total'] = summary['total_sensible'] + summary['total_latent']
+                else:
+                    # Fallback to sum of design loads if needed
+                    total_load = sum(value for key, value in design_loads.items() if key != 'design_hour')
+                    summary = {
+                        'total_sensible': total_load * 0.7,  # Approximate sensible ratio
+                        'total_latent': total_load * 0.3,    # Approximate latent ratio
+                        'total': total_load
+                    }
+            
             # Format results for results_display.py
             floor_area = building_info.get('floor_area', 100.0) or 100.0
             results = {