main.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. import json
  2. import openai
  3. from flask import Flask, request, jsonify
  4. import re
  5. import os
  6. import time
  7. from openai import ChatCompletion
  8. app = Flask(__name__)
  9. ssl = None
  10. openai_key = os.environ.get("OPENAI_KEY","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")
  11. openai.api_key = openai_key
  12. app.chat_messages = [
  13. {"role": "system", "content": "Kamu adalah asisten yang membantu dan selalu menjawab dengan singkat"}
  14. ]
  15. @app.route('/', methods=['GET', 'POST'])
  16. def test():
  17. return jsonify({"status": "0"})
  18. @app.route('/gpt', methods=['POST'])
  19. def gpt():
  20. chat_messages = app.chat_messages.copy()
  21. json_payload = request.get_json()
  22. if not json_payload:
  23. json_payload = []
  24. for message in json_payload:
  25. chat_messages.append(message)
  26. response = openai.ChatCompletion.create(model="gpt-3.5-turbo", # GPT-3.5 Turbo engine
  27. messages=json_payload,
  28. max_tokens=400,)
  29. return response.choices[0].message
  30. # Press the green button in the gutter to run the script.
  31. if __name__ == '__main__':
  32. app.run(host='0.0.0.0', port=8348, debug=True, ssl_context=ssl)
  33. # See PyCharm help at https://www.jetbrains.com/help/pycharm/