Ali Mohsin commited on
Commit
d9371aa
Β·
1 Parent(s): 9c8bd0b

fix4453112

Browse files
Files changed (1) hide show
  1. app.py +25 -5
app.py CHANGED
@@ -940,23 +940,43 @@ def gradio_recommend(
940
  {"id": f"item_{i}", "image": images[i], "category": None}
941
  for i in range(len(images))
942
  ]
 
943
  res = service.compose_outfits(items, context=context)
944
 
 
 
 
 
945
  # Check if compose_outfits returned an error
946
  if res and isinstance(res[0], dict) and "error" in res[0]:
 
947
  return [], res[0]
948
 
949
  # Prepare stitched previews
950
  strips: List[Image.Image] = []
951
- for r in res:
 
952
  idxs = []
953
- for iid in r.get("item_ids", []):
 
 
954
  try:
955
- idxs.append(int(str(iid).split("_")[-1]))
956
- except Exception:
 
 
 
957
  continue
958
  imgs = [images[i] for i in idxs if 0 <= i < len(images)]
959
- strips.append(_stitch_strip(imgs))
 
 
 
 
 
 
 
 
960
  return strips, {"outfits": res}
961
 
962
 
 
940
  {"id": f"item_{i}", "image": images[i], "category": None}
941
  for i in range(len(images))
942
  ]
943
+ print(f"πŸ” DEBUG: Calling compose_outfits with {len(items)} items, context={context}")
944
  res = service.compose_outfits(items, context=context)
945
 
946
+ print(f"πŸ” DEBUG: compose_outfits returned {len(res) if res else 0} results")
947
+ if res:
948
+ print(f"πŸ” DEBUG: First result type: {type(res[0])}, keys: {res[0].keys() if isinstance(res[0], dict) else 'N/A'}")
949
+
950
  # Check if compose_outfits returned an error
951
  if res and isinstance(res[0], dict) and "error" in res[0]:
952
+ print(f"πŸ” DEBUG: Error in compose_outfits result: {res[0]}")
953
  return [], res[0]
954
 
955
  # Prepare stitched previews
956
  strips: List[Image.Image] = []
957
+ print(f"πŸ” DEBUG: Preparing stitched previews for {len(res)} outfits...")
958
+ for i, r in enumerate(res):
959
  idxs = []
960
+ item_ids = r.get("item_ids", [])
961
+ print(f"πŸ” DEBUG: Outfit {i+1}: item_ids={item_ids}")
962
+ for iid in item_ids:
963
  try:
964
+ idx = int(str(iid).split("_")[-1])
965
+ idxs.append(idx)
966
+ print(f"πŸ” DEBUG: Mapped {iid} -> index {idx}")
967
+ except Exception as e:
968
+ print(f"πŸ” DEBUG: Failed to parse {iid}: {e}")
969
  continue
970
  imgs = [images[i] for i in idxs if 0 <= i < len(images)]
971
+ print(f"πŸ” DEBUG: Extracted {len(imgs)} images from indices {idxs}")
972
+ if imgs:
973
+ strip = _stitch_strip(imgs)
974
+ strips.append(strip)
975
+ print(f"πŸ” DEBUG: Created stitched image: {strip.size}")
976
+ else:
977
+ print(f"⚠️ DEBUG: No images extracted for outfit {i+1}")
978
+
979
+ print(f"πŸ” DEBUG: Returning {len(strips)} stitched images and {len(res)} outfit results")
980
  return strips, {"outfits": res}
981
 
982