Revert "debug"

This reverts commit b83cf9477adf35e17f9cabfb5cbb40d082ede6b2.

app.py

@@ -3,6 +3,7 @@ from typing import Tuple
 
 import gradio as gr
 import numpy as np
+import requests
 import spaces
 import torch
 from PIL import Image, ImageFilter
@@ -24,6 +25,53 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 # HF_TOKEN = os.environ.get("HF_TOKEN", None)
 # client = Client("SkalskiP/florence-sam-masking", hf_token=HF_TOKEN)
 
+
+def remove_background(image: Image.Image, threshold: int = 50) -> Image.Image:
+    image = image.convert("RGBA")
+    data = image.getdata()
+    new_data = []
+    for item in data:
+        avg = sum(item[:3]) / 3
+        if avg < threshold:
+            new_data.append((0, 0, 0, 0))
+        else:
+            new_data.append(item)
+
+    image.putdata(new_data)
+    return image
+
+
+# EXAMPLES = [
+#     [
+#         None,
+#         {
+#             "background": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-image.png", stream=True).raw),
+#             "layers": [remove_background(Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-mask-2.png", stream=True).raw))],
+#             "composite": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-composite-2.png", stream=True).raw),
+#         },
+#         "little lion",
+#         None,
+#         42,
+#         False,
+#         0.85,
+#         30
+#     ],
+#     [
+#         None,
+#         {
+#             "background": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-image.png", stream=True).raw),
+#             "layers": [remove_background(Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-mask-3.png", stream=True).raw))],
+#             "composite": Image.open(requests.get("https://media.roboflow.com/spaces/doge-2-composite-3.png", stream=True).raw),
+#         },
+#         "tattoos",
+#         None,
+#         42,
+#         False,
+#         0.85,
+#         30
+#     ]
+# ]
+
 pipe = FluxInpaintPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
 
@@ -75,8 +123,6 @@ def process(
         gr.Info("Please enter a text prompt.")
         return None, None
 
-    print(input_image_editor)
-
     image_path = input_image_editor['background']
     mask_path = input_image_editor['layers'][0]
 
@@ -194,6 +240,27 @@ with gr.Blocks() as demo:
             with gr.Accordion("Debug", open=False):
                 output_mask_component = gr.Image(
                     type='pil', image_mode='RGB', label='Input mask', format="png")
+    # with gr.Row():
+        # gr.Examples(
+        #     fn=process,
+        #     examples=EXAMPLES,
+        #     inputs=[
+        #         input_image_editor_component,
+        #         inpainting_prompt_text_component,
+        #         masking_prompt_text_component,
+        #         masking_prompt_text_component,
+        #         seed_slicer_component,
+        #         randomize_seed_checkbox_component,
+        #         strength_slider_component,
+        #         num_inference_steps_slider_component
+        #     ],
+        #     outputs=[
+        #         output_image_component,
+        #         output_mask_component
+        #     ],
+        #     run_on_click=True,
+        #     cache_examples=True
+        # )
 
     submit_button_component.click(
         fn=process,