added paired and unpaired

api/app.py

@@ -37,7 +37,7 @@ def current_endpoint():
     # Return the current URL to the client
     return f"Current Endpoint: {full_url}"
 
-@app.route('/generate-design', methods=['GET'])
+@app.route('/generate-design-paired', methods=['GET'])
 def generate_design():
     
     '''# Esegui il comando di generazione
@@ -92,6 +92,67 @@ def generate_design():
 
 
 
+        # Return the image as a file download
+        return send_file(img_io, mimetype='image/jpeg', as_attachment=True, download_name='in memory_image.jpg')
+
+    except Exception as e:
+        return str(e), 500
+
+@app.route('/generate-design-unpaired', methods=['GET'])
+def generate_design():
+    
+    '''# Esegui il comando di generazione
+    command = [
+        'python', '/api/model/src/eval.py',
+        '--dataset_path', '/api/model/assets/data/vitonhd',
+        '--batch_size', '1',
+        '--mixed_precision', 'fp16',
+        '--output_dir', output_dir,
+        '--save_name', 'generato_paired',
+        '--num_workers_test', '4',
+        '--sketch_cond_rate', '0.2',
+        '--dataset', 'vitonhd',
+        '--start_cond_rate', '0.0',
+        '--test_order', 'paired'
+    ]
+    
+    # Esegui il comando senza gestire il JSON, lascia che l'errore venga stampato
+    subprocess.run(command, check=True)'''
+    try:
+        # Creiamo una lista di argomenti come quelli che passeresti via CLI
+        sys.argv = [
+            'eval.py',
+            '--dataset_path', '/api/model/assets/data/vitonhd',
+            '--batch_size', '1',
+            '--mixed_precision', 'fp16',
+            #'--output_dir', '/api/output/generato_paired_paired/images',
+            '--save_name', 'generato_unpaired',
+            '--num_workers_test', '4',
+            '--sketch_cond_rate', '0.2',
+            '--dataset', 'vitonhd',
+            '--start_cond_rate', '0.0',
+            '--test_order', 'unpaired'
+        ]
+
+         # Esegui la funzione `main()` di eval.py passando gli argomenti
+        final_image = eval.main()
+
+         # Save the image to the current directory
+        save_path = os.path.join(os.getcwd(), 'generated_image.jpg')
+        #final_image.save(save_path, 'JPEG')  # Save the image in the current directory
+
+         # Print the contents of the current directory
+        #print("Files in current directory:")
+        #for file_name in os.listdir(os.getcwd()):
+        #    print(file_name)
+
+        # Also save the image to a BytesIO buffer to return via HTTP
+        img_io = io.BytesIO()
+        final_image.save(img_io, 'JPEG')
+        img_io.seek(0)
+
+
+
         # Return the image as a file download
         return send_file(img_io, mimetype='image/jpeg', as_attachment=True, download_name='in memory_image.jpg')
 

api/model/assets/data/vitonhd/test_pairs.txt

@@ -1,2 +1 @@
-12419_00.jpg 01944_00.jpg
 03191_00.jpg 00349_00.jpg