From 363074317472a6f6ece9106f77aa269432e4506e Mon Sep 17 00:00:00 2001 From: "majiahui@haimaqingfan.com" Date: Thu, 23 Nov 2023 15:15:38 +0800 Subject: [PATCH] =?UTF-8?q?ai=E5=88=9B=E6=84=8F=E5=B7=A5=E5=8E=82=E9=9C=80?= =?UTF-8?q?=E8=A6=81=E7=9A=84=E6=8E=A5=E5=8F=A3=E7=BB=84=E5=90=88?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .idea/.gitignore | 8 + .idea/ai_creative_workshop.iml | 8 + .idea/deployment.xml | 742 +++++++++++++++++++++++++ .idea/inspectionProfiles/Project_Default.xml | 19 + .idea/inspectionProfiles/profiles_settings.xml | 6 + .idea/misc.xml | 4 + .idea/modules.xml | 8 + .idea/vcs.xml | 6 + README.md | 0 gunicorn_config.py | 21 + main.py | 294 ++++++++++ run_api.sh | 1 + 12 files changed, 1117 insertions(+) create mode 100644 .idea/.gitignore create mode 100644 .idea/ai_creative_workshop.iml create mode 100644 .idea/deployment.xml create mode 100644 .idea/inspectionProfiles/Project_Default.xml create mode 100644 .idea/inspectionProfiles/profiles_settings.xml create mode 100644 .idea/misc.xml create mode 100644 .idea/modules.xml create mode 100644 .idea/vcs.xml create mode 100644 README.md create mode 100644 gunicorn_config.py create mode 100644 main.py create mode 100644 run_api.sh diff --git a/.idea/.gitignore b/.idea/.gitignore new file mode 100644 index 0000000..35410ca --- /dev/null +++ b/.idea/.gitignore @@ -0,0 +1,8 @@ +# 默认忽略的文件 +/shelf/ +/workspace.xml +# 基于编辑器的 HTTP 客户端请求 +/httpRequests/ +# Datasource local storage ignored files +/dataSources/ +/dataSources.local.xml diff --git a/.idea/ai_creative_workshop.iml b/.idea/ai_creative_workshop.iml new file mode 100644 index 0000000..5fee449 --- /dev/null +++ b/.idea/ai_creative_workshop.iml @@ -0,0 +1,8 @@ + + + + + + + + \ No newline at end of file diff --git a/.idea/deployment.xml b/.idea/deployment.xml new file mode 100644 index 0000000..81dcf97 --- /dev/null +++ b/.idea/deployment.xml @@ -0,0 +1,742 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml new file mode 100644 index 0000000..50c8dac --- /dev/null +++ b/.idea/inspectionProfiles/Project_Default.xml @@ -0,0 +1,19 @@ + + + + \ No newline at end of file diff --git a/.idea/inspectionProfiles/profiles_settings.xml b/.idea/inspectionProfiles/profiles_settings.xml new file mode 100644 index 0000000..105ce2d --- /dev/null +++ b/.idea/inspectionProfiles/profiles_settings.xml @@ -0,0 +1,6 @@ + + + + \ No newline at end of file diff --git a/.idea/misc.xml b/.idea/misc.xml new file mode 100644 index 0000000..5cf0187 --- /dev/null +++ b/.idea/misc.xml @@ -0,0 +1,4 @@ + + + + \ No newline at end of file diff --git a/.idea/modules.xml b/.idea/modules.xml new file mode 100644 index 0000000..b535dcc --- /dev/null +++ b/.idea/modules.xml @@ -0,0 +1,8 @@ + + + + + + + + \ No newline at end of file diff --git a/.idea/vcs.xml b/.idea/vcs.xml new file mode 100644 index 0000000..94a25f7 --- /dev/null +++ b/.idea/vcs.xml @@ -0,0 +1,6 @@ + + + + + + \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..e69de29 diff --git a/gunicorn_config.py b/gunicorn_config.py new file mode 100644 index 0000000..e874945 --- /dev/null +++ b/gunicorn_config.py @@ -0,0 +1,21 @@ +# 并行工作线程数 +workers = 1 +# 监听内网端口5000【按需要更改】 +bind = '0.0.0.0:19000' + +loglevel = 'debug' + +worker_class = "gevent" +# 设置守护进程【关闭连接时，程序仍在运行】 +daemon = True +# 设置超时时间120s，默认为30s。按自己的需求进行设置 +timeout = 120 +# 设置访问日志和错误信息日志路径 +accesslog = './logs/acess.log' +errorlog = './logs/error.log' +# access_log_format = '%(h) - %(t)s - %(u)s - %(s)s %(H)s' +# errorlog = '-' # 记录到标准输出 + + +# 设置最大并发量 +worker_connections = 20000 diff --git a/main.py b/main.py new file mode 100644 index 0000000..e53a734 --- /dev/null +++ b/main.py @@ -0,0 +1,294 @@ +import os +from flask import Flask, render_template, request, redirect, url_for, jsonify +from werkzeug.utils import secure_filename +app = Flask(__name__) +import time +import re +import requests +import uuid + +# 上传文件存储目录 +UPLOAD_FOLDER = '/home/majiahui/project/ai_creative_workshop/uploads' +app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER + +# 正则表达式 +RE_CHINA_NUMS = "[1-9].(.*)" +# 允许的文件类型 +ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'} + + +prompt_picture = { + "1": "图中的商品：{}，有什么突出亮点和卖点，请分条列举出来，要求亮点或者卖点要用一个词总结，冒号后面在进行解释，例如：1. 时尚黑色：图中的鞋子是黑色的，符合时尚潮流，适合不同场合的穿搭。", + "2": "图中的商品：{}，有什么亮点，写一段营销话语", + "3": "图中的商品：{}，有以下亮点：\n{}\n根据这些优势亮点，写一段营销文本让商品卖的更好", + "4": "图中的商品：{}，有哪些不足之处可以改进？", + "5": "图中{}的渲染图做哪些调整可以更吸引消费者", + "6": "根据图中的商品：{}，生成一个商品名称，要求商品名称格式中包含的信息（有品牌名，有产品名，有细分产品种类词，比如猫砂，篮球鞋等，有三到五个卖点和形容词）", +} + +# prompt_text = { +# "1": "图中{}有什么突出亮点，请列举出来", +# "2": "图中{}有什么亮点，写一段营销话语", +# "3": "图中{}有以下亮点：\n{}\n根据这些优势亮点，写一段营销文本让商品买的更好", +# "4": "图中{}有哪些不足之处可以改进？", +# "5": "图中{}的渲染图做哪些调整可以更吸引消费者", +# "5": "图中{}的渲染图做哪些调整可以更吸引消费者", +# } + + +def dialog_line_parse(url, text): + """ + 将数据输入模型进行分析并输出结果 + :param url: 模型url + :param text: 进入模型的数据 + :return: 模型返回结果 + """ + + response = requests.post( + url, + json=text, + timeout=1000 + ) + if response.status_code == 200: + return response.json() + else: + # logger.error( + # "【{}】 Failed to get a proper response from remote " + # "server. Status Code: {}. Response: {}" + # "".format(url, response.status_code, response.text) + # ) + print("【{}】 Failed to get a proper response from remote " + "server. Status Code: {}. Response: {}" + "".format(url, response.status_code, response.text)) + print(text) + return {} + + +class log: + def __init__(self): + pass + + def log(*args, **kwargs): + format = '%Y/%m/%d-%H:%M:%S' + format_h = '%Y-%m-%d' + value = time.localtime(int(time.time())) + dt = time.strftime(format, value) + dt_log_file = time.strftime(format_h, value) + log_file = 'log_file/access-%s' % dt_log_file + ".log" + if not os.path.exists(log_file): + with open(os.path.join(log_file), 'w', encoding='utf-8') as f: + print(dt, *args, file=f, **kwargs) + else: + with open(os.path.join(log_file), 'a+', encoding='utf-8') as f: + print(dt, *args, file=f, **kwargs) + + +# 检查文件扩展名 +def allowed_file(filename): + return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS + + +def picyure_model_predict(image_path, prompt): + # query = tokenizer.from_list_format([ + # {'image': image_path}, + # {'text': prompt}, + # ]) + # + # response, history = model.chat(tokenizer, query=query, history=None) + # return response + + url = "http://192.168.31.74:19001/predict" + data = { + "path_list": [image_path], + "prompt": prompt + } + result = dialog_line_parse(url, data)["data"] + return result + + +def picture_main(path_list, commodity, type, additional): + if type == "1": + result_list_len = False + dan_result_geshi = True + dan_result_geshi_maohao = True + + prompy_text = prompt_picture[type] + prompy_text = prompy_text.format(commodity) + while True: + result = picyure_model_predict(path_list[0], prompy_text) + result_list = str(result).split("\n") + result_list = [i for i in result_list if i != ""] + if len(result_list) > 3: + result_list_len = True + for i in result_list: + response_re = re.findall(RE_CHINA_NUMS, i) + if response_re == []: + dan_result_geshi = False + continue + if "：" not in i: + dan_result_geshi_maohao = False + continue + if result_list_len == True and dan_result_geshi == True and dan_result_geshi_maohao == True: + break + + maidian_list = [] + for i in result_list: + response_re = re.findall(RE_CHINA_NUMS, i) + guanjianci = response_re[0].split("：") + maidian_list.append([i, guanjianci]) + + return maidian_list + + elif type == "2": + prompy_text = prompt_picture[type] + prompy_text = prompy_text.format(commodity, additional) + return_list = [] + for path in path_list: + result = picyure_model_predict(path, prompy_text) + return_list.append(result) + return return_list + # + elif type == "3": + prompy_text = prompt_picture[type] + prompy_text = prompy_text.format(commodity, additional) + return_list = [] + for path in path_list: + result = picyure_model_predict(path, prompy_text) + return_list.append(result) + return return_list + # + elif type == "4": + prompy_text = prompt_picture[type] + prompy_text = prompy_text.format(commodity, additional) + return_list = [] + for path in path_list: + for i in range(5): + result = picyure_model_predict(path, prompy_text) + return_list.append(result) + return return_list + + elif type == "5": + return_list = [] + prompy_text = prompt_picture[type] + prompy_text = prompy_text.format(commodity) + result_list_type = False + + for path in path_list: + while True: + if result_list_type == True: + break + result = picyure_model_predict(path, prompy_text) + result_list = str(result).split("\n") + result_list = [i for i in result_list if i != ""] + result_list_new = [] + for i in result_list: + response_re = re.findall(RE_CHINA_NUMS, i) + if response_re == []: + continue + else: + result_list_new.append(i) + if result_list_new != []: + result_list_type = True + return_list.append(result_list_new) + + return return_list + + elif type == "6": + return_list = [] + prompy_text = prompt_picture[type] + prompy_text = prompy_text.format(commodity) + for path in path_list: + for i in range(5): + result = picyure_model_predict(path, prompy_text) + return_list.append(result) + return return_list + + elif type == "7": + prompy_text = additional + return_list = [] + for path in path_list: + result = picyure_model_predict(path, prompy_text) + return_list.append(result) + return return_list + + else: + return "1111" + +# 文件上传处理 +@app.route('/vl_chat', methods=['POST']) +def upload_file(): + + file0 = request.files.get('file0') + file1 = request.files.get('file1') + file2 = request.files.get('file2') + file3 = request.files.get('file3') + file4 = request.files.get('file4') + file5 = request.files.get('file5') + commodity = request.form.get('commodity') + type_str = request.form.get('type') + additional = request.form.get("additional") + file_list = [file0, file1, file2, file3, file4, file5] + + # if commodity == False or type_str == False and file0 == False: + # return str(400) + try: + assert file0 + except: + return_text = {"texts": "没有主图", "probabilities": None, "status_code": 400} + return jsonify(return_text) + + try: + assert commodity + except: + return_text = {"texts": "没有商品类型", "probabilities": None, "status_code": 400} + return jsonify(return_text) + + try: + assert type_str + except: + return_text = {"texts": "没有生成类型", "probabilities": None, "status_code": 400} + return jsonify(return_text) + + path_list = [] + for file in file_list: + if file and allowed_file(file.filename): + filename = secure_filename(file.filename) + kuozhan = filename.split(".")[-1] + uuid_picture = str(uuid.uuid1()) + filename = ".".join([uuid_picture, kuozhan]) + path = os.path.join(app.config['UPLOAD_FOLDER'], filename) + file.save(path) + path_list.append(path) + + # 业务逻辑 + try: + type_list = str(type_str).split(",") + + result = [] + for type_dan in type_list: + print("type:", type_dan) + result_dan = picture_main(path_list, commodity, type_dan, additional) + result.append(result_dan) + return_text = {"texts": result, "probabilities": None, "status_code": 200} + except: + return_text = {"texts": "运算出错", "probabilities": None, "status_code": 400} + log.log('start at', + 'filename:{}, commodity:{}, type:{}, additional:{}, result:{}'.format( + str(path_list), commodity, str(type_str), additional, return_text)) + return jsonify(return_text) + + + +# 无文件上传 +# @app.route('/chat', methods=['POST']) +# def upload_file(): +# +# type = request.files.get('type') +# describe = request.form.get("describe") +# advantage = request.form.get("dadvantage") +# +# return "1" + + +if __name__ == "__main__": + app.run(host="0.0.0.0", port=19000, threaded=True) diff --git a/run_api.sh b/run_api.sh new file mode 100644 index 0000000..b204ebd --- /dev/null +++ b/run_api.sh @@ -0,0 +1 @@ +gunicorn main:app -c gunicorn_config.py \ No newline at end of file