added post and get req

api/app.py

@@ -11,52 +11,12 @@ 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=['POST'])
+@app.route('/generate-design', methods=['GET', 'POST'])
 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', 'unpaired'
-    ]
-    
-    # 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 = [
@@ -75,28 +35,16 @@ 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
         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')
+        return send_file(img_io, mimetype='image/jpeg', as_attachment=True, download_name='inmemory_image.jpg')
 
     except Exception as e:
         return str(e), 500
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860)