serializing image and returning only the one modified

api/app.py

@@ -78,25 +78,22 @@ def generate_design():
 
          # 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
+        #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)
+        #print("Files in current directory:")
+        #for file_name in os.listdir(os.getcwd()):
+        #    print(file_name)
 
-        # Invia l'immagine come file scaricabile direttamente dalla directory
-        return send_file(save_path, mimetype='image/jpeg', as_attachment=True)
-    
         # 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)
+        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='generated_image.jpg')
+        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