Update app.py

app.py

@@ -1,17 +1,58 @@
-from flask import Flask, request, jsonify
-import requests
+from flask import Flask, request, send_from_directory
+from PIL import Image
+import os
+import tempfile
 
 app = Flask(__name__)
 
+def convert_images(image_paths, height, width):
+    converted_files = []
+    for image_path in image_paths:
+        img = Image.open(image_path).convert("RGBA")
+        orig_width, orig_height = img.size
+
+        if orig_height > height or orig_width > width:
+            if orig_height / height > orig_width / width:
+                new_height = height
+                new_width = int((new_height / orig_height) * orig_width)
+            else:
+                new_width = width
+                new_height = int((new_width / orig_width) * orig_height)
+        else:
+            new_width, new_height = orig_width, orig_height
+
+        img = img.resize((new_width, new_height), Image.LANCZOS)
+        
+        base_name = os.path.splitext(os.path.basename(image_path))[0]
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".webp", prefix=base_name + "_")
+        img.save(temp_file, format='WEBP')
+        temp_file.close()
+        
+        converted_files.append(temp_file.name)
+    
+    return converted_files
+
 @app.route('/convert', methods=['POST'])
 def convert():
-    files = request.files.getlist('files')
-    height = int(request.form['height'])
-    width = int(request.form['width'])
-    
-    # Process files here (e.g., send to Hugging Face API)
+    files = request.files.getlist('images')
+    height = int(request.form.get('height', 100))
+    width = int(request.form.get('width', 200))
+
+    image_paths = []
+    for file in files:
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.filename)[1])
+        file.save(temp_file.name)
+        image_paths.append(temp_file.name)
+
+    converted_files = convert_images(image_paths, height, width)
+
+    return {
+        "converted_files": [os.path.basename(f) for f in converted_files]
+    }
 
-    return jsonify({'status': 'success', 'message': 'Files processed successfully'})
+@app.route('/download/<filename>', methods=['GET'])
+def download(filename):
+    return send_from_directory(tempfile.gettempdir(), filename)
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=5000)