From b939d2c61312c56d098eeb152647a348a1a2985c Mon Sep 17 00:00:00 2001
From: daMustermann <julian.neumann@gmail.com>
Date: Sun, 12 Jan 2025 05:04:17 +0100
Subject: [PATCH] Enhance image processing: Add functionality to resize and
 convert uploaded images to WebP format

---
 app.py           | 48 +++++++++++++++++++++++++++++++++++++++++++-----
 requirements.txt |  3 ++-
 2 files changed, 45 insertions(+), 6 deletions(-)

diff --git a/app.py b/app.py
index df36d6d..471da53 100644
--- a/app.py
+++ b/app.py
@@ -8,6 +8,8 @@
 from sqlalchemy import or_
 from flask_wtf.csrf import generate_csrf
 from flask_wtf import FlaskForm
+from PIL import Image
+import io
 
 def create_app(config_class=Config):
     app = Flask(__name__)
@@ -25,6 +27,29 @@ def allowed_file(filename):
         return '.' in filename and \
                filename.rsplit('.', 1)[1].lower() in app.config['ALLOWED_EXTENSIONS']
 
+    def process_image(image_file):
+        # Read the image
+        image = Image.open(image_file)
+        
+        # Convert to RGB if needed (for PNG transparency)
+        if image.mode in ('RGBA', 'LA'):
+            background = Image.new('RGB', image.size, (255, 255, 255))
+            background.paste(image, mask=image.split()[-1])
+            image = background
+        
+        # Calculate new dimensions
+        aspect_ratio = image.size[0] / image.size[1]
+        new_height = min(800, image.size[1])  # Updated to 800px
+        new_width = int(aspect_ratio * new_height)
+        
+        # Resize image
+        image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+        
+        # Save as WebP
+        webp_buffer = io.BytesIO()
+        image.save(webp_buffer, format='WebP', quality=85, method=6)
+        return webp_buffer.getvalue()
+
     @app.route('/', methods=['GET', 'POST'])
     def index():
         if request.method == 'POST':
@@ -35,7 +60,7 @@ def index():
                 image_file = request.files['image']
                 
                 if image_file and allowed_file(image_file.filename):
-                    image_data = image_file.read()
+                    image_data = process_image(image_file)
                 else:
                     image_data = None
 
@@ -100,16 +125,29 @@ def edit_item(item_id):
                 item.description = request.form['description']
                 item.location = request.form['location']
                 item.quantity = int(request.form['quantity'])
-                image_file = request.files['image']
-                if image_file:
-                    item.image = image_file.read()
+                
+                # Add debug logging
+                print("Processing image upload in edit route")
+                image_file = request.files.get('image')
+                if image_file and image_file.filename:
+                    print(f"Image file received: {image_file.filename}")
+                    if allowed_file(image_file.filename):
+                        print("File type allowed, processing image")
+                        image_data = process_image(image_file)
+                        item.image = image_data
+                        print("Image processed and saved")
+                    else:
+                        print(f"File type not allowed: {image_file.filename}")
+                
                 db.session.commit()
                 flash('Item updated successfully!', 'success')
                 return redirect(url_for('item', item_id=item.id))
             except Exception as e:
-                print(f"Error updating item: {e}")
+                print(f"Error updating item: {str(e)}")
+                db.session.rollback()
                 flash('Error updating item. Please try again.', 'danger')
                 return redirect(url_for('edit_item', item_id=item.id))
+        
         return render_template('edit_item.html', item=item)
 
     @app.route('/item/<int:item_id>/update_quantity', methods=['POST'])
diff --git a/requirements.txt b/requirements.txt
index 801b245..b94270b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,5 @@ Flask-SQLAlchemy
 Flask-WTF
 python-dotenv
 Werkzeug
-alembic
\ No newline at end of file
+alembic
+Pillow==10.1.0
\ No newline at end of file