handling images

api/app.py

@@ -11,35 +11,23 @@ from model.src import eval
 app = Flask(__name__)
 CORS(app)
 
-# Directory di output
-output_dir = '/api/output/generato_paired_paired/images'
-image_filename = '03191_00.jpg'
-
 @app.route('/')
 def index():
-
-    # Extract the full URL where the current request was made
-    full_url = request.url
-    
-    # Log the current endpoint to the server logs
-    print(f"Current Endpoint: {full_url}")
-    
     return render_template('index.html')
 
-@app.route('/current-endpoint', methods=['GET'])
-def current_endpoint():
-    # Extract the full URL where the current request was made
-    full_url = request.url
-    
-    # Log the current endpoint to the server logs
-    print(f"Current Endpoint: {full_url}")
-    
-    # Return the current URL to the client
-    return f"Current Endpoint: {full_url}"
-
-@app.route('/generate-design', methods=['GET'])
+@app.route('/generate-design', methods=['POST'])
 def generate_design():
     try:
+
+        # Getting Image
+        image_file = request.files['image']
+        image = Image.open(image_file)
+        save_path = os.path.join('/api/model/assets/data/vitonhd/test/im_sketch', '03191_00.jpeg')
+        image.save(save_path, 'JPEG')
+
+
+
+        
         # Creiamo una lista di argomenti come quelli che passeresti via CLI
         sys.argv = [
             'eval.py',
@@ -56,22 +44,11 @@ def generate_design():
          # 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
+        # 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')