From 5cc715f6fa0dfd69be8e9f7349c53d2315251542 Mon Sep 17 00:00:00 2001
From: rogersun <roger_sun@outlook.com>
Date: Thu, 18 Jun 2026 14:45:13 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B5=A0=E5=8A=A0=E6=8E=92=E7=89=87=E7=83=AD?=
 =?UTF-8?q?=E5=BA=A6=E5=92=8C=E5=9C=BA=E6=AC=A1=E5=8D=A0=E6=AF=94=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E4=BD=9C=E4=B8=BA=E6=8F=90=E7=A4=BA=E8=AF=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ai/utils/show_process.py | 66 ++++++++++++++++++++--------------------
 1 file changed, 33 insertions(+), 33 deletions(-)

diff --git a/ai/utils/show_process.py b/ai/utils/show_process.py
index 9304b7b..9f1fb17 100644
--- a/ai/utils/show_process.py
+++ b/ai/utils/show_process.py
@@ -33,39 +33,39 @@ def show_main_process(zz_code=None):
         template_show, template_date = get_template_show(cinema, show_date)
         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)
-        # # 获取排片数据
-        # result_obj = json.loads(result)
-        # # 预测排片数据
-        # _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,
-        #         template = template_show,
-        #         temp_date=template_date,
-        #         show=_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)
+        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)
+        # 获取排片数据
+        result_obj = json.loads(result)
+        # 预测排片数据
+        _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,
+                template = template_show,
+                temp_date=template_date,
+                show=_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