Update main.py

main.py

@@ -74,9 +74,7 @@ def upload_image_to_imgur_with_token(image_path, access_token):
         with open(image_path, "rb") as img_file:
             image_binary = img_file.read()
 
-        headers = {
-            "Authorization": f"Bearer {access_token}"
-        }
+        headers = {"Authorization": f"Bearer {access_token}"}
 
         data = {
             "image": base64.b64encode(image_binary).decode("utf-8"),
@@ -103,6 +101,7 @@ def generate_image_with_gemini(prompt):
             tmp_file.write(image_data)
             return tmp_file.name
     return None
+    
 #==========================
 # 使用者上傳圖片
 #==========================
@@ -170,7 +169,7 @@ def handle_image_message(event):
         chat = model.start_chat(history=[])
         chat_sessions[user_id] = chat
 
-    user_text = event.message.text if event.message.type == "text" 
+    user_text = event.message.text if event.message.type == "text" else None
     if user_text and user_text.startswith("請幫我生成圖片"):
         prompt = user_text.replace("請幫我生成圖片", "").strip()