Update app.py

app.py

@@ -8,7 +8,7 @@ def random_hex():
 
 # Generate initial 5-color palette
 def generate_palette(locked_colors):
-    new_palette = [random_hex() if not locked else locked for locked in locked_colors]
+    new_palette = [random_hex() if locked is None else locked for locked in locked_colors]
     return new_palette
 
 # Create color swatches
@@ -27,7 +27,7 @@ initial_palette = generate_palette([None] * 5)
 def update_palette(*locks):
     locks = [color if color else None for color in locks]
     new_palette = generate_palette(locks)
-    return display_palette(new_palette)
+    return display_palette(new_palette) + tuple(new_palette)
 
 with gr.Blocks(theme=gr.themes.Soft()) as iface:
     gr.Markdown("# 🎨 AI-Powered Color Palette Generator")
@@ -39,11 +39,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     palette_output = gr.Plot(label="Generated Palette")
     generate_button = gr.Button("Regenerate Palette 🎨")
     
-    generate_button.click(update_palette, inputs=lock_inputs, outputs=[palette_output, lock_inputs])
+    generate_button.click(update_palette, inputs=lock_inputs, outputs=[palette_output] + lock_inputs)
     
     initial_fig, _ = display_palette(initial_palette)
     palette_output.value = initial_fig
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()