image input fix, button try to fix

app.py

@@ -11,7 +11,7 @@ import base64
 hf_token = os.environ.get("HF_TOKEN_API_DEMO") # we get it from a secret env variable, such that it's private
 auth_headers = {"api_token": hf_token}
 
-def convert_mask_image_to_base64_string(mask_image):
+def convert_image_to_base64_string(mask_image):
     buffer = io.BytesIO()
     mask_image.save(buffer, format="PNG")  # You can choose the format (e.g., "JPEG", "PNG")
     # Encode the buffer in base64
@@ -48,7 +48,7 @@ def predict_ref_by_text(input_image, prompt):
     # init_image = Image.fromarray(dict['background'][:, :, :3], 'RGB') #dict['background'].convert("RGB")#.resize((1024, 1024))
     # mask = Image.fromarray(dict['layers'][0][:,:,3], 'L') #dict['layers'].convert("RGB")#.resize((1024, 1024))
     
-    image_base64_file = convert_mask_image_to_base64_string(input_image)
+    image_base64_file = convert_image_to_base64_string(input_image)
     
     gen_img = lifestyle_shot_by_text_api_call(image_base64_file, prompt)
     
@@ -77,13 +77,15 @@ def lifestyle_shot_by_image_api_call(image_base64_file, ref_image_base64_file):
 
 def predict_ref_by_image(init_image, ref_image):
     
-    image_base64_file = convert_mask_image_to_base64_string(init_image)
-    ref_base64_file = convert_mask_image_to_base64_string(ref_image)
+    image_base64_file = convert_image_to_base64_string(init_image)
+    ref_base64_file = convert_image_to_base64_string(ref_image)
     
     gen_img = lifestyle_shot_by_image_api_call(image_base64_file, ref_base64_file)
     
     return gen_img
 
+def on_change_prompt(img: Image.Image | None, prompt: str | None):
+    return gr.update(interactive=bool(img and prompt))
 
 css = '''
 .gradio-container{max-width: 1100px !important}
@@ -151,9 +153,10 @@ with image_blocks as demo:
 
         with gr.Row():
             with gr.Column():
-                image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
-                                    brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
-                                    )
+                # image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
+                                    # brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
+                                    # )
+                image = gr.Image(type="pil", label="Input")
                 prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
                 with gr.Row(elem_id="prompt-container", equal_height=True):
                     with gr.Column():
@@ -163,15 +166,22 @@ with image_blocks as demo:
                 image_out = gr.Image(label="Output", elem_id="output-img")
 
         # Button click will trigger the inpainting function (now with prompt included)
+        for inp in [image, prompt]:
+            inp.change(
+                fn=on_change_prompt,
+                inputs=[image, prompt],
+                outputs=[btn],
+            )
         btn.click(fn=predict_ref_by_text, inputs=[image, prompt], outputs=[image_out], api_name='run')
     
     with gr.Tab(label="By image reference", id="tab_ref_image"):
 
         with gr.Row():
             with gr.Column():
-                image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
-                                    brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
-                                    )
+                # image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
+                                    # brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
+                                    # )
+                image = gr.Image(type="pil", label="Input")
                 ref_image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
                                     brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
                                     )