local support

.env.example

@@ -0,0 +1,6 @@
+# Credentials for the app login
+APP_USERNAME=admin
+APP_PASSWORD=password
+
+# Hugging Face token for downloading private/restricted data
+HF_TOKEN=your_huggingface_token_here

.gitignore

@@ -11,4 +11,6 @@ __pycache__/
 .gradio/
 
 # Mac
-.DS_Store
+.DS_Store
+
+.env

app.py

@@ -3,7 +3,11 @@ import os
 import json
 import gradio as gr
 import argparse
-from huggingface_hub import snapshot_download
+from huggingface_hub import hf_hub_download, snapshot_download
+from dotenv import load_dotenv
+
+# Load local .env file if it exists
+load_dotenv()
 
 HF_DATA_REPO = "doosaganesh/virtual-tryon-data"
 DATA_ROOT = None
@@ -34,6 +38,14 @@ def check_login(username, password):
 def get_data_root():
     global DATA_ROOT
 
+    # Priority 1: Check if datasets folder exists locally in the project root
+    if os.path.exists("datasets"):
+        print("Using local datasets folder.")
+        return os.getcwd()
+        
+    # Priority 2: Use tmp storage (standard for HF Spaces)
+    data_root = "/tmp/tryon-data"
+
     if DATA_ROOT is None:
 
         print("Downloading dataset from Hugging Face...")
@@ -225,16 +237,14 @@ def build_ui(site_id, person_image="source_models/female/asian/model_1.png"):
             fn=go_prev,
             inputs=[outfit_selector],
             outputs=[outfit_selector],
-            queue=False,
-            show_progress="hidden"
+            queue=False
         )
 
         next_btn.click(
             fn=go_next,
             inputs=[outfit_selector],
             outputs=[outfit_selector],
-            queue=False,
-            show_progress="hidden"
+            queue=False
         )
 
         def update_view(index):
@@ -275,24 +285,21 @@ def build_ui(site_id, person_image="source_models/female/asian/model_1.png"):
             fn=update_view,
             inputs=[outfit_selector],
             outputs=[garments_gallery, result_img, review_radio, model_img, seed_prod_txt, seed_ext_txt],
-            queue=False,
-            show_progress="hidden"
+            queue=False
         )
 
         review_radio.change(
             fn=on_review_change,
             inputs=[outfit_selector, review_radio],
             outputs=[],
-            queue=False,
-            show_progress="hidden"
+            queue=False
         )
 
         demo.load(
             fn=update_view,
             inputs=[outfit_selector],
             outputs=[garments_gallery, result_img, review_radio, model_img, seed_prod_txt, seed_ext_txt],
-            queue=False,
-            show_progress="hidden"
+            queue=False
         )
 
     return demo

requirements.txt

@@ -1,4 +1,5 @@
 gradio
 huggingface_hub
 requests
-Pillow
+Pillow
+python-dotenv