Update app.py

app.py

@@ -50,6 +50,15 @@ def on_random_click():
     updates.append(texture)
     return updates
 
+def on_image_upload(image):
+    # Process the uploaded image and get PCA coefficients
+    coefficients = process_uploaded_image(image)
+    texture = generate_texture(*coefficients)
+    # Update sliders and image
+    updates = [gr.update(value=value) for value in coefficients]
+    updates.append(texture)
+    return updates
+
 # Create Gradio interface
 with gr.Blocks() as demo:
     with gr.Row():
@@ -65,11 +74,16 @@ with gr.Blocks() as demo:
                     label=component_names[i]
                 )
                 sliders.append(slider)
-            random_button = gr.Button("Randomize Texture")
         with gr.Column():
             output_image = gr.Image(
                 label="Generated Texture"
             )
+            random_button = gr.Button("Randomize Texture")
+            upload_image = gr.Image(
+                label="Upload Image",
+                tool="editor",
+                type="numpy"
+            )
     
     # Update texture when any slider changes
     for slider in sliders:
@@ -86,4 +100,11 @@ with gr.Blocks() as demo:
         outputs=[*sliders, output_image]
     )
 
+    # Update sliders and image based on uploaded image
+    upload_image.change(
+        fn=on_image_upload,
+        inputs=upload_image,
+        outputs=[*sliders, output_image]
+    )
+
 demo.launch()