wta demo

app.py

@@ -0,0 +1,63 @@
+import keras
+import numpy as np
+import gradio as gr
+from tibblingai import wta
+from PIL import Image, ImageOps
+import matplotlib.pyplot as plt 
+
+
+def load_sample_images():
+    sample_paths = ["sample1.jpg", "sample2.jpg"]  # Must be in the same folder as your script
+    original_imgs = [Image.open(path) for path in sample_paths]
+    processed_imgs = [process_image(path) for path in sample_paths]
+    return gr.update(visible=True, value=original_imgs), gr.update(visible=True, value=processed_imgs)
+
+# Image processing function (you can replace this)
+def process_image(img: Image.Image) -> Image.Image:
+    img_tensor = wta.wta(plt.imread(img)).numpy()
+    print(img_tensor.shape)
+    # image_array = np.squeeze(img_tensor)
+    return img_tensor[0]
+
+# Function to process uploaded images
+def process_images(images):
+    if not isinstance(images, list):
+        images = [images]
+
+    original_imgs = []
+    processed_imgs = []
+
+    for img in images:
+        original_imgs.append(img)
+        processed_imgs.append(process_image(img))
+
+    return gr.update(visible=True, value=original_imgs), gr.update(visible=True, value=processed_imgs)
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("## Upload Image(s) for Processing")
+
+    # File component (no preview shown here)
+    file_input = gr.File(file_types=["image"],file_count="multiple", label="Upload Images")
+
+    with gr.Row():
+        original_gallery = gr.Gallery(label="Original Images", visible=False)
+        processed_gallery = gr.Gallery(label="Processed Images", visible=False)
+
+    file_input.change(fn=process_images, inputs=file_input, outputs=[original_gallery, processed_gallery])
+    with gr.Row():
+      with gr.Column(scale=0, min_width=300):
+        load_examples_btn = gr.Button("Run WTA on sample images")
+        load_examples_btn.click(fn=load_sample_images, outputs=[original_gallery, processed_gallery])
+
+
+    # 🆕 Examples section
+#     examples = gr.Examples(
+#     examples=[["sample1.jpg"], ["sample2.jpg"]],
+#     inputs=file_input,
+#     label="Example Images"
+# )
+
+    # example_files = [["sample1.jpg", "sample2.jpg"], ["sample2.jpg"]]
+    # gr.Examples(examples=example_files, inputs=[file_input], label="Try one of our example samples")
+demo.launch()

requirements.txt

@@ -0,0 +1,6 @@
+gradio
+pillow
+matplotlib
+numpy
+tibblingai
+keras