working adding the demands

src/pages/potential_analysis.py

@@ -2074,147 +2074,194 @@ with button_col:
 # Initialize session state for selections if not exists
 if 'selected_items' not in st.session_state:
     st.session_state['selected_items'] = {}
+if 'current_heat_demand' not in st.session_state:
+    st.session_state['current_heat_demand'] = 0.0
+if 'current_cooling_demand' not in st.session_state:
+    st.session_state['current_cooling_demand'] = 0.0
 
-# Get processes from session state
-processes = st.session_state.get('processes', [])
-
-if not processes:
-    st.info("No processes found. Please add processes in the Data Collection page first.")
-else:
-    # Create side-by-side layout: streams on left, map on right
-    streams_col, map_col = st.columns([1, 1])
+# Helper function to determine stream type and extract data
+def get_stream_info(stream):
+    """Extract Tin, Tout, mdot, cp, CP from stream and determine if Hot stream (Heat Source) or Cold Stream (Heat Sink).
+    Calculate Q = CP * (Tout - Tin).
+    CP can be provided directly, or calculated as mdot * cp.
+    """
+    properties = stream.get('properties', {})
+    values = stream.get('values', {})
+    
+    # Also check stream_values (new structure)
+    stream_values = stream.get('stream_values', {})
+    if not stream_values:
+        stream_values = stream.get('product_values', {})
+    
+    tin = None
+    tout = None
+    mdot = None
+    cp_val = None
+    CP_direct = None  # CP provided directly
+    
+    # First try stream_values (new structure)
+    if stream_values:
+        if 'Tin' in stream_values and stream_values['Tin']:
+            try:
+                tin = float(stream_values['Tin'])
+            except (ValueError, TypeError):
+                pass
+        if 'Tout' in stream_values and stream_values['Tout']:
+            try:
+                tout = float(stream_values['Tout'])
+            except (ValueError, TypeError):
+                pass
+        if 'ṁ' in stream_values and stream_values['ṁ']:
+            try:
+                mdot = float(stream_values['ṁ'])
+            except (ValueError, TypeError):
+                pass
+        if 'cp' in stream_values and stream_values['cp']:
+            try:
+                cp_val = float(stream_values['cp'])
+            except (ValueError, TypeError):
+                pass
+        if 'CP' in stream_values and stream_values['CP']:
+            try:
+                CP_direct = float(stream_values['CP'])
+            except (ValueError, TypeError):
+                pass
     
-    # Helper function to determine stream type and extract data
-    def get_stream_info(stream):
-        """Extract Tin, Tout, mdot, cp, CP from stream and determine if Hot stream (Heat Source) or Cold Stream (Heat Sink).
-        Calculate Q = CP * (Tout - Tin).
-        CP can be provided directly, or calculated as mdot * cp.
-        """
-        properties = stream.get('properties', {})
-        values = stream.get('values', {})
-        
-        # Also check stream_values (new structure)
-        stream_values = stream.get('stream_values', {})
-        if not stream_values:
-            stream_values = stream.get('product_values', {})
-        
-        tin = None
-        tout = None
-        mdot = None
-        cp_val = None
-        CP_direct = None  # CP provided directly
-        
-        # First try stream_values (new structure)
-        if stream_values:
-            if 'Tin' in stream_values and stream_values['Tin']:
+    # Check properties dict structure
+    if isinstance(properties, dict) and isinstance(values, dict):
+        for pk, pname in properties.items():
+            vk = pk.replace('prop', 'val')
+            v = values.get(vk, '')
+            
+            if pname == 'Tin' and v and tin is None:
                 try:
-                    tin = float(stream_values['Tin'])
+                    tin = float(v)
                 except (ValueError, TypeError):
                     pass
-            if 'Tout' in stream_values and stream_values['Tout']:
+            elif pname == 'Tout' and v and tout is None:
                 try:
