auth

app.py

@@ -1,3 +1,4 @@
+
 import os
 import json
 import gradio as gr
@@ -5,9 +6,20 @@ import argparse
 from huggingface_hub import snapshot_download
 
 HF_DATA_REPO = "doosaganesh/virtual-tryon-data"
-
 DATA_ROOT = None
 
+# -----------------------------
+# Simple user database from Space secrets
+# -----------------------------
+VALID_USERS = {
+    os.getenv("APP_USERNAME"): os.getenv("APP_PASSWORD")
+}
+
+def check_login(username, password):
+    if username in VALID_USERS and VALID_USERS[username] == password:
+        return True
+    return False
+
 
 def get_data_root():
     global DATA_ROOT
@@ -19,7 +31,7 @@ def get_data_root():
         DATA_ROOT = snapshot_download(
             repo_id=HF_DATA_REPO,
             repo_type="dataset",
-            token=os.environ["HF_TOKEN"],
+            token=os.environ.get("HF_TOKEN"),
             local_dir="/tmp/tryon-data",
             local_dir_use_symlinks=False
         )
@@ -34,7 +46,6 @@ def get_outfits(json_path):
         data = json.load(f)
 
     outfits = data.get('recommendedOutfits', [])
-
     json_dir = os.path.dirname(os.path.abspath(json_path))
     prod_to_ext = {}
 
@@ -105,18 +116,13 @@ def build_ui(site_id, person_image="source_models/female/asian/model_1.png"):
     DATA_ROOT = get_data_root()
 
     json_path = os.path.join(DATA_ROOT, "datasets", site_id, f"{site_id}_processed.json")
-
     results_dir = os.path.join(DATA_ROOT, "generated_tryon", site_id, "batch_results")
 
-    if not os.path.exists(json_path):
-        print(f"Warning: {json_path} not found")
-
     outfits = get_outfits(json_path)
 
     with gr.Blocks(title=f"Try-On Viewer: Site {site_id}") as demo:
 
         gr.Markdown(f"# Virtual Try-On Validator: Site {site_id}")
-        gr.Markdown(f"Viewing results from `{results_dir}`")
 
         reviews_file = f"reviews_{site_id}.json"
 
@@ -144,15 +150,6 @@ def build_ui(site_id, person_image="source_models/female/asian/model_1.png"):
             next_btn = gr.Button("Next Outfit ➡️")
 
         with gr.Row():
