update

config.py

@@ -18,6 +18,8 @@ class BaseConfig(object):
     # GPT_OPENAI_API_KEY = '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'
     # GPT_OPENAI_API_KEY = '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'
     GPT_OPENAI_API_KEY = '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'
+    GPT_URL = 'http://aigc.piaoquantv.com/aigc-server/aigc/content'
+    RETRY_MAX_COUNT = 3
 
     # 代理地址
     PROXIES = {

gpt_tag.py

@@ -1,4 +1,5 @@
 import time
+import json
 import requests
 import traceback
 from config import set_config
@@ -42,3 +43,28 @@ def get_tag(prompt):
             time.sleep(30)
             continue
     return result_content
+
+
+def request_gpt(prompt):
+    retry_count = 0
+    result_content = None
+    while retry_count < config_.RETRY_MAX_COUNT:
+        retry_count += 1
+        try:
+            # response = requests.post(url=config_.GPT_URL, json={'content': prompt, 'auth': config_.GPT_OPENAI_API_KEY})
+            response = requests.post(url=config_.GPT_URL, json={'content': prompt})
+            # print(response.json())
+            # print(response.json()['choices'][0]['message']['content'])
+            # print('\n')
+            # result_content = response.json()['choices'][0]['message']['content']
+            # log_.info(f"response.text: {response.text}")
+            res_data = json.loads(response.text)
+            if res_data['code'] != 0:
+                time.sleep(10)
+                continue
+            result_content = res_data['data']['choices'][0]['message']['content']
+            return result_content
+        except Exception:
+            time.sleep(10)
+            continue
+    return result_content

main_process.py

@@ -5,7 +5,7 @@ from feishu import FeiShuHelper
 from audio_process import get_wav
 from xunfei_asr import call_asr
 from utils import download_video, asr_validity_discrimination
-from gpt_tag import get_tag
+from gpt_tag import get_tag, request_gpt
 from config import set_config
 from log import Log
 config_ = set_config()
@@ -106,7 +106,8 @@ def main(sheet_info_config):
 
                     # 6. gpt产出结果
                     prompt = f"{config_.GPT_PROMPT['tags']['prompt5']}{asr_res.strip()}"
-                    gpt_res = get_tag(prompt=prompt)
+                    # gpt_res = get_tag(prompt=prompt)
+                    gpt_res = request_gpt(prompt=prompt)
                     # print(gpt_res)
                     log_.info(f"gpt_res = {gpt_res}, type = {type(gpt_res)}")
 
@@ -149,10 +150,10 @@ if __name__ == '__main__':
         '历史视频top5000回流倒叙排列': {
             'video_spreadsheet_token': 'L4ywsRaV2hFLv1t4Athcdw71nde',
             'video_sheet_id': 'hRjMrL',
-            'read_start_row': 1377,
+            'read_start_row': 1455,
             'res_spreadsheet_token': 'DkiUsqwJ6hmBxstBYyEcNE4ante',
             'res_sheet_id': '7Fua00',
-            'write_start_row': 1377,
+            'write_start_row': 1455,
             'write_start_col': 'A',
             'write_end_col': 'H'
         }