-                    tout = float(stream_values['Tout'])
+                    tout = float(v)
                 except (ValueError, TypeError):
                     pass
-            if 'ṁ' in stream_values and stream_values['ṁ']:
+            elif pname == 'ṁ' and v and mdot is None:
                 try:
-                    mdot = float(stream_values['ṁ'])
+                    mdot = float(v)
                 except (ValueError, TypeError):
                     pass
-            if 'cp' in stream_values and stream_values['cp']:
+            elif pname == 'cp' and v and cp_val is None:
                 try:
-                    cp_val = float(stream_values['cp'])
+                    cp_val = float(v)
                 except (ValueError, TypeError):
                     pass
-            if 'CP' in stream_values and stream_values['CP']:
+            elif pname == 'CP' and v and CP_direct is None:
                 try:
-                    CP_direct = float(stream_values['CP'])
+                    CP_direct = float(v)
                 except (ValueError, TypeError):
                     pass
+    
+    # Fallback to legacy fields
+    if tin is None and stream.get('temp_in'):
+        try:
+            tin = float(stream['temp_in'])
+        except (ValueError, TypeError):
+            pass
+    if tout is None and stream.get('temp_out'):
+        try:
+            tout = float(stream['temp_out'])
+        except (ValueError, TypeError):
+            pass
+    if mdot is None and stream.get('mdot'):
+        try:
+            mdot = float(stream['mdot'])
+        except (ValueError, TypeError):
+            pass
+    if cp_val is None and stream.get('cp'):
+        try:
+            cp_val = float(stream['cp'])
+        except (ValueError, TypeError):
+            pass
+    
+    # Determine stream type
+    stream_type = None
+    if tin is not None and tout is not None:
+        if tin > tout:
+            stream_type = "Hot stream (Heat Source)"
+        else:
+            stream_type = "Cold stream (Heat sink)"
+    
+    # Determine CP: use direct CP if provided, otherwise calculate from mdot * cp
+    CP_flow = None
+    if CP_direct is not None:
+        CP_flow = CP_direct
+    elif mdot is not None and cp_val is not None:
+        CP_flow = mdot * cp_val
+    
+    # Calculate Q = CP * |Tout - Tin| (always positive)
+    Q = None
+    if CP_flow is not None and tin is not None and tout is not None:
+        Q = abs(CP_flow * (tout - tin))
+    
+    return {
+        'tin': tin,
+        'tout': tout,
+        'mdot': mdot,
+        'cp': cp_val,
+        'CP': CP_flow,
+        'Q': Q,
+        'type': stream_type
+    }
+
+# Get processes from session state
+processes = st.session_state.get('processes', [])
+
+# Current Demands Section
+with st.expander("Current Energy Demands (They can be added to calculate the improvement with respect to the status quo)", expanded=False):
+    heat_demand = st.number_input("Heat Demand (kW)", min_value=0.0, step=0.1, value=st.session_state.get('current_heat_demand', 0.0))
+    cooling_demand = st.number_input("Cooling Demand (kW)", min_value=0.0, step=0.1, value=st.session_state.get('current_cooling_demand', 0.0))
+    
+    # Create stream options
+    stream_options = []
+    for idx, process in enumerate(processes):
+        for stream_idx, stream in enumerate(process.get('streams', [])):
+            stream_name = stream.get('name', f'Stream {stream_idx + 1} in {process.get('name', f'Subprocess {idx + 1}')}')
+            stream_options.append((idx, stream_idx, stream_name))
+    
+    selected_stream_names = st.multiselect("Streams", [name for _, _, name in stream_options], default=[name for _, _, name in stream_options])
+    
+    if st.button("Apply Demands"):
+        st.session_state['current_heat_demand'] = heat_demand
+        st.session_state['current_cooling_demand'] = cooling_demand
         
