Update app.py

app.py

@@ -14,6 +14,29 @@ step_wl = 0.01
 wavelengths = np.arange(start_wl, stop_wl + step_wl, step_wl)
 materials = ['Si', 'Si3N4', 'SiO2', 'AlN']
 
+@tool
+def simulate_spectrum_10nm(layer_order: List[str]) -> List[float]:
+    """
+    Simulates the optical transmission spectrum for a given sequence of material layers at 10nm thickness.
+
+    Args:
+        layer_order (List[str]): A list of material names (e.g., ["Si", "SiO2", "AlN"]) representing the order of layers in the optical stack.
+
+    Returns:
+        List[float]: The transmission spectrum across a predefined wavelength range.
+    """
+    source = Source(wavelength=start_wl)
+    reflection_layer = Layer(n=1.0)
+    transmission_layer = Layer(material=Material("Si"))
+    try:
+        layers = [Layer(material=Material(m), thickness=0.01) for m in layer_order]
+        stack = LayerStack(*layers, incident_layer=reflection_layer, transmission_layer=transmission_layer)
+        solver = Solver(stack, source, (1, 1))
+        result = solver.solve(wavelength=wavelengths)
+        return np.array(result['TTot']).tolist()
+    except Exception as e:
+        return []
+
 @tool
 def simulate_spectrum_100nm(layer_order: List[str]) -> List[float]:
     """
@@ -52,7 +75,6 @@ def cosine_similarity(vec1: List[float], vec2: List[float]) -> float:
     a, b = np.array(vec1), np.array(vec2)
     return float(np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b)))
 
-
 # --- Target Spectrum Generator ---
 def get_target_spectrum(layer_order, thickness=0.1):
     source = Source(wavelength=start_wl)