Update app.py

app.py

@@ -2,12 +2,21 @@ from fastapi import FastAPI
 from fastapi.responses import HTMLResponse
 from pydantic import BaseModel
 from fastapi.responses import JSONResponse
+import requests
+import json
 
 class Text(BaseModel):
     content: str = ""
-    
+
+
 app = FastAPI()
 
+url = 'https://api.openai.com/v1/chat/completions'
+headers = {
+    'Content-Type': 'application/json',
+    'Authorization': 'Bearer ' + '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'
+}
+
 
 @app.get("/")
 def home():
@@ -15,11 +24,18 @@ def home():
     return HTMLResponse(content=html_content, status_code=200)
 
 
-@app.get("/hello")
-def hello():
-    return {"hello": "world"}
-
-@app.post("/sentiment-analysis")
+@app.post("/qa_maker")
 def sentiment_analysis_ep(content: Text = None):
-    content = {"message": "hello world"}
-    return JSONResponse(content=content)
+    prompt = '根据下面的文章，生成的“问题和回答”QA对，大于5个，以一行一个json格式（{“question”:"xxx","answer":"xxx"}）生成：\n'
+    messages = [{"role": "user", "content": prompt + content.content}]
+    data = {
+        "model": "gpt-3.5-turbo",
+        "messages": messages
+    }
+    result = requests.post(url=url,
+                           data=json.dumps(data),
+                           headers=headers
+                           )
+
+    res = {'content': str(result.json()['choices'][0]['message']['content']).strip()}
+    return JSONResponse(content=res)