updated path to save sketch and then reading file

api/model/src/datasets/dresscode.py

@@ -83,28 +83,42 @@ class DressCodeDataset(data.Dataset):
         '''
 
         # Load Captions
-        model_data  = self.json_from_req.get('MODEL', {})  # Safely get the 'MODEL' key, default to an empty dictionary if it doesn't exist
+        model_data  = self.json_from_req.get('dresscode', {})  # Safely get the 'MODEL' key, default to an empty dictionary if it doesn't exist
         # Filter captions based on the length requirement (3 or more items)
         self.captions_dict = {k: v for k, v in model_data.items() if len(v) >= 3}
 
-        for c in category:
-            assert c in ['dresses', 'upper_body', 'lower_body']
+        # Retrieving bodypart sketch: dresses, upper_body, lower_body
+        body_part = self.json_from_req.get('body_part', '')
 
-            dataroot = self.dataroot / c
-            if phase == 'train':
-                filename = dataroot / f"{phase}_pairs.txt"
-            else:
-                filename = dataroot / f"{phase}_pairs_{order}.txt"
+        #for c in category:
+        #    assert c in ['dresses', 'upper_body', 'lower_body']
+
+        dataroot = self.dataroot / body_part
+        #if phase == 'train':
+        #    filename = dataroot / f"{phase}_pairs.txt"
+        #else:
+        #    filename = dataroot / f"{phase}_pairs_{order}.txt"
+
+        # Creating new file name
+        filename = dataroot / "test_pairs_paired.txt"
+
+        with open(filename, 'w') as f:
+            first_key = list(model_data.keys())[0]
+            img_name1 = f"{first_key}_0.jpg"
+            img_name2 = f"{first_key}_1.jpg"
+
+            print("###########################################           ", img_name1)
+            f.write(f"{img_name1}    {img_name2}\n")
 
-            with open(filename, 'r') as f:
-                for line in f.readlines():
-                    im_name, c_name = line.strip().split()
-                    if c_name.split('_')[0] not in self.captions_dict:
-                        continue
+        with open(filename, 'r') as f:
+            for line in f.readlines():
+                im_name, c_name = line.strip().split()
+                if c_name.split('_')[0] not in self.captions_dict:
+                    continue
 
-                    im_names.append(im_name)
-                    c_names.append(c_name)
-                    dataroot_names.append(dataroot)
+                im_names.append(im_name)
+                c_names.append(c_name)
+                dataroot_names.append(dataroot)
 
         self.im_names = im_names
         self.c_names = c_names