Selaa lähdekoodia

translate and suggest

kevin 1 vuosi sitten
vanhempi
commit
3927d4d007
1 muutettua tiedostoa jossa 23 lisäystä ja 4 poistoa
  1. 23 4
      main.py

+ 23 - 4
main.py

@@ -6,7 +6,6 @@ from flask import Flask, request, jsonify, send_from_directory, url_for
 app = Flask(__name__)
 ssl = None
 # ssl =('/etc/ssl/sample.crt', '/etc/ssl/sample.pem')
-#test2
 
 openai_key = os.environ.get("OPENAI_KEY", "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")
 
@@ -15,6 +14,14 @@ app.chat_messages = [
     {"role": "system",
      "content": "Please respond professionally and in a friendly manner, using the same language as the original request."}
 ]
+app.translate_messages = [
+    {"role": "system",
+     "content": "Please translate using the requested language."}
+]
+app.suggest_messages = [
+    {"role": "system",
+     "content": "Please suggest the best response based on the previous conversations and the user's request."}
+]
 UPLOAD_FOLDER = 'files'
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
@@ -28,7 +35,7 @@ def test():
 
 @app.route('/gpt', methods=['POST'])
 def gpt():
-    chat_messages = app.chat_messages.copy()
+#     chat_messages = app.chat_messages.copy()
     use_video = False
     json_payload = request.get_json()
     if not json_payload:
@@ -40,6 +47,12 @@ def gpt():
             use_video = json_payload['use_video'] == "1"
             json_payload = json_payload['payload']
             if isinstance(json_payload, dict):
+                if 'translate' in json_payload:
+                    chat_messages = app.translate_messages.copy()
+                elif 'suggest' in json_payload:
+                    chat_messages = app.suggest_messages.copy()
+                else:
+                    chat_messages = app.chat_messages.copy()
                 json_payload = [json_payload]
         elif 'greeting' in json_payload:
             company_name = json_payload['greeting']['company_name']
@@ -54,9 +67,9 @@ def gpt():
                                                      messages=islam_messages,
                                                      max_tokens=2, temperature=0.5)
             if 'Ya' in islamic_response.choices[0].message['content']:
-                greeting_message = f"Buatkan respons chatbot berupa greeting dari chat {company_name} pada jam {timestamp}, tidak perlu mention waktu, dan jawab dengan salam terlebih dahulu apabila ada unsur islami di nama perusahaan."
+                greeting_message = f"Buatkan respons chatbot berupa greeting dari chat perusahaan bernama {company_name} pada jam {timestamp}, tidak perlu mention waktu, dan jawab dengan 'Assalamu'alaikum...' terlebih dahulu"
             else:
-                greeting_message = f"Buatkan respons chatbot berupa greeting dari chat {company_name} pada jam {timestamp}, tidak perlu mention waktu"
+                greeting_message = f"Buatkan respons chatbot berupa greeting dari chat perusahaan bernama {company_name} pada jam {timestamp}, tidak perlu mention waktu"
             json_payload = [
                 {
                     "role": "user",
@@ -64,6 +77,12 @@ def gpt():
                 }
             ]
         else:
+            if 'translate' in json_payload:
+                chat_messages = app.translate_messages.copy()
+            elif 'suggest' in json_payload:
+                chat_messages = app.suggest_messages.copy()
+            else:
+                chat_messages = app.chat_messages.copy()
             json_payload = [json_payload]
     json_payload = json_payload[-5:]
     for message in json_payload: