Update app/results_display.py

app/results_display.py

@@ -299,7 +299,7 @@ class ResultsDisplay:
                         "Name": roof["name"],
                         "Orientation": roof["orientation"],
                         "Area (m²)": roof["area"],
-                        "U-Value (W/m²·K)": roof["u_value"],
+                        "U-Value (W/m²·K)": wall["u_value"],
                         "Temperature Difference (°C)": roof["delta_t"],
                         "Load (kW)": roof["load"]
                     })
@@ -401,11 +401,65 @@ class ResultsDisplay:
             st.warning("No component breakdown data available.")
             return
         
-        # Use component visualization module
-        self.component_visualization.display_component_breakdown(
-            session_state["calculation_results"],
-            session_state["components"]
-        )
+        # Try to use component visualization module
+        try:
+            self.component_visualization.display_component_breakdown(
+                session_state["calculation_results"],
+                session_state["components"]
+            )
+        except AttributeError:
+            # Fallback visualization if display_component_breakdown is not available
+            st.info("Component visualization module not fully implemented. Displaying default breakdown.")
+            
+            results = session_state["calculation_results"]
+            
+            # Cooling load bar chart
+            if results.get("cooling"):
+                cooling_breakdown = {
+                    "Walls": results["cooling"]["component_loads"]["walls"],
+                    "Roof": results["cooling"]["component_loads"]["roof"],
+                    "Windows": results["cooling"]["component_loads"]["windows"],
+                    "Doors": results["cooling"]["component_loads"]["doors"],
+                    "People": results["cooling"]["component_loads"]["people"],
+                    "Lighting": results["cooling"]["component_loads"]["lighting"],
+                    "Equipment": results["cooling"]["component_loads"]["equipment"],
+                    "Infiltration": results["cooling"]["component_loads"]["infiltration"],
+                    "Ventilation": results["cooling"]["component_loads"]["ventilation"]
+                }
+                
+                fig_cooling = px.bar(
+                    x=list(cooling_breakdown.keys()),
+                    y=list(cooling_breakdown.values()),
+                    title="Cooling Load by Component",
+                    labels={"x": "Component", "y": "Load (kW)"},
+                    color_discrete_sequence=px.colors.qualitative.Pastel
+                )
+                
+                fig_cooling.update_layout(showlegend=False)
+                st.plotly_chart(fig_cooling, use_container_width=True)
+            
+            # Heating load bar chart
+            if results.get("heating"):
+                heating_breakdown = {
+                    "Walls": results["heating"]["component_loads"]["walls"],
+                    "Roof": results["heating"]["component_loads"]["roof"],
+                    "Floor": results["heating"]["component_loads"]["floor"],
+                    "Windows": results["heating"]["component_loads"]["windows"],
+                    "Doors": results["heating"]["component_loads"]["doors"],
+                    "Infiltration": results["heating"]["component_loads"]["infiltration"],
+                    "Ventilation": results["heating"]["component_loads"]["ventilation"]
+                }
+                
+                fig_heating = px.bar(
+                    x=list(heating_breakdown.keys()),
+                    y=list(heating_breakdown.values()),
+                    title="Heating Load by Component",
+                    labels={"x": "Component", "y": "Load (kW)"},
+                    color_discrete_sequence=px.colors.qualitative.Pastel
+                )
+                
+                fig_heating.update_layout(showlegend=False)
+                st.plotly_chart(fig_heating, use_container_width=True)
     
     def _display_psychrometric_analysis(self, session_state: Dict[str, Any]) -> None:
         """