Update main.py

main.py

@@ -91,13 +91,21 @@ def upload_image_to_imgur_with_token(image_path, access_token):
         print("圖片上傳例外：", e)
         return None
         
-def generate_image_with_gemini(prompt):
+def generate_image_with_gemini(prompt, size=(512, 512)):
     model_name = "gemini-2.0-flash-exp-image-generation"
     image_model = genai.GenerativeModel(model_name)
+    prompt += f"，尺寸：{size[0]}x{size[1]}"
     response = image_model.generate_content(prompt)
     if response.parts and hasattr(response.parts[0], "inline_data"):
         image_data = response.parts[0].inline_data.data
         with tempfile.NamedTemporaryFile(delete=False, suffix=".png", dir="/tmp") as tmp_file:
+            # 清理先前生成的暫存圖檔（選擇性，可加強）
+            for f in os.listdir("/tmp"):
+                if f.endswith(".png") and os.path.getmtime(f"/tmp/{f}") < os.path.getmtime(tmp_file.name):
+                    try:
+                        os.remove(f"/tmp/{f}")
+                    except:
+                        pass
             tmp_file.write(image_data)
             return tmp_file.name
     return None
@@ -159,104 +167,70 @@ def get_previous_message(user_id):
 # 主程式(圖片與文字)
 #==========================
 # 建立 chat_sessions 字典
-chat_sessions = {}
-@line_handler.add(MessageEvent, message=(ImageMessage,TextMessage))
+@line_handler.add(MessageEvent, message=(ImageMessage, TextMessage))
 def handle_image_message(event):
     user_id = event.source.user_id
-    if user_id in chat_sessions:
-        chat = chat_sessions[user_id]
-    else:
-        chat = model.start_chat(history=[])
-        chat_sessions[user_id] = chat
+    chat = chat_sessions.get(user_id) or model.start_chat(history=[])
+    chat_sessions[user_id] = chat
 
-    
-    # 從 LINE API 獲取圖片數據
-    image_path = get_image_url(event.message.id)
-    if image_path:
-        try:
-            # 處理圖片或進行進一步分析
-            print(f"圖片已保存至: {image_path}")
+    user_text = event.message.text if event.message.type == "text" else None
+
+    if user_text and user_text.startswith("請幫我生成圖片"):
+        prompt = user_text.replace("請幫我生成圖片", "").strip()
+
+        image_path = generate_image_with_gemini(prompt, size=(512, 512))
+        if image_path:
+            image_url = f"https://your-domain.com/tmp/{os.path.basename(image_path)}"
+            line_bot_api.reply_message(
+                event.reply_token,
+                [
+                    TextSendMessage(text="這是我為你生成的圖片喔～ ✨"),
+                    ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
+                ]
+            )
+        else:
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成失敗，請稍後再試~"))
+        return
+
+    if event.message.type == "image":
+        image_path = get_image_url(event.message.id)
+        if image_path:
             store_user_message(user_id, "image", image_path)
-            # 回應用戶
-            line_bot_api.reply_message(event.reply_token,TextSendMessage(text="圖片已接收成功囉，幫我輸入你想詢問的問題喔~"))
-        except:
-            line_bot_api.reply_message(event.reply_token,TextSendMessage(text="沒有接收到圖片~"))
-    
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片已接收成功囉，幫我輸入你想詢問的問題喔~"))
+        else:
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="沒有接收到圖片~"))
+        return
+
     previous_message = get_previous_message(user_id)
-    # 獲取傳送圖片後的用戶問題（文字）
     if 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)
-    
         try:
-            # 確保圖片存在
             if not os.path.exists(image_path):
                 raise FileNotFoundError(f"圖片路徑無效：{image_path}")
-        
             organ = PIL.Image.open(image_path)
             completion = chat.send_message([user_text, organ])
             out = completion.text
-            
         except Exception as e:
             out = f"發生錯誤: {e}"
-        
     else:
-        global working_status
-        # 檢查事件類型和訊息類型
-        if event.type != "message" or event.message.type != "text":
-            # 回覆錯誤訊息
-            line_bot_api.reply_message(
-            event.reply_token,
-            TextSendMessage(text="Event type error:[No message or the message does not contain text]")
-            )
-        
-        # 檢查使用者是否輸入 "再見"
-        elif event.message.text == "再見":
-        # 回覆 "Bye!"
-            line_bot_api.reply_message(event.reply_token,TextSendMessage(text="Bye!"))
+        if event.message.type != "text":
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="請輸入文字或圖片~"))
             return
-       
-        # 檢查是否正在與使用者交談
-        elif working_status:
-            try: 
-
-                 if user_text and user_text.startswith("請幫我生成圖片"):
-                    prompt = user_text.replace("請幫我生成圖片", "").strip()
-
-                    image_path_gen = generate_image_with_gemini(prompt)
-                    if image_path:
-                    # 使用 Access Token 上傳至 Imgur
-                    IMGUR_ACCESS_TOKEN = os.environ.get("IMGUR_ACCESS_TOKEN")
-                    image_url = upload_image_to_imgur_with_token(image_path, IMGUR_ACCESS_TOKEN)
-
-                    if image_url:
-                        line_bot_api.reply_message(event.reply_token,ImageSendMessage(original_content_url=image_url, preview_image_url=image_url))
-        else:
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片上傳失敗，請稍後再試~"))
-                # 取得使用者輸入的文字
+        if event.message.text == "再見":
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="Bye!"))
+            return
+        if working_status:
+            try:
                 prompt = event.message.text
-                store_user_message(user_id, "text", prompt) 
-   
-                # 使用 Gemini 模型生成文字
+                store_user_message(user_id, "text", prompt)
                 completion = chat.send_message(prompt)
-                # 檢查生成結果是否為空
-                if (completion.text != None):
-                    # 取得生成結果
-                    out = completion.text
-                else:
-                    # 回覆 "Gemini沒答案!請換個說法！"
-                    out = "我不太懂什麼意思也~"
+                out = completion.text if completion.text else "我不太懂什麼意思也~"
             except:
-                # 處理錯誤
-                out = "執行出錯!請換個說法！" 
+                out = "執行出錯!請換個說法！"
 
-        
-    # 回覆生成結果
-    line_bot_api.reply_message(event.reply_token,TextSendMessage(text=out))
+    line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
 
 
 if __name__ == "__main__":