| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- from flask import Flask, request, jsonify
- import os
- from urllib.request import urlretrieve
- import time
- from image_search import ImageSearchEngine
- app = Flask(__name__)
- # 配置常量
- UPLOAD_FOLDER = 'static/images'
- ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'bmp'}
- # 搜索参数默认值
- TOP_K = 5 # 默认返回的最大结果数
- MIN_SCORE = 0.0 # 默认最小相似度分数
- MAX_SCORE = 100.0 # 默认最大相似度分数
- os.makedirs(UPLOAD_FOLDER, exist_ok=True)
- # 初始化图像搜索引擎
- search_engine = ImageSearchEngine()
- # @app.route("/initialize", methods=["POST"])
- # def initialize():
- # # See FC docs for all the HTTP headers: https://www.alibabacloud.com/help/doc-detail/132044.htm#common-headers
- # request_id = request.headers.get("x-fc-request-id", "")
- # print("FC Initialize Start RequestId: " + request_id)
- # # do your things
- # # Use the following code to get temporary credentials
- # # access_key_id = request.headers['x-fc-access-key-id']
- # # access_key_secret = request.headers['x-fc-access-key-secret']
- # # access_security_token = request.headers['x-fc-security-token']
- # print("FC Initialize End RequestId: " + request_id)
- # return "Function is initialized, request_id: " + request_id + "\n"
- @app.route("/", methods=["POST"])
- def invoke():
- """处理图片上传请求"""
- try:
- # 获取并验证参数
- data = request.get_json()
- if not data:
- return jsonify({'error': '请求必须包含JSON数据'}), 400
-
- print(f"Received data: {data}")
- image_url = data.get('url')
-
- time_base = str(time.time() * 1000)
- image_path = os.path.join(UPLOAD_FOLDER, time_base + os.path.basename(image_url))
- urlretrieve(image_url, image_path)
-
- start_time = time.time()
- search_engine._process_image(image_path)
- os.remove(image_path)
- return jsonify({"vector_spend_time": time.time() - start_time}),200
-
- except Exception as e:
- return jsonify({'error': str(e)}), 500
- if __name__ == '__main__':
- app.run(host='0.0.0.0', port=5001, debug=False)
|
