Update app/results_display.py

app/results_display.py

@@ -94,84 +94,92 @@ class ResultsDisplay:
         with col1:
             st.write("### Cooling Load Results")
             
-            # Display cooling load metrics
-            cooling_metrics = [
-                {"name": "Total Cooling Load", "value": results["cooling"]["total_load"], "unit": "kW"},
-                {"name": "Sensible Cooling Load", "value": results["cooling"]["sensible_load"], "unit": "kW"},
-                {"name": "Latent Cooling Load", "value": results["cooling"]["latent_load"], "unit": "kW"},
-                {"name": "Cooling Load per Area", "value": results["cooling"]["load_per_area"], "unit": "W/m²"}
-            ]
-            
-            for metric in cooling_metrics:
-                st.metric(
-                    label=metric["name"],
-                    value=f"{metric['value']:.2f} {metric['unit']}"
+            # Check if cooling results are available
+            if not results.get("cooling") or "total_load" not in results["cooling"]:
+                st.warning("Cooling load results are not available. Please check calculation inputs and try again.")
+            else:
+                # Display cooling load metrics
+                cooling_metrics = [
+                    {"name": "Total Cooling Load", "value": results["cooling"]["total_load"], "unit": "kW"},
+                    {"name": "Sensible Cooling Load", "value": results["cooling"]["sensible_load"], "unit": "kW"},
+                    {"name": "Latent Cooling Load", "value": results["cooling"]["latent_load"], "unit": "kW"},
+                    {"name": "Cooling Load per Area", "value": results["cooling"]["load_per_area"], "unit": "W/m²"}
+                ]
+                
+                for metric in cooling_metrics:
+                    st.metric(
+                        label=metric["name"],
+                        value=f"{metric['value']:.2f} {metric['unit']}"
+                    )
+                
+                # Display cooling load pie chart
+                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 = px.pie(
+                    values=list(cooling_breakdown.values()),
+                    names=list(cooling_breakdown.keys()),
+                    title="Cooling Load Breakdown",
+                    color_discrete_sequence=px.colors.qualitative.Pastel
                 )
-            
-            # Display cooling load pie chart
-            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 = px.pie(
-                values=list(cooling_breakdown.values()),
-                names=list(cooling_breakdown.keys()),
-                title="Cooling Load Breakdown",
-                color_discrete_sequence=px.colors.qualitative.Pastel
-            )
-            
-            fig.update_traces(textposition='inside', textinfo='percent+label')
-            fig.update_layout(uniformtext_minsize=12, uniformtext_mode='hide')
-            
-            st.plotly_chart(fig, use_container_width=True)
+                
+                fig.update_traces(textposition='inside', textinfo='percent+label')
+                fig.update_layout(uniformtext_minsize=12, uniformtext_mode='hide')
+                
+                st.plotly_chart(fig, use_container_width=True)
         
         with col2:
             st.write("### Heating Load Results")
             
-            # Display heating load metrics
-            heating_metrics = [
-                {"name": "Total Heating Load", "value": results["heating"]["total_load"], "unit": "kW"},
-                {"name": "Heating Load per Area", "value": results["heating"]["load_per_area"], "unit": "W/m²"},
-                {"name": "Design Heat Loss", "value": results["heating"]["design_heat_loss"], "unit": "kW"},
-                {"name": "Safety Factor", "value": results["heating"]["safety_factor"], "unit": "%"}
-            ]
-            
-            for metric in heating_metrics:
-                st.metric(
-                    label=metric["name"],
-                    value=f"{metric['value']:.2f} {metric['unit']}"
+            # Check if heating results are available
+            if not results.get("heating") or "total_load" not in results["heating"]:
+                st.warning("Heating load results are not available. Please check calculation inputs and try again.")
+            else:
+                # Display heating load metrics
+                heating_metrics = [
+                    {"name": "Total Heating Load", "value": results["heating"]["total_load"], "unit": "kW"},
+                    {"name": "Heating Load per Area", "value": results["heating"]["load_per_area"], "unit": "W/m²"},
+                    {"name": "Design Heat Loss", "value": results["heating"]["design_heat_loss"], "unit": "kW"},
+                    {"name": "Safety Factor", "value": results["heating"]["safety_factor"], "unit": "%"}
+                ]
+                
+                for metric in heating_metrics:
+                    st.metric(
+                        label=metric["name"],
+                        value=f"{metric['value']:.2f} {metric['unit']}"
+                    )
+                
+                # Display heating load pie chart
+                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 = px.pie(
+                    values=list(heating_breakdown.values()),
+                    names=list(heating_breakdown.keys()),
+                    title="Heating Load Breakdown",
+                    color_discrete_sequence=px.colors.qualitative.Pastel
                 )
-            
-            # Display heating load pie chart
-            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 = px.pie(
-                values=list(heating_breakdown.values()),
-                names=list(heating_breakdown.keys()),
-                title="Heating Load Breakdown",
-                color_discrete_sequence=px.colors.qualitative.Pastel
-            )
-            
-            fig.update_traces(textposition='inside', textinfo='percent+label')
-            fig.update_layout(uniformtext_minsize=12, uniformtext_mode='hide')
-            
-            st.plotly_chart(fig, use_container_width=True)
+                
+                fig.update_traces(textposition='inside', textinfo='percent+label')
+                fig.update_layout(uniformtext_minsize=12, uniformtext_mode='hide')
+                
+                st.plotly_chart(fig, use_container_width=True)
         
         # Display tabular results
         st.subheader("Detailed Results")
@@ -180,168 +188,174 @@ class ResultsDisplay:
         tab1, tab2 = st.tabs(["Cooling Load Details", "Heating Load Details"])
         
         with tab1:
-            # Create cooling load details table
-            cooling_details = []
-            
-            # Add envelope components
-            for wall in results["cooling"]["detailed_loads"]["walls"]:
-                cooling_details.append({
-                    "Component Type": "Wall",
-                    "Name": wall["name"],
-                    "Orientation": wall["orientation"],
-                    "Area (m²)": wall["area"],
-                    "U-Value (W/m²·K)": wall["u_value"],
-                    "CLTD (°C)": wall["cltd"],
-                    "Load (kW)": wall["load"]
-                })
-            
-            for roof in results["cooling"]["detailed_loads"]["roofs"]:
-                cooling_details.append({
-                    "Component Type": "Roof",
-                    "Name": roof["name"],
-                    "Orientation": roof["orientation"],
-                    "Area (m²)": roof["area"],
-                    "U-Value (W/m²·K)": roof["u_value"],
-                    "CLTD (°C)": roof["cltd"],
-                    "Load (kW)": roof["load"]
-                })
-            
-            for window in results["cooling"]["detailed_loads"]["windows"]:
-                cooling_details.append({
-                    "Component Type": "Window",
-                    "Name": window["name"],
-                    "Orientation": window["orientation"],
-                    "Area (m²)": window["area"],
-                    "U-Value (W/m²·K)": window["u_value"],
-                    "SHGC": window["shgc"],
-                    "SCL (W/m²)": window["scl"],
-                    "Load (kW)": window["load"]
-                })
-            
-            for door in results["cooling"]["detailed_loads"]["doors"]:
+            if not results.get("cooling") or "detailed_loads" not in results["cooling"]:
+                st.warning("Cooling load details are not available.")
+            else:
+                # Create cooling load details table
+                cooling_details = []
+                
+                # Add envelope components
+                for wall in results["cooling"]["detailed_loads"]["walls"]:
+                    cooling_details.append({
+                        "Component Type": "Wall",
+                        "Name": wall["name"],
+                        "Orientation": wall["orientation"],
+                        "Area (m²)": wall["area"],
+                        "U-Value (W/m²·K)": wall["u_value"],
+                        "CLTD (°C)": wall["cltd"],
+                        "Load (kW)": wall["load"]
+                    })
+                
+                for roof in results["cooling"]["detailed_loads"]["roofs"]:
+                    cooling_details.append({
+                        "Component Type": "Roof",
+                        "Name": roof["name"],
+                        "Orientation": roof["orientation"],
+                        "Area (m²)": roof["area"],
+                        "U-Value (W/m²·K)": roof["u_value"],
+                        "CLTD (°C)": roof["cltd"],
+                        "Load (kW)": roof["load"]
+                    })
+                
+                for window in results["cooling"]["detailed_loads"]["windows"]:
+                    cooling_details.append({
+                        "Component Type": "Window",
+                        "Name": window["name"],
+                        "Orientation": window["orientation"],
+                        "Area (m²)": window["area"],
+                        "U-Value (W/m²·K)": window["u_value"],
+                        "SHGC": window["shgc"],
+                        "SCL (W/m²)": window["scl"],
+                        "Load (kW)": window["load"]
+                    })
+                
+                for door in results["cooling"]["detailed_loads"]["doors"]:
+                    cooling_details.append({
+                        "Component Type": "Door",
+                        "Name": door["name"],
+                        "Orientation": door["orientation"],
+                        "Area (m²)": door["area"],
+                        "U-Value (W/m²·K)": door["u_value"],
+                        "CLTD (°C)": door["cltd"],
+                        "Load (kW)": door["load"]
+                    })
+                
+                # Add internal loads
+                for internal_load in results["cooling"]["detailed_loads"]["internal"]:
+                    cooling_details.append({
+                        "Component Type": internal_load["type"],
+                        "Name": internal_load["name"],
+                        "Quantity": internal_load["quantity"],
+                        "Heat Gain (W)": internal_load["heat_gain"],
+                        "CLF": internal_load["clf"],
+                        "Load (kW)": internal_load["load"]
+                    })
+                
+                # Add infiltration and ventilation
                 cooling_details.append({
-                    "Component Type": "Door",
-                    "Name": door["name"],
-                    "Orientation": door["orientation"],
-                    "Area (m²)": door["area"],
-                    "U-Value (W/m²·K)": door["u_value"],
-                    "CLTD (°C)": door["cltd"],
-                    "Load (kW)": door["load"]
+                    "Component Type": "Infiltration",
+                    "Name": "Air Infiltration",
+                    "Air Flow (m³/s)": results["cooling"]["detailed_loads"]["infiltration"]["air_flow"],
+                    "Sensible Load (kW)": results["cooling"]["detailed_loads"]["infiltration"]["sensible_load"],
+                    "Latent Load (kW)": results["cooling"]["detailed_loads"]["infiltration"]["latent_load"],
+                    "Load (kW)": results["cooling"]["detailed_loads"]["infiltration"]["total_load"]
                 })
-            
-            # Add internal loads
-            for internal_load in results["cooling"]["detailed_loads"]["internal"]:
+                
                 cooling_details.append({
-                    "Component Type": internal_load["type"],
-                    "Name": internal_load["name"],
-                    "Quantity": internal_load["quantity"],
-                    "Heat Gain (W)": internal_load["heat_gain"],
-                    "CLF": internal_load["clf"],
-                    "Load (kW)": internal_load["load"]
+                    "Component Type": "Ventilation",
+                    "Name": "Fresh Air",
+                    "Air Flow (m³/s)": results["cooling"]["detailed_loads"]["ventilation"]["air_flow"],
+                    "Sensible Load (kW)": results["cooling"]["detailed_loads"]["ventilation"]["sensible_load"],
+                    "Latent Load (kW)": results["cooling"]["detailed_loads"]["ventilation"]["latent_load"],
+                    "Load (kW)": results["cooling"]["detailed_loads"]["ventilation"]["total_load"]
                 })
-            
-            # Add infiltration and ventilation
-            cooling_details.append({
-                "Component Type": "Infiltration",
-                "Name": "Air Infiltration",
-                "Air Flow (m³/s)": results["cooling"]["detailed_loads"]["infiltration"]["air_flow"],
-                "Sensible Load (kW)": results["cooling"]["detailed_loads"]["infiltration"]["sensible_load"],
-                "Latent Load (kW)": results["cooling"]["detailed_loads"]["infiltration"]["latent_load"],
-                "Load (kW)": results["cooling"]["detailed_loads"]["infiltration"]["total_load"]
-            })
-            
-            cooling_details.append({
-                "Component Type": "Ventilation",
-                "Name": "Fresh Air",
-                "Air Flow (m³/s)": results["cooling"]["detailed_loads"]["ventilation"]["air_flow"],
-                "Sensible Load (kW)": results["cooling"]["detailed_loads"]["ventilation"]["sensible_load"],
-                "Latent Load (kW)": results["cooling"]["detailed_loads"]["ventilation"]["latent_load"],
-                "Load (kW)": results["cooling"]["detailed_loads"]["ventilation"]["total_load"]
-            })
-            
-            # Display cooling details table
-            cooling_df = pd.DataFrame(cooling_details)
-            st.dataframe(cooling_df, use_container_width=True)
+                
+                # Display cooling details table
+                cooling_df = pd.DataFrame(cooling_details)
+                st.dataframe(cooling_df, use_container_width=True)
         
         with tab2:
-            # Create heating load details table
-            heating_details = []
-            
-            # Add envelope components
-            for wall in results["heating"]["detailed_loads"]["walls"]:
-                heating_details.append({
-                    "Component Type": "Wall",
-                    "Name": wall["name"],
-                    "Orientation": wall["orientation"],
-                    "Area (m²)": wall["area"],
-                    "U-Value (W/m²·K)": wall["u_value"],
-                    "Temperature Difference (°C)": wall["delta_t"],
-                    "Load (kW)": wall["load"]
-                })
-            
-            for roof in results["heating"]["detailed_loads"]["roofs"]:
-                heating_details.append({
-                    "Component Type": "Roof",
-                    "Name": roof["name"],
-                    "Orientation": roof["orientation"],
-                    "Area (m²)": roof["area"],
-                    "U-Value (W/m²·K)": roof["u_value"],
-                    "Temperature Difference (°C)": roof["delta_t"],
-                    "Load (kW)": roof["load"]
-                })
-            
-            for floor in results["heating"]["detailed_loads"]["floors"]:
-                heating_details.append({
-                    "Component Type": "Floor",
-                    "Name": floor["name"],
-                    "Area (m²)": floor["area"],
-                    "U-Value (W/m²·K)": floor["u_value"],
-                    "Temperature Difference (°C)": floor["delta_t"],
-                    "Load (kW)": floor["load"]
-                })
-            
-            for window in results["heating"]["detailed_loads"]["windows"]:
+            if not results.get("heating") or "detailed_loads" not in results["heating"]:
+                st.warning("Heating load details are not available.")
+            else:
+                # Create heating load details table
+                heating_details = []
+                
+                # Add envelope components
+                for wall in results["heating"]["detailed_loads"]["walls"]:
+                    heating_details.append({
+                        "Component Type": "Wall",
+                        "Name": wall["name"],
+                        "Orientation": wall["orientation"],
+                        "Area (m²)": wall["area"],
+                        "U-Value (W/m²·K)": wall["u_value"],
+                        "Temperature Difference (°C)": wall["delta_t"],
+                        "Load (kW)": wall["load"]
+                    })
+                
+                for roof in results["heating"]["detailed_loads"]["roofs"]:
+                    heating_details.append({
+                        "Component Type": "Roof",
+                        "Name": roof["name"],
+                        "Orientation": roof["orientation"],
+                        "Area (m²)": roof["area"],
+                        "U-Value (W/m²·K)": roof["u_value"],
+                        "Temperature Difference (°C)": roof["delta_t"],
+                        "Load (kW)": roof["load"]
+                    })
+                
+                for floor in results["heating"]["detailed_loads"]["floors"]:
+                    heating_details.append({
+                        "Component Type": "Floor",
+                        "Name": floor["name"],
+                        "Area (m²)": floor["area"],
+                        "U-Value (W/m²·K)": floor["u_value"],
+                        "Temperature Difference (°C)": floor["delta_t"],
+                        "Load (kW)": floor["load"]
+                    })
+                
+                for window in results["heating"]["detailed_loads"]["windows"]:
+                    heating_details.append({
+                        "Component Type": "Window",
+                        "Name": window["name"],
+                        "Orientation": window["orientation"],
+                        "Area (m²)": window["area"],
+                        "U-Value (W/m²·K)": window["u_value"],
+                        "Temperature Difference (°C)": window["delta_t"],
+                        "Load (kW)": window["load"]
+                    })
+                
+                for door in results["heating"]["detailed_loads"]["doors"]:
+                    heating_details.append({
+                        "Component Type": "Door",
+                        "Name": door["name"],
+                        "Orientation": door["orientation"],
+                        "Area (m²)": door["area"],
+                        "U-Value (W/m²·K)": door["u_value"],
+                        "Temperature Difference (°C)": door["delta_t"],
+                        "Load (kW)": door["load"]
+                    })
+                
+                # Add infiltration and ventilation
                 heating_details.append({
-                    "Component Type": "Window",
-                    "Name": window["name"],
-                    "Orientation": window["orientation"],
-                    "Area (m²)": window["area"],
-                    "U-Value (W/m²·K)": window["u_value"],
-                    "Temperature Difference (°C)": wall["delta_t"],
-                    "Load (kW)": wall["load"]
+                    "Component Type": "Infiltration",
+                    "Name": "Air Infiltration",
+                    "Air Flow (m³/s)": results["heating"]["detailed_loads"]["infiltration"]["air_flow"],
+                    "Temperature Difference (°C)": results["heating"]["detailed_loads"]["infiltration"]["delta_t"],
+                    "Load (kW)": results["heating"]["detailed_loads"]["infiltration"]["load"]
                 })
-            
-            for door in results["heating"]["detailed_loads"]["doors"]:
+                
                 heating_details.append({
-                    "Component Type": "Door",
-                    "Name": door["name"],
-                    "Orientation": door["orientation"],
-                    "Area (m²)": door["area"],
-                    "U-Value (W/m²·K)": door["u_value"],
-                    "Temperature Difference (°C)": door["delta_t"],
-                    "Load (kW)": door["load"]
+                    "Component Type": "Ventilation",
+                    "Name": "Fresh Air",
+                    "Air Flow (m³/s)": results["heating"]["detailed_loads"]["ventilation"]["air_flow"],
+                    "Temperature Difference (°C)": results["heating"]["detailed_loads"]["ventilation"]["delta_t"],
+                    "Load (kW)": results["heating"]["detailed_loads"]["ventilation"]["load"]
                 })
-            
-            # Add infiltration and ventilation
-            heating_details.append({
-                "Component Type": "Infiltration",
-                "Name": "Air Infiltration",
-                "Air Flow (m³/s)": results["heating"]["detailed_loads"]["infiltration"]["air_flow"],
-                "Temperature Difference (°C)": results["heating"]["detailed_loads"]["infiltration"]["delta_t"],
-                "Load (kW)": results["heating"]["detailed_loads"]["infiltration"]["load"]
-            })
-            
-            heating_details.append({
-                "Component Type": "Ventilation",
-                "Name": "Fresh Air",
-                "Air Flow (m³/s)": results["heating"]["detailed_loads"]["ventilation"]["air_flow"],
-                "Temperature Difference (°C)": results["heating"]["detailed_loads"]["ventilation"]["delta_t"],
-                "Load (kW)": results["heating"]["detailed_loads"]["ventilation"]["load"]
-            })
-            
-            # Display heating details table
-            heating_df = pd.DataFrame(heating_details)
-            st.dataframe(heating_df, use_container_width=True)
+                
+                # Display heating details table
+                heating_df = pd.DataFrame(heating_details)
+                st.dataframe(heating_df, use_container_width=True)
         
         # Add download buttons for results
         st.subheader("Download Results")
@@ -349,39 +363,44 @@ class ResultsDisplay:
         col1, col2 = st.columns(2)
         
         with col1:
-            if st.button("Download Cooling Load Results (CSV)"):
-                cooling_csv = cooling_df.to_csv(index=False)
-                st.download_button(
-                    label="Download CSV",
-                    data=cooling_csv,
-                    file_name=f"cooling_load_results_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
-                    mime="text/csv"
-                )
+            if results.get("cooling") and "detailed_loads" in results["cooling"]:
+                if st.button("Download Cooling Load Results (CSV)"):
+                    cooling_csv = cooling_df.to_csv(index=False)
+                    st.download_button(
+                        label="Download CSV",
+                        data=cooling_csv,
+                        file_name=f"cooling_load_results_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
+                        mime="text/csv"
+                    )
         
         with col2:
-            if st.button("Download Heating Load Results (CSV)"):
-                heating_csv = heating_df.to_csv(index=False)
-                st.download_button(
-                    label="Download CSV",
-                    data=heating_csv,
-                    file_name=f"heating_load_results_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
-                    mime="text/csv"
-                )
+            if results.get("heating") and "detailed_loads" in results["heating"]:
+                if st.button("Download Heating Load Results (CSV)"):
+                    heating_csv = heating_df.to_csv(index=False)
+                    st.download_button(
+                        label="Download CSV",
+                        data=heating_csv,
+                        file_name=f"heating_load_results_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
+                        mime="text/csv"
+                    )
         
         # Add button to download full report
         if st.button("Generate Full Report (Excel)"):
-            # This would be implemented with the export functionality
             st.info("Excel report generation will be implemented in the Export module.")
     
     def _display_component_breakdown(self, session_state: Dict[str, Any]) -> None:
         """
-        Display component breakdown visualizationolden visualization.
+        Display component breakdown visualization.
         
         Args:
             session_state: Streamlit session state containing calculation results
         """
         st.subheader("Component Breakdown")
         
+        if not session_state["calculation_results"].get("cooling") and not session_state["calculation_results"].get("heating"):
+            st.warning("No component breakdown data available.")
+            return
+        
         # Use component visualization module
         self.component_visualization.display_component_breakdown(
             session_state["calculation_results"],
@@ -397,6 +416,10 @@ class ResultsDisplay:
         """
         st.subheader("Psychrometric Analysis")
         
+        if not session_state["calculation_results"].get("cooling"):
+            st.warning("Psychrometric analysis requires cooling load results.")
+            return
+        
         # Use psychrometric visualization module
         self.psychrometric_visualization.display_psychrometric_chart(
             session_state["calculation_results"],
@@ -412,6 +435,10 @@ class ResultsDisplay:
         """
         st.subheader("Time Analysis")
         
+        if not session_state["calculation_results"].get("cooling"):
+            st.warning("Time analysis requires cooling load results.")
+            return
+        
         # Use time-based visualization module
         self.time_based_visualization.display_time_analysis(
             session_state["calculation_results"]