Update app.py

app.py

@@ -106,6 +106,25 @@ def sale_ep(content: Text = None):
     return JSONResponse(content=content)
 
 
+@app.post("/chat_gpt")
+def chat_gpt_ep(content: Text = None):
+    url = 'https://api.openai.com/v1/chat/completions'
+    obj = json.loads(content.content)
+    data = {
+        "model": "gpt-3.5-turbo",
+        "messages": obj['messages']
+    }
+    print("data = \n", data)
+    result = requests.post(url=url,
+                           data=json.dumps(data),
+                           headers=headers
+                           )
+    res = str(result.json()['choices'][0]['message']['content']).strip()
+    content = {'content': res}
+    print('content:', content)
+    return JSONResponse(content=content)
+
+
 @app.post("/embeddings")
 def embeddings_ep(content: Text = None):
     url = 'https://api.openai.com/v1/embeddings'