浏览代码

translate and suggest

kevin 1 年之前
父节点
当前提交
895db25a68
共有 1 个文件被更改,包括 16 次插入15 次删除
  1. 16 15
      main.py

+ 16 - 15
main.py

@@ -42,19 +42,25 @@ def gpt():
         json_payload = []
     has_named_params = False
     if isinstance(json_payload, dict):
-        has_named_params = 'use_video' in json_payload or 'payload' in json_payload
-        if 'use_video' in json_payload and 'payload' in json_payload:
-            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:
+        has_named_params = 'payload' in json_payload
+        if 'payload' in json_payload:
+            if 'use_video' in json_payload:
+                use_video = json_payload['use_video'] == "1"
+            if 'translate' in json_payload:
+                chat_messages = app.translate_messages.copy()
+                json_payload['payload'][0]['content'] = json_payload['payload'][0]['content'] + f" (Translate to {json_payload['translate']})"
+            elif 'suggest' in json_payload:
+                if json_payload['suggest'] == "1":
                     chat_messages = app.suggest_messages.copy()
                 else:
                     chat_messages = app.chat_messages.copy()
+            else:
+                chat_messages = app.chat_messages.copy()
+            json_payload = json_payload['payload']
+            if isinstance(json_payload, dict):
                 json_payload = [json_payload]
         elif 'greeting' in json_payload:
+            chat_messages = app.chat_messages.copy()
             company_name = json_payload['greeting']['company_name']
             timestamp = json_payload['greeting']['timestamp']
             islamic_message = f"Apakah Nama '{company_name}' terdapat unsur islami? Jawab dengan 'Ya' atau 'Tidak'"
@@ -77,14 +83,9 @@ 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()
+            chat_messages = app.chat_messages.copy()
             json_payload = [json_payload]
-    json_payload = json_payload[-5:]
+    json_payload = json_payload[-7:]
     for message in json_payload:
         content = message['content']
         content_arr = content.split(" ")