在Python Flask中将在线图片URL转换为Blurhash编码

pip install Flask requests blurhash

import blurhash

# 假设 'image.jpg' 是一个本地图片文件
with open('image.jpg', 'rb') as image_file:
    # x_components 和 y_components 控制生成模糊图像的精度
    hash_string = blurhash.encode(image_file, x_components=4, y_components=3)
    print(f"本地图片 Blurhash: {hash_string}")

import requests
import io
import blurhash

def get_blurhash_from_url(image_url: str, x_components: int = 4, y_components: int = 3) -> str:
    """
    从在线图片URL生成Blurhash编码。
    """
    try:
        response = requests.get(image_url, stream=True, timeout=10)
        response.raise_for_status() # 检查HTTP请求是否成功

        # 将图片内容读取到内存文件对象中
        image_data = io.BytesIO(response.content)

        # 使用blurhash库进行编码
        hash_string = blurhash.encode(image_data, x_components=x_components, y_components=y_components)
        return hash_string
    except requests.exceptions.RequestException as e:
        print(f"请求图片失败: {e}")
        return None
    except Exception as e:
        print(f"Blurhash编码失败: {e}")
        return None

# 示例使用
online_image_url = "https://example.com/some-image.jpg" # 替换为实际的图片URL
# blurhash_key = get_blurhash_from_url(online_image_url)
# if blurhash_key:
#     print(f"在线图片 Blurhash: {blurhash_key}")

from flask import Flask, request, jsonify
import requests
import io
import blurhash

app = Flask(__name__)

@app.route('/generate_blurhash', methods=['GET'])
def generate_blurhash():
    """
    API端点：接收图片URL，返回其Blurhash编码。
    示例: GET /generate_blurhash?url=https://example.com/your-image.jpg
    """
    image_url = request.args.get('url')
    if not image_url:
        return jsonify({"error": "缺少 'url' 参数"}), 400

    try:
        # 使用requests下载图片内容
        response = requests.get(image_url, stream=True, timeout=15)
        response.raise_for_status() # 检查HTTP请求是否成功

        # 验证内容类型是否为图片
        content_type = response.headers.get('Content-Type', '')
        if not content_type.startswith('image/'):
            return jsonify({"error": "URL指向的不是图片资源"}), 415

        # 将图片二进制数据封装到io.BytesIO对象中
        image_data = io.BytesIO(response.content)

        # 使用blurhash进行编码，可自定义组件数量
        # 默认使用 4x3 的组件数量，可以在生产环境中根据需求调整
        hash_string = blurhash.encode(image_data, x_components=4, y_components=3)

        return jsonify({"blurhash": hash_string}), 200

    except requests.exceptions.Timeout:
        return jsonify({"error": "下载图片超时"}), 408
    except requests.exceptions.RequestException as e:
        return jsonify({"error": f"请求图片失败: {str(e)}"}), 502 # Bad Gateway
    except Exception as e:
        # 捕获其他可能的编码错误，例如图片格式不支持
        return jsonify({"error": f"Blurhash编码失败: {str(e)}"}), 500

if __name__ == '__main__':
    # 运行Flask应用，生产环境请使用Gunicorn等WSGI服务器
    app.run(debug=True, port=5000)