1234567891011121314151617181920212223242526272829303132333435363738 |
- 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/
|