|
|
@@ -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='')
|
