Update app.py

app.py

@@ -1,8 +1,6 @@
 import numpy as np
 import cv2
 import gradio as gr
-import asyncio
-import time
 
 PCA_MODEL_PATH = "pca_texture_model.npy"
 COMPONENT_NAMES_PATH = "component_names.txt"
@@ -22,17 +20,13 @@ 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"{line.strip()} (Component {i + 1})" if line.strip() else f"Component {i + 1}"
-            for i, line in enumerate(f.readlines())
-        ]
+        component_names = [f"Component {i+1} ({line.strip()})" 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)
@@ -40,37 +34,24 @@ 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
 
-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):
+def on_random_click():
     random_values = randomize_texture()
-    state["last_update"] = time.time()
-    texture = generate_texture(random_values)
+    texture = generate_texture(*random_values)
     # Prepare updates for sliders and the image
-    slider_updates = [gr.Slider.update(value=value) for value in random_values]
-    return (*slider_updates, texture)
-
+    updates = [gr.update(value=value) for value in random_values]
+    updates.append(texture)
+    return updates
 
 # 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 = []
@@ -81,27 +62,27 @@ with gr.Blocks() as demo:
                     maximum=range_limit,
                     step=10,
                     value=0,
-                    label=f"{component_names[i]}"
+                    label=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"
             )
-
-    # Update texture when any slider changes with debouncing
+    
+    # Update texture when any slider changes
     for slider in sliders:
         slider.change(
-            fn=debounced_update_texture,
-            inputs=[state] + sliders,
-            outputs=output_image,
+            fn=update_texture,
+            inputs=sliders,
+            outputs=output_image
         )
 
-    # Randomize texture and update sliders and image with debouncing
+    # Randomize texture and update sliders and image
     random_button.click(
         fn=on_random_click,
-        inputs=[state],
+        inputs=None,
         outputs=[*sliders, output_image]
     )