v5

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from pytube import YouTube
 
 
-import requests
+import requests,json
 
 headers = {
     'accept': 'application/json',
@@ -13,13 +13,47 @@ headers = {
 }
 
 
+openai_key = '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'
+
+headers = {
+        'Content-Type': 'application/json',
+        'Authorization': f'Bearer {openai_key}',
+       
+    }
+
+def create(prompt):
+    data =   {
+        "model": "text-davinci-003",
+        "prompt": prompt,
+        "temperature": 0.7,
+        "max_tokens": 1024,
+        "top_p": 1.0,
+        "frequency_penalty": 0.0,
+        "presence_penalty": 0.0
+    }
+    url = 'https://api.openai.com/v1/completions'
+    r = requests.post(url,headers=headers,
+            data=json.dumps(data))
+    return r.json()
+
+
 def convert(res):
     data = res.json()
     prediction = data['prediction']
     content = []
     for x in prediction:
         content.append(x['transcription'])
-    return '\n'.join(content)
+    auido_txt = '\n'.join(content)
+
+    prompt = f"将下面的内容，总结10条要点出来，\n{auido_txt}"
+    open_ai_res = create(prompt)
+    answer = open_ai_res['choices'][0]['text']
+
+    res_content = f'音频内容：\n{auido_txt}\nGPT3总结的要点：\n{answer}'
+
+    return res_content
+
+    
         
 
 def get_audio(url):