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/