updated backend to handle new models

api/app.py

@@ -51,31 +51,48 @@ def generate_design():
             print(f"### Errore nell'apertura dell'immagine: {str(e)} ###")  # Stampa per il debug
             return f"Failed to open the image: {str(e)}", 400
 
+        if 'vitonhd' in json_data_from_req:
+            dataset_name = 'vitonhd'
+            image_suffix = '_00.jpeg'
+            dataset_path = '/api/model/assets/data/vitonhd'
+            image_save_dir = os.path.join(dataset_path, 'test/im_sketch')
+        elif 'dresscode' in json_data_from_req:
+            dataset_name = 'dresscode'
+            image_suffix = '_01.jpeg'
+            body_part = json_data_from_req['body_part']
+            dataset_path = f'/api/model/assets/data/dresscode/{body_part}'
+            image_save_dir = os.path.join(dataset_path, 'im_sketch')
+        else:
+            print("### Chiave 'vitonhd' o 'dresscode' non trovata nel JSON ###")  # Stampa per il debug
+            return "Invalid JSON data, expected 'vitonhd' or 'dresscode'", 400
 
+        
         # Saving sketch in filesystem
         model_key = list(json_data_from_req["MODEL"].keys())[0]  # Es: "03191"
-        image_filename = f"{model_key}_00.jpeg"  # Nome file immagine es: "03191_00.jpg"
+        image_filename = f"{model_key}{image_suffix}"  # Nome file immagine es: "03191_00.jpg"
+        print("###################################### NOME SKETCH: ", image_filename)
+        print("###################################### DIRECTORY DOVE LO SALVO: ", image_save_dir)
         image_save_path = os.path.join(image_save_dir, image_filename)
         image.save(image_save_path, format='JPEG')
-
-        with open(text_file_path, "w") as text_file:
-            text_file.write(f"{image_filename} {image_filename}\n")
+        
+        #with open(text_file_path, "w") as text_file:
+        #    text_file.write(f"{image_filename} {image_filename}\n")
 
         
         # Argomenti per eval.main
         sys.argv = [
             'eval.py',
-            '--dataset_path', '/api/model/assets/data/vitonhd',
+            '--dataset_path', dataset_path,
             '--batch_size', '1',
             '--mixed_precision', 'fp16',
             '--num_workers_test', '4',
             '--sketch_cond_rate', '0.2',
-            '--dataset', 'vitonhd',
+            '--dataset', dataset_name,
             '--start_cond_rate', '0.0',
             '--test_order', 'paired'
         ]
         print("### Esecuzione eval.main ###")  # Stampa per il debug
-
+        
         import traceback
 
         # Esecuzione del modello