在Python Flask中从在线图片URL生成Blurhash占位符

pip install Flask blurhash requests Pillow

import requests
import io
from PIL import Image
import blurhash

def generate_blurhash_from_url(image_url: str, x_components: int = 4, y_components: int = 3) -> str | None:
    """
    从在线图片URL生成Blurhash键值。

    Args:
        image_url: 在线图片的URL。
        x_components: Blurhash的X轴组件数量，影响模糊细节。
        y_components: Blurhash的Y轴组件数量，影响模糊细节。

    Returns:
        生成的Blurhash字符串，如果处理失败则返回None。
    """
    try:
        # 1. 下载图片内容
        response = requests.get(image_url, stream=True, timeout=10)
        response.raise_for_status()  # 检查HTTP请求是否成功

        # 2. 将图片内容读入BytesIO对象
        image_data = io.BytesIO(response.content)

        # 3. 使用Pillow打开图片（blurhash库内部会用到）
        # 这一步是可选的，blurhash.encode可以直接处理BytesIO，
        # 但显式使用Image.open可以提前处理一些图片格式问题或验证图片有效性。
        # 如果图片非常大，直接传递BytesIO可能更高效，让blurhash内部处理。
        # 这里为了演示清晰性，先通过Pillow打开。
        # image = Image.open(image_data)

        # 4. 生成Blurhash
        # blurhash.encode可以直接接受文件类对象
        hash_key = blurhash.encode(image_data, x_components=x_components, y_components=y_components)
        return hash_key

    except requests.exceptions.RequestException as e:
        print(f"下载图片失败: {e}")
        return None
    except Image.UnidentifiedImageError:
        print(f"无法识别的图片格式或无效图片URL: {image_url}")
        return None
    except Exception as e:
        print(f"生成Blurhash时发生未知错误: {e}")
        return None

# 示例使用
if __name__ == "__main__":
    test_image_url = "https://www.w3schools.com/w3css/img_lights.jpg" # 替换为你的在线图片URL
    blurhash_key = generate_blurhash_from_url(test_image_url)
    if blurhash_key:
        print(f"图片URL: {test_image_url}")
        print(f"Blurhash Key: {blurhash_key}")
    else:
        print("未能生成Blurhash Key。")

    test_invalid_url = "http://invalid-url.com/image.jpg"
    blurhash_key_invalid = generate_blurhash_from_url(test_invalid_url)
    if not blurhash_key_invalid:
        print(f"尝试处理无效URL，结果符合预期: {test_invalid_url}")

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

app = Flask(__name__)

@app.route('/generate_blurhash', methods=['GET'])
def generate_blurhash_endpoint():
    """
    Flask路由：接收一个图片URL参数，返回其Blurhash键值。
    示例用法: GET /generate_blurhash?url=https://example.com/image.jpg
    """
    image_url = request.args.get('url')

    if not image_url:
        return jsonify({"error": "请提供'url'参数"}), 400

    try:
        # 下载图片内容
        response = requests.get(image_url, stream=True, timeout=10)
        response.raise_for_status()

        # 将图片内容读入BytesIO对象
        image_data = io.BytesIO(response.content)

        # 生成Blurhash
        # 可以通过请求参数调整x_components和y_components
        x_components = int(request.args.get('x', 4))
        y_components = int(request.args.get('y', 3))

        hash_key = blurhash.encode(image_data, x_components=x_components, y_components=y_components)

        return jsonify({"image_url": image_url, "blurhash_key": hash_key}), 200

    except requests.exceptions.Timeout:
        return jsonify({"error": "下载图片超时"}), 408
    except requests.exceptions.RequestException as e:
        return jsonify({"error": f"下载图片失败: {str(e)}"}), 500
    except Image.UnidentifiedImageError:
        return jsonify({"error": "无法识别的图片格式或无效图片URL"}), 400
    except ValueError:
        return jsonify({"error": "x或y组件参数无效，必须是整数"}), 400
    except Exception as e:
        return jsonify({"error": f"服务器内部错误: {str(e)}"}), 500

if __name__ == '__main__':
    # 在生产环境中，请勿使用开发服务器，而应使用WSGI服务器如Gunicorn
    app.run(debug=True, port=5000)