Update app.py

app.py

@@ -18,21 +18,21 @@ from mosaic_generator.mosaic_builder import MosaicBuilder
 from mosaic_generator.metrics import mse, ssim_rgb
 
 
-# -------------------------------------------------------------------
-# GLOBAL TILE MANAGER → loaded ONCE for the entire Space
-# -------------------------------------------------------------------
+# ---------------------------------------------------------
+# GLOBAL TILE MANAGER — Loaded ONCE when the Space starts
+# ---------------------------------------------------------
 TM = TileManager()
 TM.load(sample_size=20000)
 
 
-# -------------------------------------------------------------------
+# ---------------------------------------------------------
 # MAIN PIPELINE
-# -------------------------------------------------------------------
+# ---------------------------------------------------------
 def run_pipeline(
     img, grid_size, tile_px, tile_sample,
     quantize_on, quantize_colors, show_grid
 ):
-    """Full end-to-end mosaic generation pipeline."""
+    """Full end-to-end mosaic pipeline."""
 
     if img is None:
         return None, None, None, "Upload an image first."
@@ -40,10 +40,10 @@ def run_pipeline(
     img_np = np.array(img.convert("RGB"))
     grid_n = int(grid_size)
 
-    # Crop
+    # --- Crop image to perfect grid division ---
     base = crop_to_multiple(img_np, grid_n)
 
-    # Optional quantization
+    # --- Optional quantization ---
     if quantize_on:
         try:
             q = Image.fromarray(base).quantize(
@@ -55,7 +55,7 @@ def run_pipeline(
         except Exception as e:
             return None, None, None, f"Quantization failed: {e}"
 
-    # Compute LAB means
+    # --- Compute LAB means ---
     try:
         t0 = time.perf_counter()
         cell_means, dims = compute_cell_means_lab(base, grid_n)
@@ -65,16 +65,16 @@ def run_pipeline(
 
     w, h, cell_w, cell_h = dims
 
-    # Prepare tiles
+    # --- Prepare scaled tiles ---
     TM.prepare_scaled_tiles(cell_w, cell_h)
 
-    # FAISS lookup
+    # --- FAISS lookup ---
     try:
         idxs = TM.lookup_tiles(cell_means)
     except Exception as e:
         return None, None, None, f"Tile lookup failed: {e}"
 
-    # Mosaic build
+    # --- Build mosaic ---
     builder = MosaicBuilder(TM)
     try:
         mosaic_np = builder.build(idxs, dims, grid_n)
@@ -83,14 +83,14 @@ def run_pipeline(
 
     t2 = time.perf_counter()
 
-    # Metrics
+    # --- Metrics ---
     try:
         mse_val = mse(base, mosaic_np)
         ssim_val = ssim_rgb(base, mosaic_np)
-    except:
+    except Exception:
         mse_val, ssim_val = -1, -1
 
-    # Grid overlay
+    # --- Optional grid overlay ---
     segmented = Image.fromarray(base)
     if show_grid:
         seg = segmented.copy()
@@ -101,7 +101,7 @@ def run_pipeline(
             draw.line([(0, y), (w, y)], fill="red", width=1)
         segmented = seg
 
-    # Text report
+    # --- Build report text ---
     report = (
         f"MSE: {mse_val:.2f}\n"
         f"SSIM: {ssim_val:.4f}\n\n"
@@ -118,26 +118,32 @@ def run_pipeline(
     )
 
 
-# -------------------------------------------------------------------
+# ---------------------------------------------------------
 # GRADIO UI
-# -------------------------------------------------------------------
+# ---------------------------------------------------------
 def build_demo():
     with gr.Blocks(title="High-Performance Mosaic Generator") as demo:
 
         gr.Markdown("# ⚡ High-Performance Mosaic Generator (Lab 5)")
-        gr.Markdown("FAISS-accelerated mosaic generation • 20×–100× faster than Lab 1.\n")
+        gr.Markdown("Ultra-fast FAISS-powered image mosaic generator.\n")
 
         with gr.Row():
 
-            # ------------------------------------------------
-            # LEFT — Inputs
-            # ------------------------------------------------
+            # ---------------- LEFT COLUMN --------------------
             with gr.Column(scale=1):
 
                 img_in = gr.Image(type="pil", label="Upload Image")
 
-                grid_size = gr.Radio(["16", "32", "64", "128"], value="32", label="Grid Size")
-                tile_px = gr.Radio(["8", "16", "24", "32"], value="16", label="Tile Resolution (px)")
+                grid_size = gr.Radio(
+                    ["16", "32", "64", "128"],
+                    value="32",
+                    label="Grid Size"
+                )
+                tile_px = gr.Radio(
+                    ["8", "16", "24", "32"],
+                    value="16",
+                    label="Tile Resolution (px)"
+                )
 
                 tile_sample = gr.Slider(
                     512, 20000, step=256, value=2048,
@@ -154,26 +160,23 @@ def build_demo():
 
                 run_btn = gr.Button("Generate Mosaic", variant="primary")
 
-                # ------------------------------------------------
-                # EXAMPLE IMAGES (URL-based → HF compatible)
-                # ------------------------------------------------
-                gr.Markdown("### Example Images")
+                # ---------------- Example Images (HF-safe URLs) ----------------
+                gr.Markdown("### Example Images (Click to Load)")
 
                 example_urls = [
-                    ["https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/gradio/image_classifier/squirrel.jpg"],
-                    ["https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/gradio/image_classifier/cat.jpg"],
-                    ["https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/gradio/image_classifier/cheetah.jpg"],
+                    ["https://huggingface.co/datasets/hf-internal-testing/dog/resolve/main/dog.jpg"],
+                    ["https://huggingface.co/datasets/mishig/sample_images/resolve/main/cat.png"],
+                    ["https://huggingface.co/datasets/mishig/sample_images/resolve/main/mountain.png"],
                 ]
 
                 gr.Examples(
                     examples=example_urls,
                     inputs=[img_in],
+                    label="",
                     cache_examples=False
                 )
 
-            # ------------------------------------------------
-            # RIGHT — Outputs
-            # ------------------------------------------------
+            # ---------------- RIGHT COLUMN --------------------
             with gr.Column(scale=2):
 
                 with gr.Tab("Original"):
@@ -187,7 +190,7 @@ def build_demo():
 
                 report = gr.Textbox(label="Timing & Metrics", lines=10)
 
-        # Connect Generate button
+        # Connect button → function
         run_btn.click(
             fn=run_pipeline,
             inputs=[img_in, grid_size, tile_px, tile_sample,
@@ -198,9 +201,9 @@ def build_demo():
     return demo
 
 
-# -------------------------------------------------------------------
+# ---------------------------------------------------------
 # LAUNCH
-# -------------------------------------------------------------------
+# ---------------------------------------------------------
 if __name__ == "__main__":
     demo = build_demo()
     demo.launch()