Add more LUT interaction

app.py

@@ -119,6 +119,7 @@ from utils.version_info import (
 
 input_image_palette = []
 current_prerendered_image = gr.State("./images/images/Bee-test-2.png")
+current_lut_example_img = gr.State(constants.default_lut_example_img)
 user_dir = constants.TMPDIR
 lora_models = get_lora_models()
 selected_index = gr.State(value=-1)
@@ -768,9 +769,11 @@ def composite_with_control_sync(input_image, sketch_image, slider_value):
     result_img = alpha_composite_with_control(in_img, sk_img, slider_value)
     return result_img
 
-def replace_input_with_sketch_image(sketch_image):
+def replace_with_sketch_image(sketch_image, replace_current_lut_example_img: bool = False):
     print(f"Sketch Image: {sketch_image}\n")
     sketch, is_dict = get_image_from_dict(sketch_image)
+    if replace_current_lut_example_img:
+        current_lut_example_img = sketch
     return sketch
 ####################################### DEPTH ESTIMATION #######################################
 
@@ -1193,7 +1196,7 @@ with gr.Blocks(css_paths="style_20250314.css", title=title, theme='Surn/beeuty',
                     with gr.Column(scale=1):
                         with gr.Row():
                             sketch_replace_input_image_button = gr.Button("Replace Input Image with sketch", elem_id="sketch_replace_input_image_button", elem_classes="solid")
-                            sketch_replace_lut_example_image_button = gr.Button("Replace Filter Example Image with sketch", elem_id="sketch_replace_lut_example_image_button", elem_classes="solid")
+                            #sketch_replace_lut_example_image_button = gr.Button("Replace Filter Example Image with sketch", elem_id="sketch_replace_lut_example_image_button", elem_classes="solid")
                         with gr.Row():
                             sketch_alpha_composite_slider = gr.Slider(0,100,50,0.5, label="Sketch Transparancy", elem_id="alpha_composite_slider")
                             btn_sketch_alpha_composite = gr.Button("Overlay Sketch on Input Image", elem_id="btn_sketchninput", elem_classes="solid")
@@ -1246,6 +1249,12 @@ with gr.Blocks(css_paths="style_20250314.css", title=title, theme='Surn/beeuty',
                     outputs=[input_image], 
                     scroll_to_output=True
                 )
+                apply_lut_to_sketch_button.click(
+                    lambda lut_filename, sketch_image, lut_intensity: gr.Warning("Please upload an Input Image to get started.") if input_image is None else apply_lut_to_image_path(lut_filename, replace_with_sketch_image(sketch_image, False), lut_intensity)[1], 
+                    inputs=[lut_filename, sketch_image, lut_intensity], 
+                    outputs=[sketch_image], 
+                    scroll_to_output=True
+                )
             with gr.Accordion("Color Composite", open = False):
                 with gr.Row():
                     composite_color = gr.ColorPicker(label="Color", value="#ede9ac44")
@@ -1537,10 +1546,15 @@ with gr.Blocks(css_paths="style_20250314.css", title=title, theme='Surn/beeuty',
         scroll_to_output=True
     )
     sketch_replace_input_image_button.click(
-        lambda sketch_image: replace_input_with_sketch_image(sketch_image),
+        lambda sketch_image: replace_with_sketch_image(sketch_image, False),
         inputs=[sketch_image],
         outputs=[input_image], scroll_to_output=True
     )
+    # sketch_replace_lut_example_image_button.click(
+    #     lambda sketch_image: replace_with_sketch_image(sketch_image, True),
+    #     inputs=[sketch_image],
+    #     outputs=[lut_example_image], scroll_to_output=True
+    # )
     ##################### model #######################################
     model_textbox.change(
         fn=update_prompt_notes,