Update app.py

app.py

@@ -3,6 +3,8 @@ import math
 import time
 import random
 import tempfile
+from datetime import datetime, date
+
 import gradio as gr
 from huggingface_hub import HfApi, hf_hub_download
 
@@ -49,6 +51,18 @@ if not APP_PASSWORD:
 api = HfApi()
 
 
+# ======================================================
+# DATE PARSING
+# ======================================================
+def parse_date_safe(name: str) -> datetime:
+    for fmt in ("%Y-%m-%d", "%Y_%m_%d", "%Y%m%d"):
+        try:
+            return datetime.strptime(name, fmt)
+        except ValueError:
+            pass
+    return datetime.combine(date.today(), datetime.min.time())
+
+
 # ======================================================
 # FREE IMAGE PREVIEW
 # ======================================================
@@ -62,11 +76,8 @@ def load_free_images():
         and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp"))
     ]
 
-    if not images:
-        return []
-
     random.shuffle(images)
-    picked = (images * 5)[:5] if len(images) < 5 else images[:5]
+    picked = images[:5]
 
     local = []
     for img in picked:
@@ -86,19 +97,25 @@ FREE_IMAGES = load_free_images()
 # ======================================================
 def list_date_folders():
     files = api.list_repo_files(REPO_ID)
-    return sorted({f.split("/")[0] for f in files if "/" in f})
+    folders = {f.split("/")[0] for f in files if "/" in f}
+
+    return sorted(folders, key=parse_date_safe, reverse=True)
 
 
 def load_images_for_date(date_folder):
     files = api.list_repo_files(REPO_ID)
-    items = []
+    prefix = f"{date_folder}/"
 
+    images = []
     for f in files:
-        if f.startswith(date_folder + "/") and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp")):
-            url = f"https://huggingface.co/{REPO_ID}/resolve/main/{f}"
-            items.append(url)
+        if f.startswith(prefix) and f.lower().endswith(
+            (".png", ".jpg", ".jpeg", ".webp")
+        ):
+            images.append(
+                f"https://huggingface.co/{REPO_ID}/resolve/main/{f}"
+            )
 
-    return items
+    return images
 
 
 # ======================================================
@@ -112,7 +129,7 @@ def render_grid(date_folder, page, page_size):
     start, end = page * page_size, (page + 1) * page_size
 
     cards = [
-        f"<div class='card'><img src='{img}'></div>"
+        f"<div class='card'><img src='{img}' loading='lazy'></div>"
         for img in data[start:end]
     ]
 
@@ -125,12 +142,10 @@ def render_grid(date_folder, page, page_size):
 def next_page(date, page, size):
     return render_grid(date, page + 1, size)
 
+
 def prev_page(date, page, size):
     return render_grid(date, page - 1, size)
 
-def reset_page(date, size):
-    return render_grid(date, 0, size)
-
 
 # ======================================================
 # AUTH
@@ -145,11 +160,12 @@ def check_password(pwd):
             "",
             0,
             "❌ Wrong password",
-             latest
+            ""
         )
 
     dates = list_date_folders()
-    latest = dates[-1]
+    latest = dates[0]  # newest date folder
+
     html, page = render_grid(latest, 0, 6)
 
     return (
@@ -181,9 +197,7 @@ css = """
   width:100%;
   border-radius:10px;
 }
-
 footer, .gradio-footer { display:none!important; }
-
 #ohamlab-footer {
   position:fixed;
   bottom:0;
@@ -200,10 +214,9 @@ footer, .gradio-footer { display:none!important; }
 # ======================================================
 # UI
 # ======================================================
-with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    gr.Markdown("# 🔐 OhamLab Image Showcase- For PRO Users 18+ only unlock")
+with gr.Blocks() as demo:
+    gr.Markdown("# 🔐 OhamLab Image Showcase – PRO Users Only")
 
-    # ---------- LOGIN ----------
     with gr.Column(visible=True) as login_box:
         pwd = gr.Textbox(type="password", label="Password")
         login_btn = gr.Button("Unlock")
@@ -215,7 +228,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 for img in FREE_IMAGES:
                     gr.Image(img, show_label=False)
 
-    # ---------- PRO ----------
     with gr.Column(visible=False) as app:
         current_date = gr.State("")
         page_state = gr.State(0)
@@ -236,7 +248,14 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         outputs=[login_box, app, free_preview, gallery, page_state, status, current_date]
     )
 
-    gr.HTML("<div id='ohamlab-footer'>© OhamLab Copyright</div>")
+    gr.HTML("<div id='ohamlab-footer'>© OhamLab</div>")
 
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch(
+        theme=gr.themes.Soft(),
+        css=css,
+        prevent_thread_lock=True,
+        ssr_mode=False,
+        show_error=True
+    )