diff --git a/ai/utils/show_process.py b/ai/utils/show_process.py index 24708fc..e603ec7 100644 --- a/ai/utils/show_process.py +++ b/ai/utils/show_process.py @@ -39,55 +39,56 @@ def show_main_process(zz_code=None, day_delta=None): show_ai = ShowAI(cinema, show_date) # template_show, template_date = get_template_show(cinema, show_date) template_show, template_date = get_template_show_tomorrow(cinema) - if get_runtime_val('real_target') == 1: + if str(get_runtime_val('real_target')) == '1': target_show = get_target_show(cinema, show_date) else: target_show = '影厅别名,影厅id,影片别名,本地影片id,语言,放映日期,开始时间,结束时间,片长,场间' prompt = show_ai.general_prompt() - start = datetime.datetime.now() # 开始计时 - result, message, tokens = show_ai.get_show_result_ai() - end = datetime.datetime.now() # 结束计时 - print('prompt:', prompt) - print('result:', result) - print('message:', message) - print('tokens:', tokens) + if str(get_runtime_val('test')) == '0': + start = datetime.datetime.now() # 开始计时 + result, message, tokens = show_ai.get_show_result_ai() + end = datetime.datetime.now() # 结束计时 + print('prompt:', prompt) + print('result:', result) + print('message:', message) + print('tokens:', tokens) - # 获取排片数据 - print('result:', result) - if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']: - result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip()) - else: - result_obj = json.loads(result.strip()) + # 获取排片数据 + print('result:', result) + if model.config_name in ['lmstudio_qwen3.6_35b', 'company_qwen3.5_35b']: + result_obj = json.loads(result.replace('```json', '').replace('}\n```', '}').strip()) + else: + result_obj = json.loads(result.strip()) - # 预测排片数据 - _show = result_obj['show'].replace('```', '').replace('csv', '').strip() + # 预测排片数据 + _show = result_obj['show'].replace('```', '').replace('csv', '').strip() - # 预测销售数据 - _sales = str(result_obj['income']).strip() + # 预测销售数据 + _sales = str(result_obj['income']).strip() - # 处理返回结果 - try: - AiShow.objects.create( - cinema=cinema.name, - zz_code=cinema.zz_code, - show_date=show_date, - is_ai_show=True, - is_manual_test=False, - ai_model=model.config_name, - template=template_show, - temp_date=template_date, - target=target_show, - show=data_cleansing(_show), - sales=_sales, - prompt=prompt, - result=result, - message=message, - take_times=int((end - start).seconds), - take_tokens=tokens, - prompt_version=prompt_ver, - ) - except Exception as e: - print(e) + # 处理返回结果 + try: + AiShow.objects.create( + cinema=cinema.name, + zz_code=cinema.zz_code, + show_date=show_date, + is_ai_show=True, + is_manual_test=False, + ai_model=model.config_name, + template=template_show, + temp_date=template_date, + target=target_show, + show=data_cleansing(_show), + sales=_sales, + prompt=prompt, + result=result, + message=message, + take_times=int((end - start).seconds), + take_tokens=tokens, + prompt_version=prompt_ver, + ) + except Exception as e: + print(e) return True