fixed image path

api/app.py

@@ -22,7 +22,7 @@ def index():
 @app.route('/generate-design', methods=['GET','POST'])
 def generate_design():
 
-    image_save_dir = "/api/model/assets/data/vitonhd/test/im_sketch"
+    #image_save_dir = "/api/model/assets/data/vitonhd/test/im_sketch"
     text_file_path = "/api/model/assets/data/vitonhd/test/test_paired.txt"
 
     try:
@@ -60,8 +60,9 @@ def generate_design():
             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')
+            dataset_path = '/api/model/assets/data/dresscode'
+            sketch_path = f'/api/model/assets/data/dresscode/{body_part}'
+            image_save_dir = os.path.join(sketch_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