init

Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+COPY . .
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+EXPOSE 7860
+CMD ["python", "app.py"]

README.md

@@ -1,5 +1,5 @@
 ---
-title: Thumbor
+title: 获取图片的主色调
 emoji: 📊
 colorFrom: indigo
 colorTo: red
@@ -8,4 +8,4 @@ pinned: false
 short_description: thumbor 是 globo.com 提供的开源照片缩略图服务
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Use `/color?url=https://your-image.jpg` to get dominant color

app.py

@@ -0,0 +1,45 @@
+from flask import Flask, request, jsonify
+from PIL import Image
+import requests
+from io import BytesIO
+from sklearn.cluster import KMeans
+import numpy as np
+
+app = Flask(__name__)
+
+def extract_dominant_color(image_url):
+    try:
+        response = requests.get(image_url)
+        img = Image.open(BytesIO(response.content)).convert('RGB')
+        img = img.resize((100, 100))  # 降低像素加快处理
+        pixels = np.array(img).reshape(-1, 3)
+
+        kmeans = KMeans(n_clusters=1, random_state=0).fit(pixels)
+        dominant_color = kmeans.cluster_centers_[0].astype(int)
+        hex_color = '#{:02x}{:02x}{:02x}'.format(*dominant_color)
+
+        return hex_color
+    except Exception as e:
+        return str(e)
+
+@app.route('/')
+def index():
+    return jsonify({
+        "message": "Use /color?url=https://your-image.jpg to get dominant color."
+    })
+
+@app.route('/color')
+def get_color():
+    image_url = request.args.get('url')
+    if not image_url:
+        return jsonify({'error': 'Missing image URL ?url=xxx'}), 400
+
+    hex_color = extract_dominant_color(image_url)
+
+    if hex_color.startswith('#'):
+        return jsonify({'hex': hex_color})
+    else:
+        return jsonify({'error': 'Failed to extract color', 'detail': hex_color}), 500
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)

requirements.txt

@@ -0,0 +1,4 @@
+flask
+requests
+pillow
+scikit-learn

thumbor.conf

@@ -0,0 +1,10 @@
+# thumbor.conf（只保留必要项）
+QUALITY = 85
+PORT = 7860
+ALLOW_UNSAFE_URL = True
+MAX_AGE = 86400
+
+# 开启 dominant_color 插件
+FILTERS = [
+  'thumbor.filters.dominant_color'
+]