zhangyong 6 months ago
parent
commit
fe02cf9afc
1 changed files with 28 additions and 31 deletions
  1. 28 31
      google_ai/generativeai_video.py

+ 28 - 31
google_ai/generativeai_video.py

@@ -79,15 +79,12 @@ class VideoAnalyzer:
 
         return model
     async def analyze_video(self, model, questions, sample_data):
-            chat_session = model.start_chat(
-                history=[
-                ]
-            )
+            chat_session = model.start_chat(history=[])
             message_content = {
                 "parts": [
                     self.video_file,
                     "你是一个专业的视频分析师,负责根据访问的视频文件回答用户的所有问题\\n"+questions +
-                    "输出返回格式样例:\n"+ str(sample_data)
+                    "输出返回格式样例:\n" + str(sample_data)
 
                 ]
             }
@@ -114,36 +111,36 @@ async def main(video_path):
     # 创建缓存模型
     model =await analyzer.create_cache()
     sample_data = {
-        "基础信息": {
-            "视觉/音乐/文字": "",
-            "内容选题": "",
-            "视频主题": ""
+        "一、基础信息": {
+            "1.视觉/音乐/文字": "",
+            "2.内容选题": "",
+            "3.视频主题": ""
         },
-        "主体和场景": {
-            "视频主体": "",
-            "视频场景": []
+        "二、主体和场景": {
+            "1.视频主体": "",
+            "2.视频场景": []
         },
-        "情感与风格": {},
-        "视频传播性与观众": {
-            "片尾引导": {},
-            "传播性判断": "",
-            "观众画像": {}
+        "三、情感与风格": {},
+        "四、视频传播性与观众": {
+            "1.片尾引导": {},
+            "2.传播性判断": "",
+            "3.观众画像": {}
         },
-        "音画细节": {
-            "音频细节": {},
-            "视频水印": {},
-            "视频字幕": {},
-            "视频口播": ""
+        "五、音画细节": {
+            "1.音频细节": {},
+            "2.视频水印": {},
+            "3.视频字幕": {},
+            "4.视频口播": ""
         },
-        "人物与场景": {
-            "知名人物": {},
-            "人物年龄段": "",
-            "场景描述": []
+        "六、人物与场景": {
+            "1.知名人物": {},
+            "2.人物年龄段": "",
+            "3.场景描述": []
         },
-        "时效性与分类": {
-            "时效性": {},
-            "视频一级分类": "",
-            "二级分类": ""
+        "七、时效性与分类": {
+            "1.时效性": {},
+            "2.视频一级分类": "",
+            "3.二级分类": ""
         }
     }
 
@@ -365,5 +362,5 @@ if __name__ == "__main__":
     os.environ["http_proxy"] = proxy_url
     os.environ["https_proxy"] = proxy_url
     # video_path = 'http://temp.yishihui.com/longvideo/transcode/video/vpc/20240926/66510681PACx7zsp2wDBHJlicE.mp4'
-    video_path = 'http://temp.yishihui.com/jq_oss/video/2024092618393680952.mp4'
+    video_path = 'http://temp.yishihui.com/longvideo/transcode/video/vpc/20240605/68754804SJz5E9JNe5hAdSkRwF.mp4'
     asyncio.run(main(video_path))