Spaces:
Paused
Paused
Ali Mohsin
commited on
Commit
Β·
d9371aa
1
Parent(s):
9c8bd0b
fix4453112
Browse files
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 |
-
|
|
|
|
| 952 |
idxs = []
|
| 953 |
-
|
|
|
|
|
|
|
| 954 |
try:
|
| 955 |
-
|
| 956 |
-
|
|
|
|
|
|
|
|
|
|
| 957 |
continue
|
| 958 |
imgs = [images[i] for i in idxs if 0 <= i < len(images)]
|
| 959 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|