update

docker-compose.yml

@@ -1,5 +1,5 @@
 services:
-  worker1:
+  select:
     build:
       context: .
       dockerfile: Dockerfile
@@ -25,5 +25,5 @@ services:
       replicas: 2
     entrypoint: sh /app/sh/select.sh
 networks:
-  google_net:
+  supply_net:
     name: supply_net

utils/google_ai_studio.py

@@ -1,9 +1,8 @@
 import os
 import time
 import uuid
-from typing import  Optional, Tuple
+from typing import  Optional
 
-import cv2
 import google.generativeai as genai
 import orjson
 import requests
@@ -18,15 +17,6 @@ CACHE_DIR = '/app/cache/'
 # os.environ['https_proxy'] = PROXY_ADDR
 
 class GoogleAI(object):
-    @classmethod
-    def get_video_duration(cls, video_link: str) -> int:
-        cap = cv2.VideoCapture(video_link)
-        if cap.isOpened():
-            rate = cap.get(5)
-            frame_num = cap.get(7)
-            duration = int(frame_num / rate)
-            return duration
-        return 0
 
     @classmethod
     def download_video(cls, video_link: str) -> Optional[str]:

workers/consumption_work.py

@@ -1,5 +1,6 @@
 import asyncio
 import os
+import sys
 import time
 from datetime import datetime
 
@@ -7,7 +8,7 @@ import orjson
 from apscheduler.schedulers.asyncio import AsyncIOScheduler
 from apscheduler.triggers.cron import CronTrigger
 from loguru import logger
-
+sys.path.append('/app')
 from utils.feishu_utils import Feishu
 from utils.google_ai_studio import GoogleAI
 from utils.gpt4o_mimi import GPT4oMini
@@ -15,6 +16,7 @@ from utils.piaoquan import PQ
 from utils.redis import RedisHelper
 
 
+
 class ConsumptionRecommend(object):
     @classmethod
     async def run(cls):

workers/select_work.py

@@ -1,12 +1,13 @@
 import asyncio
 import datetime
+import sys
 import time
 
 import orjson
 from apscheduler.schedulers.asyncio import AsyncIOScheduler
 from apscheduler.triggers.cron import CronTrigger
 from loguru import logger
-
+sys.path.append('/app')
 from utils.feishu_utils import Feishu
 from utils.odps_data import OdpsDataCount
 from utils.redis import RedisHelper