update

gpt_process.py

@@ -1,3 +1,4 @@
+import os
 import time
 import requests
 import json
@@ -36,11 +37,9 @@ def request_gpt(prompt):
     # print(response.json()['choices'][0]['message']['content'])
     # print('\n')
     # result_content = response.json()['choices'][0]['message']['content']
-    log_.info(f"response: {response}")
     log_.info(f"response.text: {response.text}")
-
     res_data = json.loads(response.text)
-    result_content = res_data['choices'][0]['message']['content']
+    result_content = res_data['data']['choices'][0]['message']['content']
     return result_content
 
 
@@ -69,6 +68,9 @@ def title_generate(video_id, video_path):
     log_.info(f"debug: title_generate 4")
     prompt = f"{config_.GPT_PROMPT['title']['prompt2']}{asr_res.strip()}"
     gpt_res = request_gpt(prompt=prompt)
+    # 5. 删除相关文件
+    for file in [video_file_path, audio_path, dialogue_path]:
+        os.remove(file)
     return gpt_res
     # except ConnectionResetError:
     #     log_.info(video_id)
@@ -78,10 +80,5 @@ def title_generate(video_id, video_path):
     #     print(gpt_res)
 
 
-    # print(gpt_res)
-    # log_.info(f"gpt_res = {gpt_res}")
-
-
-
 if __name__ == '__main__':
     title_generate(video_id='001', video_path='')