Update app.py

app.py

@@ -2,6 +2,25 @@
 import gradio as gr
 from pipeline import run_search_and_generate
 
+# --- Quick Starters: load example images directly from the HF dataset repo ---
+from huggingface_hub import hf_hub_download
+from pathlib import Path
+from PIL import Image
+
+HF_DATASET_ID = "LevyJonas/sat_land_patches"
+QS_CACHE = Path("qs_cache")
+QS_CACHE.mkdir(parents=True, exist_ok=True)
+
+def load_qs_image(rel_path: str) -> Image.Image:
+    p = hf_hub_download(
+        repo_id=HF_DATASET_ID,
+        repo_type="dataset",
+        filename=rel_path,
+        local_dir=str(QS_CACHE),
+        local_dir_use_symlinks=False,
+    )
+    return Image.open(p).convert("RGB")
+
 def run_app(user_img, prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed):
     try:
         retrieved, gen_i2i, gen_t2i, info = run_search_and_generate(
@@ -23,12 +42,31 @@ def run_app(user_img, prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen
     except Exception as e:
         return [], [], [], f"Error: {e}"
 
+# Quick starter runner: loads image from HF and calls the same run_app function
+def run_quickstarter(rel_path, prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed):
+    img = load_qs_image(rel_path)
+    return run_app(img, prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed)
+
+
 with gr.Blocks(title="Satellite Patch: Retrieve + Generate") as demo:
     gr.Markdown(
         "# Satellite Patch: Retrieve + Generate\n"
         "Upload a satellite patch + write a prompt → get (1) similar images from the dataset and (2) newly generated images."
     )
 
+    # --- Quick Starters (1-click examples) ---
+    gr.Markdown("### Quick Starters (1-click examples)")
+
+    with gr.Row():
+        qs1 = gr.Button("Quick Starter: LakeWater")
+        qs2 = gr.Button("Quick Starter: DenseForest")
+        qs3 = gr.Button("Quick Starter: ResidentialDense")
+
+    # Change these paths if needed (must exist in your dataset repo)
+    QS_1_PATH = "images/LakeWater/LakeWater_000000.jpg"
+    QS_2_PATH = "images/DenseForest/DenseForest_000000.jpg"
+    QS_3_PATH = "images/ResidentialDense/ResidentialDense_000000.jpg"
+
     with gr.Row():
         with gr.Column(scale=1):
             user_img = gr.Image(type="pil", label="Upload satellite patch")
@@ -55,8 +93,28 @@ with gr.Blocks(title="Satellite Patch: Retrieve + Generate") as demo:
             out_t2i = gr.Gallery(label="Generated (txt2img)", columns=5, height=260)
             out_txt = gr.Textbox(label="Summary", lines=8)
 
-    btn.click(run_app,
-              inputs=[user_img, prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed],
-              outputs=[out_retr, out_i2i, out_t2i, out_txt])
+    # Normal run (user upload)
+    btn.click(
+        run_app,
+        inputs=[user_img, prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed],
+        outputs=[out_retr, out_i2i, out_t2i, out_txt],
+    )
+
+    # Quick Starter runs (1 click)
+    qs1.click(
+        run_quickstarter,
+        inputs=[gr.State(QS_1_PATH), prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed],
+        outputs=[out_retr, out_i2i, out_t2i, out_txt],
+    )
+    qs2.click(
+        run_quickstarter,
+        inputs=[gr.State(QS_2_PATH), prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed],
+        outputs=[out_retr, out_i2i, out_t2i, out_txt],
+    )
+    qs3.click(
+        run_quickstarter,
+        inputs=[gr.State(QS_3_PATH), prompt, k_retrieve, n_i2i, n_t2i, strength_i2i, steps, gen_size, seed],
+        outputs=[out_retr, out_i2i, out_t2i, out_txt],
+    )
 
 demo.launch()