Update app.py

app.py

@@ -1,82 +1,72 @@
+import os
 import numpy as np
 import cv2
 import gradio as gr
 from PIL import Image
 
-# Load the PCA model
+# Paths (Adjust these paths according to your environment)
 PCA_MODEL_PATH = "pca_texture_model.npy"
+
+# Load PCA model
 pca = np.load(PCA_MODEL_PATH, allow_pickle=True).item()
+TEXTURE_SIZE = int(np.sqrt(pca.mean_.shape[0] // 3))
 
 # PCA attributes
 mean_texture = pca.mean_
 components = pca.components_
 n_components = components.shape[0]
-TEXTURE_SIZE = np.sqrt(mean_texture.shape[0] // 3).astype(int)
 
-# Slider ranges based on PCA variance
+# Calculate slider ranges
 slider_ranges = [3 * np.sqrt(var) for var in pca.explained_variance_]
 
-# Function to reconstruct texture from PCA components
 def reconstruct_texture(component_values):
-    # Calculate the texture using PCA components
+    # Reconstruct the texture using the PCA components
     new_texture = mean_texture + np.dot(component_values, components)
-    new_texture = np.clip(new_texture, 0, 255).astype(np.uint8)  # Ensure valid pixel range
-    new_texture = new_texture.reshape((TEXTURE_SIZE, TEXTURE_SIZE, 3))  # Reshape back to image
-    new_texture = cv2.cvtColor(new_texture, cv2.COLOR_BGR2RGB)  # Convert to RGB for display
-    return Image.fromarray(new_texture)
-
-# Function to handle slider values and generate the texture
-def generate_texture(*component_values):
-    component_values = np.array(component_values)
-    return reconstruct_texture(component_values)
-
-# Function to generate random textures
-def random_texture():
-    # Generate random PCA component values
-    component_values = np.random.normal(0, 1, n_components)
-    # Generate the texture from the random PCA values
-    generated_image = reconstruct_texture(component_values)
-    # Return the image and each component value separately
-    return [generated_image] + list(component_values)
-
-# Gradio interface
-def create_app():
-    sliders = [
-        gr.Slider(
-            minimum=-slider_ranges[i],
-            maximum=slider_ranges[i],
-            step=0.1,
-            label=f"Component {i + 1}",
-            value=0,
-        )
-        for i in range(n_components)
-    ]
+    new_texture = np.clip(new_texture, 0, 255).astype(np.uint8)
+    new_texture = new_texture.reshape((TEXTURE_SIZE, TEXTURE_SIZE, 3))
+    new_texture = cv2.cvtColor(new_texture, cv2.COLOR_BGR2RGB)
+    image = Image.fromarray(new_texture)
+    return image
 
-    # Define interface elements
-    with gr.Blocks() as app:
+def main():
+    # Create a list of sliders
+    sliders = []
+    for i in range(n_components):
+        range_limit = slider_ranges[i]
+        sliders.append(gr.Slider(minimum=-range_limit, maximum=range_limit, step=10, label=f"Component {i+1}", value=0))
+    
+    with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
-                gr.Markdown("### PCA Texture Generator")
-                sliders_container = gr.Group(sliders)
-                generate_button = gr.Button("Generate")
-                random_button = gr.Button("Random")
-
+                # Create sliders
+                for slider in sliders:
+                    slider.render()
             with gr.Column():
-                preview = gr.Image(label="Preview")
-
-        # Link sliders and preview
-        generate_button.click(
-            generate_texture, inputs=sliders, outputs=preview
-        )
-
-        random_button.click(
-            random_texture, inputs=None, outputs=[preview] + sliders
-        )
+                image_output = gr.Image(label="Generated Texture", type="pil")
+                image_output.render()
+                randomize_button = gr.Button("Randomize")
+                randomize_button.render()
+        
+        # Update texture when sliders change
+        def on_slider_change(*args):
+            component_values = np.array([s.value for s in sliders])
+            image = reconstruct_texture(component_values)
+            image_output.update(value=image)
 
-    return app
+        for slider in sliders:
+            slider.change(fn=on_slider_change, inputs=None, outputs=None)
+        
+        # Randomize button
+        def on_randomize_click():
+            sampled_coefficients = np.random.normal(0, np.sqrt(pca.explained_variance_), size=n_components)
+            for i, slider in enumerate(sliders):
+                slider.update(value=sampled_coefficients[i])
+            image = reconstruct_texture(sampled_coefficients)
+            image_output.update(value=image)
 
+        randomize_button.click(fn=on_randomize_click, inputs=None, outputs=None)
+    
+    demo.launch()
 
 if __name__ == "__main__":
-    # Create and launch the app
-    app = create_app()
-    app.launch()
+    main()