Update app.py

app.py

@@ -28,12 +28,38 @@ def home():
 def sentiment_analysis_ep(content: Text = None):
     url = 'https://api.openai.com/v1/chat/completions'
     prompt = '根据下面的文章，生成的“问题和回答”QA对，大于5个，以一行一个json格式（{“question”:"xxx","answer":"xxx"}）生成：\n'
-    print("content = \n", prompt + content.content)
     messages = [{"role": "user", "content": prompt + content.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()
+    print('res:', res)
+    res = {'content': res}
+    return JSONResponse(content=res)
+
+
+@app.post("/chat_pdf")
+def chat_pdf_ep(content: Text = None):
+    url = 'https://api.openai.com/v1/chat/completions'
+    messages = [
+        {
+            "role": "system",
+            "content": "你是一个有用的助手，可以使用文章内容准确地回答问题。使用提供的文章来形成你的答案，但避免逐字复制文章。尽可能使用自己的话。准确、有用、简洁、清晰。"
+        }
+    ]
+    obj = json.loads(content.content)
+    messages.append({"role": "user", "content": obj['question'] + '\n根据以下文章回答：\n' + obj['doc']})
+    data = {
+        "model": "gpt-3.5-turbo",
+        "messages": messages
+    }
+    print("messages = \n", messages)
     result = requests.post(url=url,
                            data=json.dumps(data),
                            headers=headers
@@ -55,4 +81,4 @@ def embeddings_ep(content: Text = None):
                            data=json.dumps(data),
                            headers=headers
                            )
-    return JSONResponse(content=result.json())
+    return JSONResponse(content=result.json())