Update app.py

app.py

@@ -1,9 +1,12 @@
 import numpy as np
 import cv2
 import gradio as gr
+import asyncio
+import time
 
+# Paths to your PCA model and component names file
 PCA_MODEL_PATH = "pca_texture_model.npy"
-COMPONENT_NAMES_PATH = "component_names.txt"
+COMPONENT_NAMES_PATH = "component_names.txt"  # File containing component names (one name per line)
 
 # Load PCA model
 pca = np.load(PCA_MODEL_PATH, allow_pickle=True).item()
@@ -20,13 +23,17 @@ 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 i, line in enumerate(f.readlines())]
+        component_names = [
+            f"{line.strip()} (Component {i + 1})" if line.strip() else f"Component {i + 1}"
+            for i, line in enumerate(f.readlines())
+        ]
     if len(component_names) < n_components:
-        component_names += [f"Component {i+1}" for i in range(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)]
+    component_names = [f"Component {i + 1}" for i in range(n_components)]
 
-def generate_texture(*component_values):
+
+def generate_texture(component_values):
     component_values = np.array(component_values)
     new_texture = mean_texture + np.dot(component_values, components)
     new_texture = np.clip(new_texture, 0, 255).astype(np.uint8)
@@ -34,24 +41,37 @@ def generate_texture(*component_values):
     new_texture = cv2.cvtColor(new_texture, cv2.COLOR_BGR2RGB)
     return new_texture
 
+
 def randomize_texture():
     sampled_coefficients = np.random.normal(0, np.sqrt(explained_variance), size=n_components)
     return sampled_coefficients.tolist()
 
-def update_texture(*component_values):
-    texture = generate_texture(*component_values)
-    return texture
 
-def on_random_click():
+async def debounced_update_texture(state, *component_values):
+    current_time = time.time()
+    state["last_update"] = current_time
+    await asyncio.sleep(0.2)  # Debounce delay in seconds
+    if state["last_update"] == current_time:
+        texture = generate_texture(component_values)
+        return texture
+    else:
+        return gr.update()
+
+
+async def on_random_click(state):
     random_values = randomize_texture()
-    texture = generate_texture(*random_values)
+    state["last_update"] = time.time()
+    texture = generate_texture(random_values)
     # Prepare updates for sliders and the image
-    updates = [gr.update(value=value) for value in random_values]
-    updates.append(texture)
-    return updates
+    slider_updates = [gr.Slider.update(value=value) for value in random_values]
+    return (*slider_updates, texture)
+
 
 # Create Gradio interface
 with gr.Blocks() as demo:
+    # Initialize a state to keep track of the last update time
+    state = gr.State({"last_update": 0})
+
     with gr.Row():
         with gr.Column():
             sliders = []
@@ -62,27 +82,29 @@ with gr.Blocks() as demo:
                     maximum=range_limit,
                     step=10,
                     value=0,
-                    label=component_names[i]
+                    label=f"{component_names[i]}"
                 )
                 sliders.append(slider)
             random_button = gr.Button("Randomize Texture")
         with gr.Column():
             output_image = gr.Image(
-                label="Generated Texture"
+                label="Generated Texture",
+                shape=(TEXTURE_SIZE, TEXTURE_SIZE)
             )
-    
-    # Update texture when any slider changes
+
+    # Update texture when any slider changes with debouncing
     for slider in sliders:
         slider.change(
-            fn=update_texture,
-            inputs=sliders,
-            outputs=output_image
+            fn=debounced_update_texture,
+            inputs=[state] + sliders,
+            outputs=output_image,
+            _js=None  # No JavaScript, handled in Python
         )
 
-    # Randomize texture and update sliders and image
+    # Randomize texture and update sliders and image with debouncing
     random_button.click(
         fn=on_random_click,
-        inputs=None,
+        inputs=[state],
         outputs=[*sliders, output_image]
     )