|
from flask import Flask, request, jsonify |
|
from flask_cors import CORS |
|
from PIL import Image |
|
|
|
|
|
myapp = Flask(__name__) |
|
|
|
|
|
CORS(myapp) |
|
|
|
@myapp.route('/') |
|
def index(): |
|
return "Welcome to the AI Model Upscale!" |
|
|
|
@myapp.route('/resize', methods=['POST']) |
|
def resize_image(): |
|
data = request.json |
|
input_path = data.get('input_path') |
|
output_path = data.get('output_path') |
|
width = data.get('width') |
|
height = data.get('height') |
|
|
|
|
|
with Image.open(input_path) as img: |
|
resized_img = img.resize((width, height), Image.LANCZOS) |
|
resized_img.save(output_path) |
|
|
|
return jsonify({'message': 'Image resized successfully', 'output_path': output_path}) |
|
|
|
if __name__ == "__main__": |
|
myapp.run(host='0.0.0.0', port=5000) |