Update main.py

main.py

@@ -85,11 +85,30 @@ def handle_image_message(event):
     chat = chat_sessions.get(user_id) or model.start_chat(history=[])
     chat_sessions[user_id] = chat
     previous_message = get_previous_message(user_id)
-
     user_text = event.message.text if event.message.type == "text" else None
     image_url = None
-    
-    if previous_message and previous_message["type"] == "image" and event.message.type == "text":
+
+    if event.message.type != "text" or event.message.type != "image":
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="請輸入文字或圖片~"))
+            return
+    elif user_text == "再見":
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="Bye!"))
+            return
+    # ========
+    # 純文字
+    # ========
+    elif event.message.type == "text" and previous_message["type"] != "image":
+        try: 
+            response = chat.send_message(user_text)
+            if (response.text != None):
+                out = response.text
+            else:
+                out = "Gemini沒答案!請換個說法！"
+        except:
+            # 處理錯誤
+            out = "Gemini執行出錯!請換個說法！" 
+            
+    elif previous_message and previous_message["type"] == "image" and event.message.type == "text":
         image_path = previous_message["content"]
         user_text = event.message.text
         store_user_message(user_id, "text", user_text)
@@ -101,7 +120,7 @@ def handle_image_message(event):
             out = completion.text
         except Exception as e:
             out = f"發生錯誤: {e}"
-            
+
     # ========
     # 上傳圖片
     # ========
@@ -150,24 +169,6 @@ def handle_image_message(event):
             )
         return
 
-    # ========
-    # 純文字
-    # ========
-    elif event.message.type != "text" or event.message.type != "image":
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="請輸入文字或圖片~"))
-            return
-    elif event.message.text == "再見":
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="Bye!"))
-            return
-    elif working_status:
-        try:
-            prompt = event.message.text
-            store_user_message(user_id, "text", prompt)
-            completion = chat.send_message(prompt)
-            out = completion.text if completion.text else "我不太懂什麼意思也~"
-        except:
-            out = "執行出錯!請換個說法！"
-
     line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))