Update app.py

app.py

@@ -30,18 +30,15 @@ def generate_ai_palette():
     return [rgb_to_hex([random.randint(0, 255) for _ in range(3)]) for _ in range(5)]
 
 # Function to update palette with user selection
-def update_palette(image, selected_x, selected_y, locked_colors):
+def update_palette(image, locked_colors):
     if image is not None:
-        selected_color = image[int(selected_y), int(selected_x)]
-        selected_hex = rgb_to_hex(selected_color)
-        
         colors = extract_colors(image, 5)
         hex_colors = [rgb_to_hex(color) for color in colors]
         
         # Replace unlocked colors with extracted ones
         for i in range(5):
             if locked_colors[i] == "":
-                locked_colors[i] = selected_hex if i == 0 else hex_colors[i]
+                locked_colors[i] = hex_colors[i]
     else:
         # If no image, generate AI colors
         locked_colors = [color if color != "" else rgb_to_hex([random.randint(0, 255) for _ in range(3)]) for color in locked_colors]
@@ -70,13 +67,18 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     hex_output = gr.Textbox(label="Hex Codes", interactive=True)
     
     with gr.Row():
-        locked_colors = [gr.Textbox(label=f"Color {i+1}", interactive=True) for i in range(5)]
+        locked_colors = [gr.Textbox(label=f"Color {i+1}", interactive=True, value="") for i in range(5)]
+    
+    # Initialize with a random palette on startup
+    initial_palette = generate_ai_palette()
+    initial_fig, initial_colors = update_palette(None, initial_palette)
+    palette_output.value = initial_fig
+    for i in range(5):
+        locked_colors[i].value = initial_colors[i]
     
     generate_ai_button.click(generate_ai_palette, inputs=[], outputs=locked_colors)
     regenerate_button.click(update_palette, inputs=[image_input, locked_colors], outputs=[palette_output, locked_colors])
-    
-    image_input.select(update_palette, inputs=[image_input, image_input.select, image_input.select, locked_colors], outputs=[palette_output, locked_colors])
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()