No image editor

app.py

@@ -385,31 +385,6 @@ def process(input_image, prompt,
         default_height, default_width = 640, 640
         input_image = np.ones((default_height, default_width, 3), dtype=np.uint8) * 255
         print("No input image provided. Using a blank white image.")
-    else:
-        composite_rgba_uint8 = input_image["composite"] 
-
-        # rgb_uint8 will be (H, W, 3), dtype uint8
-        rgb_uint8 = composite_rgba_uint8[:, :, :3]
-        # mask_uint8 will be (H, W), dtype uint8
-        mask_uint8 = composite_rgba_uint8[:, :, 3]
-    
-        # Create background
-        h, w = rgb_uint8.shape[:2]
-        # White background, (H, W, 3), dtype uint8
-        background_uint8 = np.full((h, w, 3), 255, dtype=np.uint8) 
-        
-        # Normalize mask to range [0.0, 1.0]. 
-        alpha_normalized_float32 = mask_uint8.astype(np.float32) / 255.0
-        
-        # Expand alpha to 3 channels to match RGB images for broadcasting.
-        # alpha_mask_float32 will have shape (H, W, 3)
-        alpha_mask_float32 = np.stack([alpha_normalized_float32] * 3, axis=2)
-        
-        # alpha blending
-        blended_image_float32 = rgb_uint8.astype(np.float32) * alpha_mask_float32 + \
-                                background_uint8.astype(np.float32) * (1.0 - alpha_mask_float32)
-
-        input_image = np.clip(blended_image_float32, 0, 255).astype(np.uint8)
     
     yield None, None, '', '', gr.update(interactive=False), gr.update(interactive=True)
 
@@ -454,7 +429,7 @@ with block:
     """)
     with gr.Row():
         with gr.Column():
-            input_image = gr.ImageEditor(type="numpy", label="Image", height=320, brush=gr.Brush(colors=["#ffffff"]))
+            input_image = gr.Image(sources='upload', type="numpy", label="Image", height=320)
             prompt = gr.Textbox(label="Prompt", value='')
             t2v = gr.Checkbox(label="do text-to-video", value=False)
             example_quick_prompts = gr.Dataset(samples=quick_prompts, label='Quick List', samples_per_page=1000, components=[prompt])
@@ -485,7 +460,7 @@ with block:
                     mp4_crf = gr.Slider(label="MP4 Compression", minimum=0, maximum=100, value=16, step=1, info="Lower means better quality. 0 is uncompressed. Change to 16 if you get black outputs. ")
 
                 with gr.Accordion("Debug", open=False):
-                    input_image_debug = gr.ImageEditor(type="numpy", label="Image Debug", height=320, brush=gr.Brush(colors=["#ffffff"]))
+                    input_image_debug = gr.Image(type="numpy", label="Image Debug", height=320)
                     prompt_debug = gr.Textbox(label="Prompt Debug", value='')
                     total_second_length_debug = gr.Slider(label="Additional Video Length to Generate (Seconds) Debug", minimum=1, maximum=120, value=5, step=0.1)