Update app.py

app.py

@@ -187,3 +187,68 @@ def create_image_ep(content: Text = None):
                            headers=headers
                            )
     return JSONResponse(content=result.json())
+
+
+
+
+
+
+
+
+
+
+
+
+
+import requests
+from concurrent.futures import ThreadPoolExecutor, as_completed
+import openai
+import json
+import time
+key = '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'
+openai.api_key = key
+headers = {
+    'Content-Type': 'application/json',
+    'Authorization': 'Bearer ' + key
+}
+
+# 初始化线程池
+pool = ThreadPoolExecutor(max_workers=4)
+url = 'https://api.openai.com/v1/chat/completions'
+prompt = '根据下面的文章，生成的“问题和回答”QA对。要求：不少于3个，问题不能重复，，回答不是"未知"，以一行一个json格式（{“question”:"xxx","answer":"xxx"}）生成：\n'
+
+
+# 每个线程去调api，异常后 retry的次数 和 间隔时间
+def task(content, retry=3, interval=2):
+    while retry > 0:
+        try:
+            messages = [{"role": "user", "content": prompt + content}]
+            data = {
+                "model": "gpt-3.5-turbo",
+                "messages": messages
+            }
+            print("messages = \n", messages)
+            result = requests.post(url=url,
+                                   data=json.dumps(data),
+                                   headers=headers
+                                   )
+            res = str(result.json()['choices'][0]['message']['content']).strip()
+            if res:
+                return res
+            else:
+                retry -= 1
+        except Exception as e:
+            print(e)
+            retry -= 1
+        time.sleep(interval)
+
+
+doc_list = ['周深，1992年9月29日出生于湖南省邵阳市，成长于贵州省贵阳市，中国内地男歌手，毕业于乌克兰利沃夫国立谢科音乐科学院美声专业。2014年，参加浙江卫视歌唱选秀节目《中国好声音第三季》，从而正式出道。','2016年，获得东方风云榜最佳新人奖。同年，为动画电影《大鱼海棠》演唱印象曲《大鱼》，并凭借该单曲获得多个音乐奖项。2020年，作为首发歌手加盟《歌手当打之年》，其演唱的《达拉崩吧》以破亿播放量斩获云村播放量第一。','2017年3月14-15日，周深率领动感101爱心支教团奔赴家乡贵州，前往兴义市威舍镇阿依小学支教，助力偏远地区的孩子们实现音乐梦想，不仅为孩子们准备了礼物，还自创教学歌，与孩子们一起参加运动会。']
+
+# 用线程池去调用api
+all_task = [pool.submit(task, doc) for doc in doc_list]
+# 等待所有线程完成
+for future in as_completed(all_task):
+    print(future.result())
+
+