|
@@ -0,0 +1,38 @@
|
|
|
+import json
|
|
|
+import openai
|
|
|
+from flask import Flask, request, jsonify
|
|
|
+import re
|
|
|
+import os
|
|
|
+import time
|
|
|
+from openai import ChatCompletion
|
|
|
+
|
|
|
+app = Flask(__name__)
|
|
|
+ssl = None
|
|
|
+openai_key = os.environ.get("OPENAI_KEY","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")
|
|
|
+openai.api_key = openai_key
|
|
|
+app.chat_messages = [
|
|
|
+ {"role": "system", "content": "Kamu adalah asisten yang membantu dan selalu menjawab dengan singkat"}
|
|
|
+]
|
|
|
+
|
|
|
+@app.route('/', methods=['GET', 'POST'])
|
|
|
+def test():
|
|
|
+ return jsonify({"status": "0"})
|
|
|
+
|
|
|
+@app.route('/gpt', methods=['POST'])
|
|
|
+def gpt():
|
|
|
+ chat_messages = app.chat_messages.copy()
|
|
|
+ json_payload = request.get_json()
|
|
|
+ if not json_payload:
|
|
|
+ json_payload = []
|
|
|
+ for message in json_payload:
|
|
|
+ chat_messages.append(message)
|
|
|
+ response = openai.ChatCompletion.create(model="gpt-3.5-turbo", # GPT-3.5 Turbo engine
|
|
|
+ messages=json_payload,
|
|
|
+ max_tokens=400,)
|
|
|
+ return response.choices[0].message
|
|
|
+
|
|
|
+# Press the green button in the gutter to run the script.
|
|
|
+if __name__ == '__main__':
|
|
|
+ app.run(host='0.0.0.0', port=8348, debug=True, ssl_context=ssl)
|
|
|
+
|
|
|
+# See PyCharm help at https://www.jetbrains.com/help/pycharm/
|