Update main.py

main.py

@@ -2,6 +2,7 @@ import os
 import io
 import PIL.Image
 import uvicorn
+import re
 from collections import defaultdict
 from fastapi import FastAPI, Request, Header, BackgroundTasks, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
@@ -18,11 +19,16 @@ from linebot.models import (
     ImageMessage,
 )
 
-# --- 修正後的 LangChain 匯入路徑 ---
+# --- 核心修正：改用絕對路徑匯入 AgentExecutor ---
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.tools import tool
 from langchain_google_genai import ChatGoogleGenerativeAI
-from langchain.agents import AgentExecutor, create_tool_calling_agent
+# 如果 langchain.agents 匯入失敗，這裡改從 .agent 模組直接抓取
+try:
+    from langchain.agents import create_tool_calling_agent, AgentExecutor
+except ImportError:
+    from langchain.agents.agent import AgentExecutor
+    from langchain.agents import create_tool_calling_agent
 
 # ==========================
 #  環境設定與工具函式
@@ -34,10 +40,8 @@ line_bot_api = LineBotApi(os.environ.get("CHANNEL_ACCESS_TOKEN"))
 line_handler = WebhookHandler(os.environ.get("CHANNEL_SECRET"))
 
 user_message_history = defaultdict(list)
-
 app = FastAPI()
 
-# 確保 static 資料夾存在
 if not os.path.exists("static"):
     os.makedirs("static")
 app.mount("/static", StaticFiles(directory="static"), name="static")
@@ -53,7 +57,7 @@ app.add_middleware(
 def get_image_url_from_line(message_id):
     try:
         message_content = line_bot_api.get_message_content(message_id)
-        file_path = f"static/{message_id}.png" # 改存到 static 方便讀取
+        file_path = f"static/{message_id}.png"
         with open(file_path, "wb") as f:
             for chunk in message_content.iter_content():
                 f.write(chunk)
@@ -78,7 +82,6 @@ def get_previous_message(user_id):
 def generate_and_upload_image(prompt: str) -> str:
     """根據文字提示生成圖片。"""
     try:
-        # 修正模型名稱：目前最新穩定版為 imagen-3
         response = genai_client.models.generate_content(
             model="imagen-3.0-generate-001", 
             contents=prompt,
@@ -96,10 +99,9 @@ def generate_and_upload_image(prompt: str) -> str:
             file_name = f"static/{os.urandom(8).hex()}.png"
             image.save(file_name, format="PNG")
             
-            # 修正 URL 拼接邏輯
             base_url = os.getenv("HF_SPACE", "http://localhost:7860").rstrip("/")
-            image_url = f"{base_url}/{file_name}"
-            return image_url
+            # 如果在本地執行，請確保 URL 格式正確
+            return f"{base_url}/{file_name}"
         return "圖片生成失敗：模型未回傳數據。"
     except Exception as e:
         return f"圖片生成失敗: {e}"
@@ -110,11 +112,10 @@ def analyze_image_with_text(image_path: str, user_text: str) -> str:
     try:
         if not os.path.exists(image_path):
             return "錯誤：找不到該圖片檔案。"
-            
         img_user = PIL.Image.open(image_path)
-        # 修正模型名稱：目前為 gemini-1.5-flash 或 gemini-2.0-flash-exp
+        # 修正：目前最新為 gemini-2.0-flash 或 1.5-flash
         response = genai_client.models.generate_content(
-            model="gemini-3-flash-preview",
+            model="gemini-1.5-flash",
             contents=[img_user, user_text]
         )
         return response.text if response.text else "Gemini 沒答案！"
@@ -125,8 +126,7 @@ def analyze_image_with_text(image_path: str, user_text: str) -> str:
 #  LangChain 代理人設定
 # ==========================
 tools = [generate_and_upload_image, analyze_image_with_text]
-# 修正模型名稱
-llm = ChatGoogleGenerativeAI(model="gemini-3-flash-preview", temperature=0.2)
+llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0.2)
 
 prompt_template = ChatPromptTemplate.from_messages([
     ("system", "你是一個強大的助理。如果生成了圖片，請直接給出 URL。"),
@@ -158,20 +158,22 @@ async def webhook(request: Request, background_tasks: BackgroundTasks, x_line_si
 def handle_message(event):
     user_id = event.source.user_id
     
-    if event.message.type == "image":
+    # 處理圖片訊息
+    if isinstance(event.message, ImageMessage):
         image_path = get_image_url_from_line(event.message.id)
         if image_path:
             store_user_message(user_id, "image", image_path)
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="收到圖片了！請問你想對這張圖做什麼分析？"))
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="收到圖片了！請問你想分析什麼？"))
     
-    elif event.message.type == "text":
+    # 處理文字訊息
+    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"]
-            agent_input = {"input": f"請分析這張圖片 {image_path}，問題是：{user_text}"}
-            user_message_history[user_id].pop() # 用完即丟
+            agent_input = {"input": f"請分析圖片 {image_path}，問題：{user_text}"}
+            user_message_history[user_id].pop()
         else:
             agent_input = {"input": user_text}
         
@@ -179,25 +181,23 @@ def handle_message(event):
             response = agent_executor.invoke(agent_input)
             out = response["output"]
             
-            if "http" in out and (".png" in out or ".jpg" in out):
-                # 簡單提取第一個 URL
-                import re
-                urls = re.findall(r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', out)
-                if urls:
-                    image_url = urls[0]
-                    line_bot_api.reply_message(
-                        event.reply_token,
-                        [
-                            TextSendMessage(text="這是為你生成的圖片："),
-                            ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
-                        ]
-                    )
-                    return
-            
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
+            # 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', '.jpeg'])), None)
             
+            if image_url:
+                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=out))
         except Exception as e:
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text=f"系統忙碌中，請稍後再試。"))
+            print(f"Agent Error: {e}")
+            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="處理訊息時發生錯誤。"))
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)