checking image creation

api/app.py

@@ -77,13 +77,25 @@ def generate_design():
          # Esegui la funzione `main()` di eval.py passando gli argomenti
         final_image = eval.main(args)
 
-        # Converti l'immagine in un buffer di memoria
+         # 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')  # Cambia il formato se necessario
-        img_io.seek(0)  # Riporta il puntatore all'inizio del buffer
+        final_image.save(img_io, 'JPEG')
+        img_io.seek(0)
+
 
-        # Invia l'immagine come file scaricabile
+
+        # Return the image as a file download
         return send_file(img_io, mimetype='image/jpeg', as_attachment=True, attachment_filename='generated_image.jpg')
+
     except Exception as e:
         return str(e), 500