Update main.py

main.py

@@ -82,28 +82,6 @@ def get_previous_message(user_id):
 @tool
 def generate_and_upload_image(prompt: str) -> str:
     """根據文字提示生成圖片。"""
-    # # Gemini image 模型需付費
-    # try:
-    #     response = genai_client.models.generate_content(
-    #         model="gemini-2.5-flash-image", 
-    #         contents=prompt,
-    #         config=types.GenerateContentConfig(response_modalities=['IMAGE'])
-    #     )
-    #     image_binary = None
-    #     for part in response.candidates[0].content.parts:
-    #         if part.inline_data:
-    #             image_binary = part.inline_data.data
-    #             break
-    #     if image_binary:
-    #         image = PIL.Image.open(io.BytesIO(image_binary))
-    #         file_name = f"static/{os.urandom(8).hex()}.png"
-    #         image.save(file_name, format="PNG")
-    #         base_url = os.getenv("HF_SPACE", "http://localhost:7860").rstrip("/")
-    #         return f"{base_url}/{file_name}"
-    #     else:
-    #         return "圖片生成失敗：模型未回傳數據。"
-    # except Exception as e:
-    #     return f"圖片生成失敗: {e}"
     
     # Free api - pixazo.ai/getImage/v1/getSDXLImage
     try:
@@ -246,13 +224,9 @@ def handle_message(event):
             image_path = previous_message["content"]
             user_text = f"請分析這張圖片 {image_path}，問題是：{user_text}"
             user_message_history[user_id].pop()
-        else:
-            agent_input = {"messages": [("user", user_text)]}
-        # try:
-        #     response = agent_executor.invoke(agent_input)
-        #     out = response["output"]
 
-            # 1. 修正輸入格式：LangGraph 預期接收 'messages'，格式為 (角色, 內容) 的列表
+        agent_input = {"messages": [("user", user_text)]}
+
         try:
             response = agent_executor.invoke(agent_input)