liqian 2 years ago
parent
commit
d5b6bb6fdb
2 changed files with 11 additions and 4 deletions
  1. 7 4
      app.py
  2. 4 0
      gpt_process.py

+ 7 - 4
app.py

@@ -31,9 +31,9 @@ class FlaskApp(Flask):
         self._activate_background_job()
 
     def _activate_background_job(self):
-        log_.info("_activate_background_job")
+        log_.info("debug: _activate_background_job")
         def run_job():
-            log_.info("run_job")
+            log_.info("debug: run_job")
             title_generate_main()
 
         t1 = threading.Thread(target=run_job)
@@ -45,12 +45,12 @@ app = FlaskApp(__name__)
 
 @app.route('/healthcheck')
 def health_check():
-    log_.info('ok!')
+    log_.info('debug: ok!')
     return 'ok!'
 
 
 def title_generate_main():
-    log_.info(title_generate_main)
+    log_.info("debug: title_generate_main")
     # 初始化client
     mq_client = MQClient(
         # 设置HTTP协议客户端接入点
@@ -93,9 +93,12 @@ def title_generate_main():
                     'properties': msg.properties
                 }
                 log_.info(receive_info)
+                log_.info(f"debug: {receive_info}")
                 video_id = msg.message_body['videoId']
                 video_path = msg.message_body['videoPath']
                 try:
+                    ppp = {'videoId': video_id, 'videoPath': video_path}
+                    log_.info(f"debug: {ppp}")
                     title = title_generate(video_id=video_id, video_path=video_path)
                     log_.info({'titleGenerate': {'videoId': video_id, 'title': title}})
 

+ 4 - 0
gpt_process.py

@@ -46,17 +46,21 @@ def title_generate(video_id, video_path):
     :return:
     """
     # 1. 下载视频
+    log_.info(f"debug: title_generate 1")
     video_file_path = download_video(video_path=video_path, video_id=video_id, download_folder='videos')
     log_.info(f"video_file_path = {video_file_path}")
     # 2. 获取视频中的音频
+    log_.info(f"debug: title_generate 2")
     audio_path = get_wav(video_path=video_file_path)
     log_.info(f"audio_path = {audio_path}")
     # 3. asr
+    log_.info(f"debug: title_generate 3")
     dialogue_path, asr_res = call_asr(audio_path=audio_path)
     log_.info({
         'asrResult': {'videoId': video_id, 'asrRes': asr_res}
     })
     # 4. gpt产出结果
+    log_.info(f"debug: title_generate 4")
     prompt = f"{config_.GPT_PROMPT['title']['prompt2']}{asr_res.strip()}"
     gpt_res = request_gpt(prompt=prompt)
     return gpt_res