Update app.py

app.py

@@ -8,8 +8,7 @@ def random_hex():
 
 # Generate initial 5-color palette
 def generate_palette(locked_colors):
-    new_palette = [random_hex() if locked is None else locked for locked in locked_colors]
-    return new_palette
+    return [random_hex() if locked is None else locked for locked in locked_colors]
 
 # Create color swatches
 def display_palette(colors):
@@ -25,16 +24,16 @@ initial_palette = generate_palette([None] * 5)
 
 # Gradio UI
 def update_palette(*locks):
-    locks = [color if color else None for color in locks]
+    locks = [color if color and color != "None" else None for color in locks]
     new_palette = generate_palette(locks)
     return display_palette(new_palette) + tuple(new_palette)
 
 with gr.Blocks(theme=gr.themes.Soft()) as iface:
     gr.Markdown("# 🎨 AI-Powered Color Palette Generator")
-    gr.Markdown("Click to lock colors and generate fresh palettes!")
+    gr.Markdown("Click on a color to lock it. Generate fresh palettes for unlocked colors!")
     
     with gr.Row():
-        lock_inputs = [gr.Textbox(label=f"Color {i+1}", interactive=True, value=initial_palette[i]) for i in range(5)]
+        lock_inputs = [gr.ColorPicker(label=f"Color {i+1}", value=initial_palette[i]) for i in range(5)]
     
     palette_output = gr.Plot(label="Generated Palette")
     generate_button = gr.Button("Regenerate Palette 🎨")
@@ -46,4 +45,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()