-        # Check properties dict structure
-        if isinstance(properties, dict) and isinstance(values, dict):
-            for pk, pname in properties.items():
-                vk = pk.replace('prop', 'val')
-                v = values.get(vk, '')
-                
-                if pname == 'Tin' and v and tin is None:
-                    try:
-                        tin = float(v)
-                    except (ValueError, TypeError):
-                        pass
-                elif pname == 'Tout' and v and tout is None:
-                    try:
-                        tout = float(v)
-                    except (ValueError, TypeError):
-                        pass
-                elif pname == 'ṁ' and v and mdot is None:
-                    try:
-                        mdot = float(v)
-                    except (ValueError, TypeError):
-                        pass
-                elif pname == 'cp' and v and cp_val is None:
-                    try:
-                        cp_val = float(v)
-                    except (ValueError, TypeError):
-                        pass
-                elif pname == 'CP' and v and CP_direct is None:
-                    try:
-                        CP_direct = float(v)
-                    except (ValueError, TypeError):
-                        pass
-        
-        # Fallback to legacy fields
-        if tin is None and stream.get('temp_in'):
-            try:
-                tin = float(stream['temp_in'])
-            except (ValueError, TypeError):
-                pass
-        if tout is None and stream.get('temp_out'):
-            try:
-                tout = float(stream['temp_out'])
-            except (ValueError, TypeError):
-                pass
-        if mdot is None and stream.get('mdot'):
-            try:
-                mdot = float(stream['mdot'])
-            except (ValueError, TypeError):
-                pass
-        if cp_val is None and stream.get('cp'):
-            try:
-                cp_val = float(stream['cp'])
-            except (ValueError, TypeError):
-                pass
-        
-        # Determine stream type
-        stream_type = None
-        if tin is not None and tout is not None:
-            if tin > tout:
-                stream_type = "Hot stream (Heat Source)"
-            else:
-                stream_type = "Cold stream (Heat sink)"
+        # Get selected stream indices
+        selected_stream_indices = [(idx, sidx) for idx, sidx, name in stream_options if name in selected_stream_names]
         
-        # Determine CP: use direct CP if provided, otherwise calculate from mdot * cp
-        CP_flow = None
-        if CP_direct is not None:
-            CP_flow = CP_direct
-        elif mdot is not None and cp_val is not None:
-            CP_flow = mdot * cp_val
+        # For cooling demand, select all hot streams from selected streams
+        if cooling_demand > 0:
+            for proc_idx, stream_idx in selected_stream_indices:
+                process = processes[proc_idx]
+                stream = process.get('streams', [])[stream_idx]
+                info = get_stream_info(stream)
+                if info['type'] == "Hot stream (Heat Source)":
+                    stream_key = f"stream_{proc_idx}_{stream_idx}"
+                    st.session_state['selected_items'][stream_key] = True
         
-        # Calculate Q = CP * |Tout - Tin| (always positive)
-        Q = None
-        if CP_flow is not None and tin is not None and tout is not None:
-            Q = abs(CP_flow * (tout - tin))
+        # For heating demand, select all cold streams from selected streams
+        if heat_demand > 0:
+            for proc_idx, stream_idx in selected_stream_indices:
+                process = processes[proc_idx]
+                stream = process.get('streams', [])[stream_idx]
+                info = get_stream_info(stream)
+                if info['type'] == "Cold stream (Heat sink)":
+                    stream_key = f"stream_{proc_idx}_{stream_idx}"
+                    st.session_state['selected_items'][stream_key] = True
         
-        return {
-            'tin': tin,
-            'tout': tout,
-            'mdot': mdot,
-            'cp': cp_val,
-            'CP': CP_flow,
-            'Q': Q,
-            'type': stream_type
-        }
+        st.success("Demands applied and streams selected.")
+
+if not processes:
+    st.info("No processes found. Please add processes in the Data Collection page first.")
+else:
+    # Create side-by-side layout: streams on left, map on right
+    streams_col, map_col = st.columns([1, 1])
     
     # Left column: Display streams with selection checkboxes
     with streams_col: