首次上傳 LINE Bot

gemini.py

@@ -3,7 +3,7 @@ import logging
 import os
 import tempfile
 import uuid
-from io import BytesIO
+from io import BytesIO 
 
 import markdown
 from bs4 import BeautifulSoup
@@ -44,7 +44,11 @@ google_search_tool = Tool(
 chat = client.chats.create(
     model="gemini-2.0-flash",
     config=GenerateContentConfig(
-        system_instruction="你是一個中文的AI助手，請用繁體中文回答",
+        system_instruction=(
+        "你是一位專業健身教練與營養顧問，擁有多年重量訓練與健身飲食指導經驗。"
+        "請使用繁體中文，針對使用者的健身問題提供專業建議，包含動作教學、訓練計畫、飲食建議與常見錯誤修正等。"
+    ),
+
         tools=[google_search_tool],
         response_modalities=["TEXT"],
     )
@@ -220,63 +224,115 @@ def handle_image_message(event):
 
 # === 處理影片訊息 ===
 
-@handler.add(MessageEvent, message=VideoMessageContent)
-def handle_video_message(event):
-    # 下載影片內容
-    with ApiClient(configuration) as api_client:
-        blob_api = MessagingApiBlob(api_client)
-        video_data = blob_api.get_message_content(message_id=event.message.id)
+@handler.add(MessageEvent, message=TextMessageContent)
+def handle_text_message(event):
+    user_input = event.message.text.strip()
+
+    # === 使用 Gemini 生成圖片（AI xxx） ===
+    if user_input.startswith("AI "):
+        prompt = user_input[3:].strip()
+        try:
+            response = client.models.generate_content(
+                model="gemini-2.0-flash-exp-image-generation",
+                contents=prompt,
+                config=types.GenerateContentConfig(
+                    response_modalities=["TEXT", "IMAGE"]
+                ),
+            )
+
+            for part in response.candidates[0].content.parts:
+                if part.inline_data is not None:
+                    image = Image.open(BytesIO(part.inline_data.data))
+                    filename = f"{uuid.uuid4().hex}.png"
+                    image_path = os.path.join(static_tmp_path, filename)
+                    image.save(image_path)
+
+                    image_url = f"https://{base_url}/images/{filename}"
+                    app.logger.info(f"Image URL: {image_url}")
+
+                    with ApiClient(configuration) as api_client:
+                        line_bot_api = MessagingApi(api_client)
+                        line_bot_api.reply_message(
+                            ReplyMessageRequest(
+                                reply_token=event.reply_token,
+                                messages=[
+                                    ImageMessage(
+                                        original_content_url=image_url,
+                                        preview_image_url=image_url,
+                                    )
+                                ],
+                            )
+                        )
+
+        except Exception as e:
+            app.logger.error(f"Gemini API error: {e}")
+            with ApiClient(configuration) as api_client:
+                line_bot_api = MessagingApi(api_client)
+                line_bot_api.reply_message(
+                    ReplyMessageRequest(
+                        reply_token=event.reply_token,
+                        messages=[TextMessage(text="抱歉，生成圖片時發生錯誤。")],
+                    )
+                )
+
+    # === 處理「菜單 xxx」功能 ===
+    elif user_input.startswith("菜單 "):
+        muscle_group = user_input[3:].strip()
+        if not muscle_group:
+            with ApiClient(configuration) as api_client:
+                line_bot_api = MessagingApi(api_client)
+                line_bot_api.reply_message(
+                    ReplyMessageRequest(
+                        reply_token=event.reply_token,
+                        messages=[TextMessage(text="請輸入要安排的部位，例如：菜單 胸肌、菜單 臀部")]
+                    )
+                )
+            return
+
+        # 同義詞簡化
+        synonym_map = {
+            "胸": "胸肌",
+            "腿": "腿部",
+            "肩": "肩膀",
+            "背": "背肌",
+            "手": "手臂",
+            "手臂": "手臂",
+            "核心": "核心肌群",
+            "臀": "臀部",
+            "臀部": "臀部",
+            "全身": "全身初學者",
+            "初學者": "全身初學者",
+        }
+        muscle_group = synonym_map.get(muscle_group, muscle_group)
+
+        # 發送訓練菜單請求
+        prompt = (
+            f"請依據「{muscle_group}」提供一份健身訓練菜單。"
+            "每份菜單包含 3~5 個動作，建議組數與次數，以及適當的休息時間。"
+            "請以繁體中文簡潔列出，使用條列方式排版，適合 LINE 顯示格式。"
+        )
+        response = query(prompt)
 
-    # 儲存影片到本地
-    if video_data is None:
-        err_msg = "抱歉，無法取得影片內容。"
-        app.logger.error(err_msg)
         with ApiClient(configuration) as api_client:
             line_bot_api = MessagingApi(api_client)
             line_bot_api.reply_message(
                 ReplyMessageRequest(
                     reply_token=event.reply_token,
-                    messages=[TextMessage(text=err_msg)]
+                    messages=[TextMessage(text=response)],
                 )
             )
-        return
-
-    with tempfile.NamedTemporaryFile(
-        dir=static_tmp_path, suffix=".mp4", delete=False
-    ) as tf:
-        tf.write(video_data)
-        filename = os.path.basename(tf.name)
 
-    video_url = f"https://{base_url}/images/{filename}"
-    app.logger.info(f"Video URL: {video_url}")
-
-    # 影片說明
-    try:
-        from io import BytesIO
-        video_bytes = BytesIO(video_data)
-        response = client.models.generate_content(
-            model="gemini-2.5-flash-preview-05-20",
-            config=types.GenerateContentConfig(
-                system_instruction="你是一個專業的影片解說員，請用繁體中文簡要說明這段影片的內容。",
-                response_modalities=["TEXT"],
-                tools=[google_search_tool],
-            ),
-            contents=[video_bytes, "用繁體中文描述這段影片"],
-        )
-        description = response.text
-    except Exception as e:
-        app.logger.error(f"Gemini API error (video): {e}")
-        description = "抱歉，無法解釋這段影片內容。"
+    # === 處理一般文字訊息 ===
+    else:
+        with ApiClient(configuration) as api_client:
+            line_bot_api = MessagingApi(api_client)
+            response = query(event.message.text)
+            html_msg = markdown.markdown(response)
+            soup = BeautifulSoup(html_msg, "html.parser")
 
-    # 回傳影片連結與說明
-    with ApiClient(configuration) as api_client:
-        line_bot_api = MessagingApi(api_client)
-        line_bot_api.reply_message(
-            ReplyMessageRequest(
-                reply_token=event.reply_token,
-                messages=[
-                    TextMessage(text=f"影片連結：{video_url}"),
-                    TextMessage(text=description),
-                ],
+            line_bot_api.reply_message_with_http_info(
+                ReplyMessageRequest(
+                    reply_token=event.reply_token,
+                    messages=[TextMessage(text=soup.get_text())],
+                )
             )
-        )