Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import pandas as pd
 import numpy as np
@@ -154,6 +153,19 @@ class AdvancedGridOptimizer:
         x_pos = 0
         for i, (width, lot_type) in enumerate(solution):
             # Get base color
+            if width in colors:
+                base_color = colors[width]
+            else:
+                # Use MD colors if in MD mode
+                if self.development_mode == 'medium_density':
+                    if width <= 6.0:
+                        base_color = colors.get(4.5, '#6B4C8A')
+                    else:
+                        base_color = colors.get(7.0, '#4A7C7E')
+                else:
+                    closest_width = min(colors.keys(), key=lambda x: abs(x - width))
+                    base_color = colors[closest_width]
+            # Get base color
         ax1.set_xlim(-5, stage_width + 5)
         # Adjust y-limits for rear laneway
         if self.md_load_type == 'rear':