Update main.py

main.py

@@ -58,29 +58,32 @@ def handle_message(event):
     global working_status
     
     if event.type != "message" or event.message.type != "text":
-        out = "Gemini: [No response or the response does not contain text]"
+        TextSendMessage(text="Gemini: [No response or the response does not contain text]")
         
-    if event.message.text == "再見":
+    elif event.message.text == "再見":
         working_status = True
         line_bot_api.reply_message(
             event.reply_token,
             TextSendMessage(text="Bye!"))
         return
-        
-    messages = []
-    if working_status:
-        message = event.message.text
-        messages.append({
-                        "role": "user",
-                        "parts": [message],
-                        })
-
-        response = model.generate_content(messages)
-        out = response.parts[0].text
-        # Check if response is valid
-        if not response.parts or not response.parts[0].text:
-            out = "Gemini: [No response or the response does not contain text]"
-
+       
+    elif working_status:
+        try: 
+            messages = []
+            message = event.message.text
+            messages.append({
+                            "role": "user",
+                            "parts": [message],
+                            })
+    
+            response = model.generate_content(messages)
+            out = response.parts[0].text
+            # Check if response is valid
+            if not response.parts or not response.parts[0].text:
+                out = "Gemini: [No response or the response does not contain text]"
+        except:
+            out = "Gemini: [No response or the response does not contain text]" 
+  
         line_bot_api.reply_message(
             event.reply_token,
             TextSendMessage(text=out))