Update app.py

app.py

@@ -3,6 +3,7 @@ import cv2
 import gradio as gr
 
 PCA_MODEL_PATH = "pca_texture_model.npy"
+COMPONENT_NAMES_PATH = "component_names.txt"
 
 # Load PCA model
 pca = np.load(PCA_MODEL_PATH, allow_pickle=True).item()
@@ -16,6 +17,15 @@ TEXTURE_SIZE = int(np.sqrt(mean_texture.shape[0] // 3))
 # Calculate slider ranges
 slider_ranges = [3 * np.sqrt(var) for var in explained_variance]
 
+# Load component names if available
+try:
+    with open(COMPONENT_NAMES_PATH, "r") as f:
+        component_names = [f"Component {i+1} ({line.strip()})" if line.strip() else f"Component {i+1}" for line in f.readlines()]
+    if len(component_names) < n_components:
+        component_names += [f"Component {i+1}" for i in range(len(component_names), n_components)]
+except FileNotFoundError:
+    component_names = [f"Component {i+1}" for i in range(n_components)]
+
 def generate_texture(*component_values):
     component_values = np.array(component_values)
     new_texture = mean_texture + np.dot(component_values, components)
@@ -52,7 +62,7 @@ with gr.Blocks() as demo:
                     maximum=range_limit,
                     step=10,
                     value=0,
-                    label=f"Component {i+1}"
+                    label=component_names[i]
                 )
                 sliders.append(slider)
             random_button = gr.Button("Randomize Texture")