Update app_demo.py

app_demo.py

@@ -98,9 +98,18 @@ def get_outfits():
                             prod_to_ext[str(parts[id_idx])] = str(parts[ext_idx])
             break
 
+    # Only include outfits that have been generated (use model_1 as reference)
+    ref_results_dir = os.path.join(get_data_root(), "generated_tryon", SITE_ID, "model_1", "batch_results")
+    generated_ids = set()
+    if os.path.exists(ref_results_dir):
+        generated_ids = {f.replace(".jpg", "") for f in os.listdir(ref_results_dir)}
+
     outfits_data = []
     for idx, o in enumerate(outfits):
         outfit_id = o.get('outfitId', o.get('styleId', idx))
+        outfit_key = f"outfit-{outfit_id}"
+        if outfit_key not in generated_ids:
+            continue
         garment_data = []
         for g in o.get('garments', []):
             prod_id = str(g.get('prodId', ''))
@@ -119,7 +128,7 @@ def get_outfits():
         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}",
+            'id': outfit_key,
             'name': o.get('styleName', 'Unknown Style'),
             'garments': garment_data,
             'seed_prod_id': seed_prod_id,