-            load_from_urls = gr.Checkbox(
-                label="Load missing garment images from URLs (Internet)",
-                value=True
-            )
-
-            refresh_btn = gr.Button("Refresh Generated Outputs", variant="secondary")
-
-        with gr.Row():
-
             review_radio = gr.Radio(
                 choices=["Pending", "✅ Good", "❌ Bad"],
                 label="Review Status",
@@ -163,35 +160,23 @@ def build_ui(site_id, person_image="source_models/female/asian/model_1.png"):
 
             with gr.Column(scale=1):
 
-                seed_info_md = gr.Markdown("### Input Garments\n**Seed Product:** N/A")
-
                 garments_gallery = gr.Gallery(
                     label="Outfit Pieces",
                     columns=2,
                     object_fit="contain"
                 )
 
-                gr.Markdown("### Input Model")
-
-                person_img = gr.Image(
-                    value=person_image,
-                    label="Base Person",
-                    interactive=False
-                )
-
             with gr.Column(scale=2):
 
-                gr.Markdown("### Gemini Generated Result")
-
                 result_img = gr.Image(
-                    label="Generated Output (Waiting...)",
+                    label="Generated Output",
                     interactive=False
                 )
 
-        def update_view(index, load_urls):
+        def update_view(index):
 
             if index is None or not outfits:
-                return None, None, "Pending", "### Input Garments\n**Seed Product:** N/A"
+                return None, None, "Pending"
 
             o = outfits[index]
             outfit_id = o['id']
@@ -202,85 +187,22 @@ def build_ui(site_id, person_image="source_models/female/asian/model_1.png"):
             gallery_items = []
 
             for g in o['garments']:
-
-                img_data = g['image']
-
-                if not img_data and load_urls and g.get('url'):
-
-                    try:
-                        import requests as req
-                        from io import BytesIO
-                        from PIL import Image
-
-                        headers = {"User-Agent": "Mozilla/5.0"}
-
-                        resp = req.get(g['url'], timeout=5, headers=headers)
-
-                        if resp.status_code == 200:
-                            img_data = Image.open(BytesIO(resp.content))
-
-                    except Exception:
-                        pass
-
-                if img_data:
-                    label = f"{g['type']}: {g['name']}\nProdID: {g.get('prod_id')} | ExtID: {g.get('ext_id')}"
-                    gallery_items.append((img_data, label))
+                if g['image']:
+                    gallery_items.append((g['image'], g['name']))
 
             output_path = os.path.join(results_dir, f"{outfit_id}.jpg")
 
             if os.path.exists(output_path):
                 res = output_path
-                res_label = f"Generated Output ({outfit_id}.jpg)"
             else:
                 res = None
-                res_label = f"Not generated yet ({output_path} not found)"
 
-            seed_text = f"### Input Garments\n**Seed Product:** ProdID: {o.get('seed_prod_id', 'N/A')} | ExtID: {o.get('seed_ext_id', 'N/A')}"
-
-            return gallery_items, gr.update(value=res, label=res_label), current_review, gr.update(value=seed_text)
-
-        def go_prev(index):
-            if index is None:
-                return 0
-            return max(0, index - 1)
-
-        def go_next(index):
-            if index is None:
-                return 0
-            return min(len(outfits) - 1, index + 1)
-
-        def on_review_change(index, status):
-            if index is not None and outfits:
-                outfit_id = outfits[index]['id']
-                save_review(outfit_id, status)
+            return gallery_items, res, current_review
 
         outfit_selector.change(
             fn=update_view,
-            inputs=[outfit_selector, load_from_urls],
-            outputs=[garments_gallery, result_img, review_radio, seed_info_md]
-        )
-
-        load_from_urls.change(
-            fn=update_view,
-            inputs=[outfit_selector, load_from_urls],
-            outputs=[garments_gallery, result_img, review_radio, seed_info_md]
-        )
-
-        refresh_btn.click(
-            fn=update_view,
-            inputs=[outfit_selector, load_from_urls],
-            outputs=[garments_gallery, result_img, review_radio, seed_info_md]
-        )
-
-        prev_btn.click(fn=go_prev, inputs=[outfit_selector], outputs=[outfit_selector])
-        next_btn.click(fn=go_next, inputs=[outfit_selector], outputs=[outfit_selector])
-
-        review_radio.change(fn=on_review_change, inputs=[outfit_selector, review_radio])
-
-        demo.load(
-            fn=update_view,
-            inputs=[outfit_selector, load_from_urls],
-            outputs=[garments_gallery, result_img, review_radio, seed_info_md]
+            inputs=[outfit_selector],
+            outputs=[garments_gallery, result_img, review_radio]
         )
 
     return demo
@@ -289,30 +211,46 @@ def build_ui(site_id, person_image="source_models/female/asian/model_1.png"):
 if __name__ == "__main__":
 
     parser = argparse.ArgumentParser(description="Virtual Try-On UI Viewer")
-
     parser.add_argument("--site-id", default="1415")
     parser.add_argument("--person", default="source_models/female/asian/model_1.png")
     parser.add_argument("--port", type=int, default=7860)
-    parser.add_argument("--share", action="store_true")
-    parser.add_argument("--username")
-    parser.add_argument("--password")
 
     args = parser.parse_args()
 
-    auth = None
+    DATA_ROOT = get_data_root()
 
-    if args.username and args.password:
-        auth = (args.username, args.password)
-        print(f"Authentication enabled for user: {args.username}")
+    viewer = build_ui(site_id=args.site_id, person_image=args.person)
 
-    print(f"Starting Viewer for Site: {args.site_id}")
+    with gr.Blocks(title="Virtual Try-On Login") as demo:
 
-    demo = build_ui(site_id=args.site_id, person_image=args.person)
+        login_state = gr.State(False)
 
-    DATA_ROOT = get_data_root()
+        with gr.Column(visible=True) as login_panel:
+
+            gr.Markdown("# 🔐 Login Required")
+
+            username = gr.Textbox(label="Username")
+            password = gr.Textbox(label="Password", type="password")
+
+            login_btn = gr.Button("Login")
+            login_status = gr.Markdown()
+
+        with gr.Column(visible=False) as app_panel:
+            viewer.render()
+
+        def login(user, pwd):
+            if check_login(user, pwd):
+                return True, gr.update(visible=False), gr.update(visible=True), "✅ Login successful"
+            else:
+                return False, gr.update(visible=True), gr.update(visible=False), "❌ Invalid credentials"
+
+        login_btn.click(
+            login,
+            inputs=[username, password],
+            outputs=[login_state, login_panel, app_panel, login_status]
+        )
 
     demo.launch(
-    server_name="0.0.0.0",
-    server_port=args.port,
-    auth=auth
-)
+        server_name="0.0.0.0",
+        server_port=args.port
+    )

