去除影片名称前后的书名号

main
rogersun 4 days ago
parent d5a48332f1
commit 07e6fd457c
  1. 88
      ai/utils/show_process.py

@ -44,50 +44,50 @@ def show_main_process(zz_code=None, day_delta=None):
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)
#
# # 获取排片数据
# 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()
#
# # 预测销售数据
# _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)
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())
# 预测排片数据
_show = result_obj['show'].replace('```', '').replace('csv', '').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)
return True

Loading…
Cancel
Save