From 4cdecfbfb307c8eb6a6f30ada5ea09624a04f8f1 Mon Sep 17 00:00:00 2001
From: RecentRichRail <80782821+RecentRichRail@users.noreply.github.com>
Date: Tue, 28 Mar 2023 12:02:31 -0400
Subject: [PATCH] Update chatgpt.py

---
 searx/engines/chatgpt.py | 24 ++++++------------------
 1 file changed, 6 insertions(+), 18 deletions(-)

diff --git a/searx/engines/chatgpt.py b/searx/engines/chatgpt.py
index af80800be..a945c486f 100644
--- a/searx/engines/chatgpt.py
+++ b/searx/engines/chatgpt.py
@@ -1,28 +1,16 @@
 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
+openai.api_key = os.environ.get("OPENAI_API_KEY")
 engine_id = "text-davinci-002"
 
-@app.route('/chatgpt', methods=['GET'])
-def chatgpt():
-    query = request.args.get('query')
+def generate_response(prompt):
     response = openai.Completion.create(
         engine=engine_id,
-        prompt=query,
+        prompt=prompt,
         max_tokens=150,
         n=1,
-        stop="\n"
+        stop=None,
+        temperature=0.7
     )
-    chatgpt_response = response.choices[0].text
-    return chatgpt_response
-
-if __name__ == '__main__':
-    app.run()
+    return response.choices[0].text.strip()