vision模型图片上传没压缩? about chatgpt-web HOT 8 OPEN

from flask import Flask, request, jsonify
import base64
import io
from PIL import Image

app = Flask(__name__)

@app.route('/', defaults={'path': ''}, methods=['POST'])
@app.route('/<path:path>', methods=['POST'])
def compress_image(path):
    data = request.get_json()

    image_data = data.get('image')

    image_bytes = base64.b64decode(image_data)

    image = Image.open(io.BytesIO(image_bytes))

    image = image.convert('RGB')

    quality = 80

    compressed_image_buffer = io.BytesIO()

    image.save(compressed_image_buffer, format='JPEG', optimize=True, quality=quality)

    compressed_image_buffer.seek(0)

    compressed_image_base64 = base64.b64encode(compressed_image_buffer.getvalue()).decode('utf-8')

    return jsonify({'image': compressed_image_base64})

if __name__ == '__main__':
    app.run()