Update main.py

main.py

@@ -19,6 +19,7 @@ import io
 import requests
 import uvicorn
 import logging
+import base64
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.tools import tool
 from langchain_google_genai import ChatGoogleGenerativeAI
@@ -31,15 +32,14 @@ from typing import List
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # 檢查必要的環境變數是否都已設定
-if not all(k in os.environ for k in ["CHANNEL_ACCESS_TOKEN", "CHANNEL_SECRET", "GOOGLE_API_KEY", "IMGBB_API_KEY", "HF_SPACE"]):
-    logging.error("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, GOOGLE_API_KEY, IMGBB_API_KEY, and HF_SPACE.")
-    raise ValueError("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, GOOGLE_API_KEY, IMGBB_API_KEY, and HF_SPACE.")
+if not all(k in os.environ for k in ["CHANNEL_ACCESS_TOKEN", "CHANNEL_SECRET", "GOOGLE_API_KEY", "HF_SPACE"]):
+    logging.error("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, GOOGLE_API_KEY, and HF_SPACE.")
+    raise ValueError("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, GOOGLE_API_KEY, and HF_SPACE.")
 
 # 獲取環境變數中的金鑰和 URL
 google_api = os.environ["GOOGLE_API_KEY"]
 line_channel_access_token = os.environ["CHANNEL_ACCESS_TOKEN"]
 line_channel_secret = os.environ["CHANNEL_SECRET"]
-IMGBB_API_KEY = os.environ["IMGBB_API_KEY"]
 HF_SPACE_URL = os.environ["HF_SPACE"]
 
 # Line Bot API 設定
@@ -54,15 +54,9 @@ user_states = defaultdict(lambda: {
     "upper_body_images": [],
     "lower_body_images": [],
     "current_mode": None,
-    "is_ready_for_outfit": False,
-    "is_ready_for_photo": False,
-    "user_info": {},
-    "personal_photo": None,
-    "personal_photo_base64": None
 })
 
 MAX_IMAGES_PER_TYPE = 3
-GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-preview-05-20:generateContent"
 IMAGIN_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-image-preview:generateContent"
 
 # 建立 FastAPI 應用程式
@@ -181,58 +175,25 @@ def generate_outfit_from_clothes(upper_body_urls: list, lower_body_urls: list) -
                     return generated_image_url
                 else:
                     logging.error("Failed to save generated outfit image locally.")
-                    return "圖片生成失敗，無法保存到伺服器。"
+                    return None
             else:
                 logging.error("Gemini image response format is invalid for outfit generation.")
-                return "圖片生成失敗，回應格式不正確。"
+                return None
         else:
             logging.error(f"Gemini outfit generation response has no candidates: {response.text}")
-            return "圖片生成失敗，請稍後再試。"
+            return None
     except requests.exceptions.RequestException as e:
         logging.error(f"Gemini outfit generation API request failed: {e}")
-        return f"圖片生成與上傳失敗: {e}"
-
-@tool
-def analyze_image_with_text(image_url: str, user_text: str) -> str:
-    """
-    這個工具可以根據圖片和文字提示來回答問題。
-
-    Args:
-        image_url: 圖片的 URL。
-        user_text: 針對圖片提出的文字問題。
-
-    Returns:
-        模型針對圖片和文字提示給出的回應。
-    """
-    try:
-        if not image_url:
-            return "圖片路徑無效，無法進行分析。"
-        
-        image_bytes = requests.get(image_url).content
-        img_user = Image.open(io.BytesIO(image_bytes))
-
-        response = genai_client.models.generate_content(
-            model="gemini-2.5-flash",
-            contents=[img_user, user_text]
-        )
-        if (response.text != None):
-            out = response.text
-        else:
-            out = "Gemini沒答案!請換個說法！"
-    except Exception as e:
-        # 處理錯誤
-        out = f"Gemini執行出錯: {e}"
-    
-    return out
+        return None
 
 # ==========================#  LangChain 代理人設定# ==========================#
 # 結合所有工具
-tools = [generate_outfit_from_clothes, analyze_image_with_text]
+tools = [generate_outfit_from_clothes]
 # 建立 LLM 模型實例
 llm = ChatGoogleGenerativeAI(google_api_key=google_api, model="gemini-2.5-flash", temperature=0.2)
 # 建立提示模板
 prompt_template = ChatPromptTemplate([
-    ("system", "你是一個強大的圖像生成與問答助理，可以根據用戶的請求使用提供的工具。當你執行 generate_and_upload_image 工具成功後會獲得一個 URL，然後你回答的 output 要包含有這個 URL 的完整資訊。如果工具有產生錯誤訊息請解讀並回應。"),
+    ("system", "你是一個強大的圖像生成與問答助理，可以根據用戶的請求使用提供的工具。當你執行 generate_outfit_from_clothes 工具成功後會獲得一個 URL，然後你回答的 output 要包含有這個 URL 的完整資訊。如果工具有產生錯誤訊息請解讀並回應。"),
     ("user", "{input}"),
     ("placeholder", "{agent_scratchpad}"),
 ])
@@ -309,9 +270,9 @@ def handle_message(event):
         
         # 處理重置功能
         if user_text in ["重置", "重來", "重新開始", "再一次"]:
-            user_states[user_id] = defaultdict(lambda: {"upper_body_images": [], "lower_body_images": [], "current_mode": None, "is_ready_for_outfit": False, "is_ready_for_photo": False, "user_info": {}, "personal_photo": None, "personal_photo_base64": None})[user_id]
+            user_states[user_id] = defaultdict(lambda: {"upper_body_images": [], "lower_body_images": [], "current_mode": None})[user_id]
             line_bot_api.reply_message(
-                reply_token, TextSendMessage(text="狀態已重置。請重新輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。例如：165,85,65,90,約會")
+                reply_token, TextSendMessage(text="狀態已重置。請先輸入「上衣」或「褲子」來開始。")
             )
             return
             
@@ -359,7 +320,7 @@ def handle_message(event):
                     line_bot_api.push_message(user_id, TextSendMessage(text=f"圖片生成失敗，請稍後再試。錯誤：{e}"))
 
                 # 生成後重置狀態以便下一次使用
-                user_states[user_id] = defaultdict(lambda: {"upper_body_images": [], "lower_body_images": [], "current_mode": None, "is_ready_for_outfit": False, "is_ready_for_photo": False, "user_info": {}, "personal_photo": None, "personal_photo_base64": None})[user_id]
+                user_states[user_id] = defaultdict(lambda: {"upper_body_images": [], "lower_body_images": [], "current_mode": None})[user_id]
                 return
             else:
                 line_bot_api.reply_message(