ptom

promot suport

app.py

@@ -8,7 +8,7 @@ from huggingface_hub import HfApi, hf_hub_download
 # ======================================================
 # CONFIG
 # ======================================================
-DEFAULT_REPO = "rahul7star/WAN22-Mar-2026"
+DEFAULT_REPO = "rahul7star/WAN22-Feb-2026"
 FREE_REPO = "rahul7star/ohamlab"
 FREE_IMAGE_DIR = "showcase/image"
 
@@ -24,7 +24,42 @@ api = HfApi()
 PAGE_SIZE = 6
 
 tmp_dir = tempfile.mkdtemp()
-images_cache = {}   # (repo, date) -> images
+images_cache = {}     # (repo, date) -> list[(img, prompt)]
+prompt_cache = {}     # folder_path -> prompt text
+
+# ======================================================
+# PROMPT LOADER (NEW)
+# ======================================================
+def load_prompt_from_folder(repo, folder):
+    """
+    Tries to find prompt text inside the same upload folder.
+    Cached for performance.
+    """
+    key = (repo, folder)
+    if key in prompt_cache:
+        return prompt_cache[key]
+
+    files = api.list_repo_files(repo, token=HF_TOKEN)
+    prompt = ""
+
+    for f in files:
+        if not f.startswith(folder + "/"):
+            continue
+
+        name = os.path.basename(f).lower()
+
+        if name in ("prompt.txt", "caption.txt"):
+            try:
+                path = hf_hub_download(repo, f, token=HF_TOKEN)
+                with open(path, "r", encoding="utf-8", errors="ignore") as fp:
+                    prompt = fp.read().strip()
+            except Exception as e:
+                print("[PROMPT ERROR]", e)
+            break
+
+    prompt_cache[key] = prompt
+    return prompt
+
 
 # ======================================================
 # FREE PREVIEW
@@ -46,8 +81,8 @@ def load_free_images():
     for f in imgs:
         try:
             out.append(hf_hub_download(FREE_REPO, f, token=HF_TOKEN, local_dir=tmp))
-        except Exception as e:
-            print("[FREE IMG ERROR]", e)
+        except Exception:
+            pass
 
     return out
 
@@ -59,7 +94,8 @@ FREE_IMAGES = load_free_images()
 # ======================================================
 def clear_cache():
     images_cache.clear()
-    print("[CACHE] cleared all")
+    prompt_cache.clear()
+    print("[CACHE] cleared")
 
 
 def list_date_folders(repo, last_n=20):
@@ -77,7 +113,7 @@ def download_images(repo, date):
         return images_cache[key]
 
     files = api.list_repo_files(repo, token=HF_TOKEN)
-    local = []
+    results = []
 
     for f in files:
         if not f.startswith(date + "/"):
@@ -85,39 +121,41 @@ def download_images(repo, date):
         if not f.lower().endswith((".png", ".jpg", ".jpeg", ".webp")):
             continue
 
+        folder = os.path.dirname(f)
         dst = os.path.join(tmp_dir, repo.replace("/", "_"), f)
         os.makedirs(os.path.dirname(dst), exist_ok=True)
 
         try:
