zhangyong há 3 meses atrás
pai
commit
b41a81ae4c
1 ficheiros alterados com 2 adições e 0 exclusões
  1. 2 0
      utils/google_ai_studio.py

+ 2 - 0
utils/google_ai_studio.py

@@ -36,6 +36,7 @@ class GoogleAI(object):
 
     @classmethod
     def run(cls, api_key, video_url):
+        video_path = None
         try:
             genai.configure(api_key=api_key)
             video_path = cls.download_video(video_link=video_url)
@@ -76,6 +77,7 @@ class GoogleAI(object):
             return text
         except Exception as e:
             logger.error(f"[内容分析] 处理异常,异常信息{e}")
+            os.remove(video_path)
             return