app_deprecated.py

@@ -0,0 +1,318 @@
+import os
+import json
+import gradio as gr
+import argparse
+from huggingface_hub import snapshot_download
+
+HF_DATA_REPO = "doosaganesh/virtual-tryon-data"
+
+DATA_ROOT = None
+
+
+def get_data_root():
+    global DATA_ROOT
+
+    if DATA_ROOT is None:
+
+        print("Downloading dataset from Hugging Face...")
+
+        DATA_ROOT = snapshot_download(
+            repo_id=HF_DATA_REPO,
+            repo_type="dataset",
+            token=os.environ["HF_TOKEN"],
+            local_dir="/tmp/tryon-data",
+            local_dir_use_symlinks=False
+        )
+
+        print("Dataset downloaded to:", DATA_ROOT)
+
+    return DATA_ROOT
+
+
+def get_outfits(json_path):
+    with open(json_path, 'r') as f:
+        data = json.load(f)
+
+    outfits = data.get('recommendedOutfits', [])
+
+    json_dir = os.path.dirname(os.path.abspath(json_path))
+    prod_to_ext = {}
+
+    for fname in os.listdir(json_dir):
+        if fname.endswith('products.txt'):
+            with open(os.path.join(json_dir, fname), 'r', encoding='utf-8') as pf:
+                first_line = pf.readline().strip()
+                if not first_line:
+                    continue
+
+                delimiter = '\x01' if '\x01' in first_line else '\t'
+                header = first_line.split(delimiter)
+
+                if 'id' in header and 'external_id' in header:
+                    id_idx = header.index('id')
+                    ext_idx = header.index('external_id')
+
+                    for line in pf:
+                        parts = line.strip().split(delimiter)
+                        if len(parts) > max(id_idx, ext_idx):
+                            prod_to_ext[str(parts[id_idx])] = str(parts[ext_idx])
+            break
+
+    outfits_data = []
+
+    for idx, o in enumerate(outfits):
+
+        outfit_id = o.get('outfitId', o.get('styleId', idx))
+        style_name = o.get('styleName', 'Unknown Style')
+
+        garment_data = []
+
+        for g in o.get('garments', []):
+
+            prod_id = str(g.get('prodId', ''))
+            ext_id = prod_to_ext.get(prod_id, prod_id)
+
+            img_path = os.path.join(json_dir, 'images', f"{ext_id}.jpg")
+
+            if not os.path.exists(img_path):
+                img_path = os.path.join(json_dir, 'images', f"{prod_id}.jpg")
+
+            garment_data.append({
+                'type': g.get('tryonType', 'unknown'),
+                'name': g.get('prodName', 'Unknown'),
+                'image': img_path if os.path.exists(img_path) else None,
+                'url': g.get('imageUrl'),
+                'prod_id': prod_id,
+                'ext_id': ext_id
+            })
+
+        seed_prod_id = str(o.get('seedProduct', ''))
+        seed_ext_id = prod_to_ext.get(seed_prod_id, seed_prod_id) if seed_prod_id else 'N/A'
+
+        outfits_data.append({
+            'id': f"outfit-{outfit_id}",
+            'name': style_name,
+            'garments': garment_data,
+            'seed_prod_id': seed_prod_id,
+            'seed_ext_id': seed_ext_id
+        })
+
+    return outfits_data
+
+
+def build_ui(site_id, person_image="source_models/female/asian/model_1.png"):
+
+    DATA_ROOT = get_data_root()
+
+    json_path = os.path.join(DATA_ROOT, "datasets", site_id, f"{site_id}_processed.json")
+
+    results_dir = os.path.join(DATA_ROOT, "generated_tryon", site_id, "batch_results")
+
+    if not os.path.exists(json_path):
+        print(f"Warning: {json_path} not found")
+
+    outfits = get_outfits(json_path)
+
+    with gr.Blocks(title=f"Try-On Viewer: Site {site_id}") as demo:
+
+        gr.Markdown(f"# Virtual Try-On Validator: Site {site_id}")
+        gr.Markdown(f"Viewing results from `{results_dir}`")
+
+        reviews_file = f"reviews_{site_id}.json"
+
+        def load_reviews():
+            if os.path.exists(reviews_file):
+                with open(reviews_file, 'r') as f:
+                    return json.load(f)
+            return {}
+
+        def save_review(outfit_id, status):
+            reviews = load_reviews()
+            reviews[outfit_id] = status
+            with open(reviews_file, 'w') as f:
+                json.dump(reviews, f, indent=2)
+
+        with gr.Row():
+            prev_btn = gr.Button("⬅️ Previous Outfit")
+
+            outfit_selector = gr.Dropdown(
+                choices=[(f"{o['id']} ({o['name']})", i) for i, o in enumerate(outfits)],
+                label="Select Outfit",
+                value=0 if outfits else None
+            )
+
+            next_btn = gr.Button("Next Outfit ➡️")
+
+        with gr.Row():
+            load_from_urls = gr.Checkbox(
+                label="Load missing garment images from URLs (Internet)",
+                value=True
+            )
+
+            refresh_btn = gr.Button("Refresh Generated Outputs", variant="secondary")
+
+        with gr.Row():
+
+            review_radio = gr.Radio(
+                choices=["Pending", "✅ Good", "❌ Bad"],
+                label="Review Status",
+                value="Pending"
+            )
+
+        with gr.Row():
+
+            with gr.Column(scale=1):
+
+                seed_info_md = gr.Markdown("### Input Garments\n**Seed Product:** N/A")
+
+                garments_gallery = gr.Gallery(
+                    label="Outfit Pieces",
+                    columns=2,
+                    object_fit="contain"
+                )
+
+                gr.Markdown("### Input Model")
+
+                person_img = gr.Image(
+                    value=person_image,
+                    label="Base Person",
+                    interactive=False
+                )
+
+            with gr.Column(scale=2):
+
+                gr.Markdown("### Gemini Generated Result")
+
+                result_img = gr.Image(
+                    label="Generated Output (Waiting...)",
+                    interactive=False
+                )
+
+        def update_view(index, load_urls):
+
+            if index is None or not outfits:
+                return None, None, "Pending", "### Input Garments\n**Seed Product:** N/A"
+
+            o = outfits[index]
+            outfit_id = o['id']
+
+            reviews = load_reviews()
+            current_review = reviews.get(outfit_id, "Pending")
+
+            gallery_items = []
+
+            for g in o['garments']:
+
+                img_data = g['image']
+
+                if not img_data and load_urls and g.get('url'):
+
+                    try:
+                        import requests as req
+                        from io import BytesIO
+                        from PIL import Image
+
+                        headers = {"User-Agent": "Mozilla/5.0"}
+
+                        resp = req.get(g['url'], timeout=5, headers=headers)
+
+                        if resp.status_code == 200:
+                            img_data = Image.open(BytesIO(resp.content))
+
+                    except Exception:
+                        pass
+
+                if img_data:
+                    label = f"{g['type']}: {g['name']}\nProdID: {g.get('prod_id')} | ExtID: {g.get('ext_id')}"
+                    gallery_items.append((img_data, label))
+
+            output_path = os.path.join(results_dir, f"{outfit_id}.jpg")
+
+            if os.path.exists(output_path):
+                res = output_path
+                res_label = f"Generated Output ({outfit_id}.jpg)"
+            else:
+                res = None
+                res_label = f"Not generated yet ({output_path} not found)"
+
+            seed_text = f"### Input Garments\n**Seed Product:** ProdID: {o.get('seed_prod_id', 'N/A')} | ExtID: {o.get('seed_ext_id', 'N/A')}"
+
+            return gallery_items, gr.update(value=res, label=res_label), current_review, gr.update(value=seed_text)
+
+        def go_prev(index):
+            if index is None:
+                return 0
+            return max(0, index - 1)
+
+        def go_next(index):
+            if index is None:
+                return 0
+            return min(len(outfits) - 1, index + 1)
+
+        def on_review_change(index, status):
+            if index is not None and outfits:
+                outfit_id = outfits[index]['id']
+                save_review(outfit_id, status)
+
+        outfit_selector.change(
+            fn=update_view,
+            inputs=[outfit_selector, load_from_urls],
+            outputs=[garments_gallery, result_img, review_radio, seed_info_md]
+        )
+
+        load_from_urls.change(
+            fn=update_view,
+            inputs=[outfit_selector, load_from_urls],
+            outputs=[garments_gallery, result_img, review_radio, seed_info_md]
+        )
+
+        refresh_btn.click(
+            fn=update_view,
+            inputs=[outfit_selector, load_from_urls],
+            outputs=[garments_gallery, result_img, review_radio, seed_info_md]
+        )
+
+        prev_btn.click(fn=go_prev, inputs=[outfit_selector], outputs=[outfit_selector])
+        next_btn.click(fn=go_next, inputs=[outfit_selector], outputs=[outfit_selector])
+
+        review_radio.change(fn=on_review_change, inputs=[outfit_selector, review_radio])
+
+        demo.load(
+            fn=update_view,
+            inputs=[outfit_selector, load_from_urls],
+            outputs=[garments_gallery, result_img, review_radio, seed_info_md]
+        )
+
+    return demo
+
+
+if __name__ == "__main__":
+
+    parser = argparse.ArgumentParser(description="Virtual Try-On UI Viewer")
+
+    parser.add_argument("--site-id", default="1415")
+    parser.add_argument("--person", default="source_models/female/asian/model_1.png")
+    parser.add_argument("--port", type=int, default=7860)
+    parser.add_argument("--share", action="store_true")
+    parser.add_argument("--username")
+    parser.add_argument("--password")
+
+    args = parser.parse_args()
+
+    auth = None
+
+    if args.username and args.password:
+        auth = (args.username, args.password)
+        print(f"Authentication enabled for user: {args.username}")
+
+    print(f"Starting Viewer for Site: {args.site_id}")
+
+    demo = build_ui(site_id=args.site_id, person_image=args.person)
+
+    DATA_ROOT = get_data_root()
+
+    demo.launch(
+    server_name="0.0.0.0",
+    server_port=args.port,
+    auth=auth
+)