赠加排片热度和场次占比数据作为提示词

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