From 57b1405cbe7948549d36db1c7c58bcd996f9c023 Mon Sep 17 00:00:00 2001
From: RecentRichRail <80782821+RecentRichRail@users.noreply.github.com>
Date: Tue, 28 Mar 2023 11:54:29 -0400
Subject: [PATCH] Create chatgpt.py

---
 searx/engines/chatgpt.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 searx/engines/chatgpt.py

diff --git a/searx/engines/chatgpt.py b/searx/engines/chatgpt.py
new file mode 100644
index 000000000..af80800be
--- /dev/null
+++ b/searx/engines/chatgpt.py
@@ -0,0 +1,28 @@
+import os
+from flask import Flask, jsonify, request
+import openai
+
+app = Flask(__name__)
+
+# retrieve ChatGPT API key from system variable
+API_KEY = os.environ.get("CHATGPT_API_KEY")
+
+# initialize OpenAI API client
+openai.api_key = API_KEY
+engine_id = "text-davinci-002"
+
+@app.route('/chatgpt', methods=['GET'])
+def chatgpt():
+    query = request.args.get('query')
+    response = openai.Completion.create(
+        engine=engine_id,
+        prompt=query,
+        max_tokens=150,
+        n=1,
+        stop="\n"
+    )
+    chatgpt_response = response.choices[0].text
+    return chatgpt_response
+
+if __name__ == '__main__':
+    app.run()