Update main.py

main.py

@@ -89,7 +89,6 @@ def generate_and_upload_image(prompt: str) -> str:
             "Cache-Control": "no-cache",
             "Ocp-Apim-Subscription-Key": pixazo_api  # 填入你的 API 金鑰
         }
-        
         # 定義生成圖片的詳細設定
         data = {
             "prompt": prompt,
@@ -100,32 +99,25 @@ def generate_and_upload_image(prompt: str) -> str:
             "guidance_scale": 5,  # 數值越高，圖片越貼近提示詞
             "seed": 40            # 固定種子碼可確保相同提示詞生成相同圖片
         }
-    
         # --- 2. 發送請求至 Pixazo API ---
         response = requests.post(url, json=data, headers=headers)
         print(response.json())  # 在控制台查看原始回應數據內容
-        
         # 解析 JSON 回應
         resp_data = response.json()
         image_url = resp_data.get("imageUrl") # 從回應中取得圖片的外部連結
-    
         if image_url:
             # --- 3. 下載生成的圖片 ---
             # 因為 API 只給 URL，我們需要再發送一次 GET 請求把圖片存下來
             image_download_response = requests.get(image_url)
-            
             if image_download_response.status_code == 200:
                 # 取得圖片的二進位數據 (Binary data)
                 image_binary = image_download_response.content
-                
                 # --- 4. 處理並儲存圖片 ---
                 # 使用 PIL 打開二進位數據並轉為圖片格式
                 image = PIL.Image.open(io.BytesIO(image_binary))
-                
                 # 產生一個隨機的檔名（避免重複），存放在 static 資料夾下
                 file_name = f"static/{os.urandom(8).hex()}.png"
                 image.save(file_name, format="PNG")
-                
                 # --- 5. 回傳圖片連結 ---
                 # 取得目前的伺服器基礎位址 (例如 Hugging Face Space 或本地端)
                 base_url = os.getenv("HF_SPACE", "http://localhost:7860").rstrip("/")
@@ -135,7 +127,6 @@ def generate_and_upload_image(prompt: str) -> str:
         else:
             # 如果 JSON 裡沒拿到 imageUrl，回傳 API 給出的錯誤訊息
             return f"圖片生成失敗: {resp_data.get('message', 'API 未回傳 URL')}"
-    
     except Exception as e:
         # 捕捉任何執行過程中的異常（如網路中斷、記憶體錯誤等）
         return f"程式執行出錯: {e}"
@@ -147,7 +138,6 @@ def analyze_image_with_text(image_path: str, user_text: str) -> str:
         if not os.path.exists(image_path):
             return "錯誤：找不到該圖片檔案。"
         img_user = PIL.Image.open(image_path)
-        
         response = genai_client.models.generate_content(
             model="gemini-3-flash-preview",
             contents=[img_user, user_text]
@@ -174,7 +164,6 @@ agent_executor = create_react_agent(
 # ==========================
 #  FastAPI 路由
 # ==========================
-
 @app.get("/")
 def root():
     return {"status": "running"}
@@ -198,22 +187,18 @@ def handle_message(event):
         if image_path:
             store_user_message(user_id, "image", image_path)
             line_bot_api.reply_message(event.reply_token, TextSendMessage(text="收到圖片了！請問你想對這張圖做什麼分析？"))
-    
     # 2. 處理文字訊息 (修正縮進，確保它與上面的 if 對齊)
     elif isinstance(event.message, TextMessage):
         user_text = event.message.text
         previous_message = get_previous_message(user_id)
-        
         if previous_message["type"] == "image":
             image_path = previous_message["content"]
             user_text = f"請分析這張圖片 {image_path}，問題是：{user_text}"
             user_message_history[user_id].pop()
 
         agent_input = {"messages": [("user", user_text)]}
-
         try:
             response = agent_executor.invoke(agent_input)
-
             # 1. 先取得原始內容 (可能是字串，也可能是列表)
             raw_content = response["messages"][-1].content
             # 2. 判斷並提取純文字
@@ -226,11 +211,9 @@ def handle_message(event):
             else:
                 # 如果已經是字串，直接指派
                 out = raw_content
-                
             # 搜尋 URL 邏輯
             urls = re.findall(r'https?://[^\s<>"]+|www\.[^\s<>"]+', out)
             image_url = next((u for u in urls if any(ext in u.lower() for ext in ['.png', '.jpg'])), None)
-            
             if image_url:
                 line_bot_api.reply_message(
                     event.reply_token,