-            path = hf_hub_download(
+            img_path = hf_hub_download(
                 repo,
                 f,
                 token=HF_TOKEN,
                 local_dir=os.path.dirname(dst),
             )
-            local.append(path)
-            print("[DOWNLOAD]", repo, f)
+            prompt = load_prompt_from_folder(repo, folder)
+            results.append((img_path, prompt))
+            print("[DOWNLOAD]", f)
         except Exception as e:
             print("[DOWNLOAD ERROR]", f, e)
 
-    images_cache[key] = local
-    print(f"[INFO] {repo} {date}: {len(local)} images")
-    return local
+    images_cache[key] = results
+    print(f"[INFO] {repo} {date}: {len(results)} images")
+    return results
 
 
 # ======================================================
 # PAGINATION
 # ======================================================
 def render_page(repo, date, page):
-    images = download_images(repo, date)
-    total_pages = max(1, math.ceil(len(images) / PAGE_SIZE))
+    items = download_images(repo, date)
+    total_pages = max(1, math.ceil(len(items) / PAGE_SIZE))
 
     page = max(0, min(int(page), total_pages - 1))
     start = page * PAGE_SIZE
     end = start + PAGE_SIZE
 
     return (
-        images[start:end],
+        items[start:end],
         page,
         f"Page {page+1} / {total_pages}",
     )
@@ -170,7 +208,7 @@ def check_password(pwd):
 
 
 # ======================================================
-# CSS
+# CSS (UNCHANGED)
 # ======================================================
 css = """
 footer, .gradio-footer { display:none!important; }
@@ -192,7 +230,6 @@ footer, .gradio-footer { display:none!important; }
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.Markdown("# 🔐 OhamLab Image Showcase (PRO)")
 
-    # ---------- LOGIN ----------
     with gr.Column(visible=True) as login_box:
         pwd = gr.Textbox(type="password", label="Password")
         login_btn = gr.Button("Unlock")
@@ -202,79 +239,45 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             for img in FREE_IMAGES:
                 gr.Image(img, show_label=False)
 
-    # ---------- APP ----------
     with gr.Column(visible=False) as app:
         repo_state = gr.State(DEFAULT_REPO)
         date_state = gr.State("")
         page_state = gr.State(0)
 
-        repo_input = gr.Textbox(
-            value=DEFAULT_REPO,
-            label="Base HuggingFace Repo",
-        )
-
+        repo_input = gr.Textbox(value=DEFAULT_REPO, label="Base HuggingFace Repo")
         date_dropdown = gr.Dropdown(label="Select Date")
+        manual_date = gr.Textbox(label="Or enter date (YYYY-MM-DD)")
 
-        manual_date = gr.Textbox(
-            label="Or enter date manually (YYYY-MM-DD)"
-        )
-
-        gallery = gr.Gallery(columns=3, height="auto")
+        gallery = gr.Gallery(columns=3, height="auto", show_label=False)
         page_info = gr.Markdown()
 
         with gr.Row():
             prev_btn = gr.Button("⬅ Previous")
             next_btn = gr.Button("Next ➡")
 
-    # ---------- EVENTS ----------
     login_btn.click(
         check_password,
         inputs=pwd,
-        outputs=[
-            login_box,
-            app,
-            gallery,
-            page_state,
-            date_state,
-            page_info,
-        ],
+        outputs=[login_box, app, gallery, page_state, date_state, page_info],
     )
 
     repo_input.change(
         change_repo,
         inputs=repo_input,
-        outputs=[
-            date_dropdown,
-            gallery,
-            page_state,
-            date_state,
-            page_info,
-        ],
-    ).then(
-        lambda r: r,
-        inputs=repo_input,
-        outputs=repo_state,
-    )
+        outputs=[date_dropdown, gallery, page_state, date_state, page_info],
+    ).then(lambda r: r, inputs=repo_input, outputs=repo_state)
 
     date_dropdown.change(
         lambda r, d: change_date(r, d),
         inputs=[repo_state, date_dropdown],
         outputs=[gallery, page_state, page_info],
-    ).then(
-        lambda d: d,
-        inputs=date_dropdown,
-        outputs=date_state,
-    )
+    ).then(lambda d: d, inputs=date_dropdown, outputs=date_state)
 
     manual_date.submit(
         lambda r, d: change_date(r, d),
         inputs=[repo_state, manual_date],
         outputs=[gallery, page_state, page_info],
-    ).then(
-        lambda d: d,
-        inputs=manual_date,
-        outputs=date_state,
-    )
+    ).then(lambda d: d, inputs=manual_date, outputs=date_state)
 
     prev_btn.click(
         lambda r, d, p: nav(-1, r, d, p),