Update xhs.py

xhs.py

@@ -1,9 +1,29 @@
 from flask import Flask, request, jsonify
 import re
 import requests
+import os
+from functools import wraps
 
 app = Flask(__name__)
 
+# 从环境变量获取API密钥，如果不存在则使用默认值
+API_KEY = os.environ.get("XHS_API_KEY", "default_secret_key")
+
+# API密钥验证装饰器
+def require_api_key(f):
+    @wraps(f)
+    def decorated_function(*args, **kwargs):
+        # 从请求头获取API密钥
+        api_key = request.headers.get('x-api-key')
+        
+        # 验证API密钥
+        if api_key and api_key == API_KEY:
+            return f(*args, **kwargs)
+        else:
+            return jsonify({"status": "error", "message": "无效的API密钥"}), 401
+    
+    return decorated_function
+
 def extract_url(text):
     """从文本中提取小红书链接（短链接或完整链接）"""
     # 匹配形如 http://xhslink.com/xxxx 的短链接
@@ -69,6 +89,7 @@ def convert_links(text):
     return {"status": "success", "message": "链接转换成功", "results": results}
 
 @app.route('/api/convert', methods=['POST'])
+@require_api_key
 def api_convert():
     """API端点，接收POST请求并转换链接"""
     if not request.is_json:
@@ -85,6 +106,7 @@ def api_convert():
     return jsonify(result)
 
 @app.route('/api/health', methods=['GET'])
+@require_api_key
 def health_check():
     """健康检查端点"""
     return jsonify({"status": "ok", "message": "服务正常运行"})
@@ -108,6 +130,13 @@ def home():
             <p>这是一个将小红书分享链接转换为真实可访问链接的API服务。</p>
             
             <h2>API使用说明</h2>
+            <h3>认证</h3>
+            <p>所有API请求都需要在请求头中包含有效的API密钥：</p>
+            <pre>
+x-api-key: [您的API密钥]
+            </pre>
+            <p>API密钥可以通过环境变量 <code>XHS_API_KEY</code> 设置</p>
+            
             <h3>转换链接</h3>
             <p><strong>端点:</strong> <code>/api/convert</code></p>
             <p><strong>方法:</strong> POST</p>
@@ -148,4 +177,4 @@ def home():
 
 if __name__ == '__main__':
     # 默认运行在7860端口，适配Hugging Face Spaces
-    app.run(host='0.0.0.0', port=7860, debug=False,threaded=True) 
+    app.run(host='0.0.0.0', port=7860, debug=False, threaded=True)