From afd7d9a150185845768c33f504abef1818df4d6e Mon Sep 17 00:00:00 2001
From: explainthis <hello@explainthis.io>
Date: Mon, 20 Mar 2023 20:40:55 +0800
Subject: [PATCH] Refactor

---
 main.py | 23 +++++++++++++++--------
 1 file changed, 15 insertions(+), 8 deletions(-)

diff --git a/main.py b/main.py
index 93602eb65..b308da59e 100644
--- a/main.py
+++ b/main.py
@@ -78,11 +78,12 @@ def handle_text_message(event):
             is_successful, response, error_message = model_management[user_id].image_generations(prompt)
             if not is_successful:
                 raise Exception(error_message)
+            url = response['data'][0]['url']
             msg = ImageSendMessage(
-                original_content_url=response,
-                preview_image_url=response
+                original_content_url=url,
+                preview_image_url=url
             )
-            memory.append(user_id, 'assistant', response)
+            memory.append(user_id, 'assistant', url)
 
         else:
             memory.append(user_id, 'user', text)
@@ -97,7 +98,10 @@ def handle_text_message(event):
         msg = TextSendMessage(text='Token 無效，請重新註冊，格式為 /註冊 sk-xxxxx')
     except Exception as e:
         memory.remove(user_id)
-        msg = TextSendMessage(text=str(e))
+        if str(e).startswith('Incorrect API key provided'):
+            msg = TextSendMessage(text='OpenAI API Token 有誤，請重新註冊。')
+        else:
+            msg = TextSendMessage(text=str(e))
     line_bot_api.reply_message(event.reply_token, msg)
 
 
@@ -114,10 +118,10 @@ def handle_audio_message(event):
         if not model_management.get(user_id):
             raise ValueError('Invalid API token')
         else:
-            transciption, error_message = model_management[user_id].audio_transcriptions(input_audio_path, 'whisper-1')
-            memory.append(user_id, 'user', transciption)
-            if error_message:
+            is_successful, response, error_message = model_management[user_id].audio_transcriptions(input_audio_path, 'whisper-1')
+            if not is_successful:
                 raise Exception(error_message)
+            memory.append(user_id, 'user', response['text'])
             is_successful, response, error_message = model_management[user_id].chat_completions(memory.get(user_id), 'gpt-3.5-turbo')
             if not is_successful:
                 raise Exception(error_message)
@@ -128,7 +132,10 @@ def handle_audio_message(event):
         msg = TextSendMessage(text='請先註冊你的 API Token，格式為 /註冊 [API TOKEN]')
     except Exception as e:
         memory.remove(user_id)
-        msg = TextSendMessage(text=str(e))
+        if str(e).startswith('Incorrect API key provided'):
+            msg = TextSendMessage(text='OpenAI API Token 有誤，請重新註冊。')
+        else:
+            msg = TextSendMessage(text=str(e))
     os.remove(input_audio_path)
     line_bot_api.reply_message(event.reply_token, msg)