Update VULCAN/mcp_output/mcp_plugin/mcp_service.py

VULCAN/mcp_output/mcp_plugin/mcp_service.py

@@ -246,12 +246,32 @@ def tool_build_atmospheric_structure(dataset: str = 'HD189') -> dict:
                 if data.ndim == 1:
                     data = data.reshape(1, -1)
                 
-                if data.shape[0] >= nz:
-                    if data.shape[1] >= 2:
-                        pressure_data = data[:, 0]  # Pressure column
-                        Tco = data[:nz, 1]  # Temperature column (2nd column)
-                    if data.shape[1] >= 3:
-                        Kzz = data[:nz, 2]  # Kzz column (3rd column)
+                n_data = data.shape[0]
+                
+                # Read columns: pressure, temperature, Kzz
+                if data.shape[1] >= 2:
+                    pressure_read = data[:, 0]  # Pressure column
+                    Tco_read = data[:, 1]  # Temperature column (2nd column)
+                    
+                    # If data size differs from nz, interpolate
+                    if n_data != nz:
+                        # Create interpolation indices
+                        indices_old = np.linspace(0, n_data - 1, n_data)
+                        indices_new = np.linspace(0, n_data - 1, nz)
+                        Tco = np.interp(indices_new, indices_old, Tco_read)
+                        pressure_data = np.interp(indices_new, indices_old, pressure_read)
+                    else:
+                        Tco = Tco_read
+                        pressure_data = pressure_read
+                
+                if data.shape[1] >= 3:
+                    Kzz_read = data[:, 2]  # Kzz column (3rd column)
+                    if n_data != nz:
+                        indices_old = np.linspace(0, n_data - 1, n_data)
+                        indices_new = np.linspace(0, n_data - 1, nz)
+                        Kzz = np.interp(indices_new, indices_old, Kzz_read)
+                    else:
+                        Kzz = Kzz_read
             except Exception as read_err:
                 pass  # Use defaults on read error