main.py 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  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. # ssl =('/etc/ssl/sample.crt', '/etc/ssl/sample.pem')
  11. openai_key = os.environ.get("OPENAI_KEY", "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")
  12. openai.api_key = openai_key
  13. app.chat_messages = [
  14. {"role": "system",
  15. "content": "Jawab hanya dengan bahasa Indonesia"}
  16. ]
  17. @app.route('/', methods=['GET', 'POST'])
  18. def test():
  19. return jsonify({"status": "0"})
  20. @app.route('/gpt', methods=['POST'])
  21. def gpt():
  22. chat_messages = app.chat_messages.copy()
  23. json_payload = request.get_json()
  24. if not json_payload:
  25. json_payload = []
  26. if isinstance(json_payload, dict):
  27. json_payload = [json_payload]
  28. json_payload = json_payload[-5:]
  29. for message in json_payload:
  30. content = message['content']
  31. content_arr = content.split(" ")
  32. new_content_arr = content[:250].split(" ")
  33. new_content_len = len(new_content_arr)
  34. arr = []
  35. for i in range(new_content_len):
  36. arr.append(content_arr[i])
  37. message['content'] = " ".join(arr)
  38. chat_messages.append(message)
  39. print(chat_messages)
  40. result = {}
  41. try:
  42. json_response = openai.ChatCompletion.create(model="gpt-3.5-turbo", # GPT-3.5 Turbo engine
  43. messages=chat_messages,
  44. max_tokens=300, temperature=0.6)
  45. print(json_response.choices[0].message)
  46. result = json_response.choices[0].message
  47. except Exception as error_print:
  48. print(error_print)
  49. result = {}, 405
  50. return result
  51. # Press the green button in the gutter to run the script.
  52. if __name__ == '__main__':
  53. app.run(host='0.0.0.0', port=8348, debug=True, ssl_context=ssl)
  54. # See PyCharm help at https://www.jetbrains.com/help/pycharm/