linted

gemini.py

@@ -25,27 +25,27 @@ from linebot.v3.messaging import (
     TextMessage,
 )
 from linebot.v3.webhooks import (
-    ImageMessageContent, 
-    MessageEvent, 
+    ImageMessageContent,
+    MessageEvent,
     TextMessageContent,
 )
 
 from PIL import Image
 
-
 # === 初始化 Google Gemini ===
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 client = genai.Client(api_key=GOOGLE_API_KEY)
 
 google_search_tool = Tool(
-    google_search = GoogleSearch()
+    google_search=GoogleSearch()
 )
 
-chat = client.chats.create(model="gemini-2.5-pro-preview-05-06",
+chat = client.chats.create(
+    model="gemini-2.5-pro-preview-05-06",
     config=GenerateContentConfig(
         system_instruction="你是一個中文的AI助手，請用繁體中文回答",
         tools=[google_search_tool],
-        response_modalities=["TEXT"], 
+        response_modalities=["TEXT"],
     )
 )
 
@@ -57,7 +57,8 @@ base_url = os.getenv("SPACE_HOST")  # e.g., "your-space-name.hf.space"
 # === Flask 應用初始化 ===
 app = Flask(__name__)
 logging.basicConfig(
-    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+    level=logging.INFO,
+    format="%(asctime)s - %(levelname)s - %(message)s"
 )
 app.logger.setLevel(logging.INFO)
 
@@ -73,6 +74,7 @@ def query(payload):
     response = chat.send_message(message=payload)
     return response.text
 
+
 # === 靜態圖檔路由 ===
 @app.route("/images/<filename>")
 def serve_image(filename):
@@ -142,7 +144,7 @@ def handle_text_message(event):
                                 ],
                             )
                         )
-                    
+
         except Exception as e:
             app.logger.error(f"Gemini API error: {e}")
             with ApiClient(configuration) as api_client:
@@ -196,7 +198,6 @@ def handle_image_message(event):
     app.logger.info(response.text)
 
     # === 以下是回傳圖片部分 === #
-
     with ApiClient(configuration) as api_client:
         line_bot_api = MessagingApi(api_client)
         line_bot_api.reply_message(
@@ -209,70 +210,4 @@ def handle_image_message(event):
                     TextMessage(text=response.text),
                 ],
             )
-        )
-'''
-@handler.add(MessageEvent, message=ImageMessageContent)
-def handle_image_message(event):
-
-    # === 以下是處理圖片回傳部分 === #
-
-    with ApiClient(configuration) as api_client:
-        blob_api = MessagingApiBlob(api_client)
-        content = blob_api.get_message_content(message_id=event.message.id)
-        image_bytes = content
-
-    # Step 2：轉成 base64 字串
-    base64_string = base64.b64encode(image_bytes).decode("utf-8")
-
-    # Step 3：組成 OpenAI 的 data URI 格式
-    data_uri = f"data:image/png;base64,{base64_string}"
-    app.logger.info(f"Data URI: {data_uri}")
-
-    # Step 4：將圖片存到本地端
-    with tempfile.NamedTemporaryFile(
-        dir=static_tmp_path, suffix=".jpg", delete=False
-    ) as tf:
-        tf.write(content)
-        filename = os.path.basename(tf.name)
-
-    image_url = f"https://{base_url}/images/{filename}"
-
-    app.logger.info(f"Image URL: {image_url}")
-
-    # === 以下是處理解釋圖片部分 === #
-    response = client.responses.create(
-        model="gpt-4.1-nano",
-        input=[
-            {
-                "role": "user",
-                "content": [
-                    {
-                        "type": "input_text",
-                        "text": "describe the image in traditional chinese",
-                    },
-                    {
-                        "type": "input_image",
-                        "image_url": data_uri,
-                    },
-                ],
-            }
-        ],
-    )
-    app.logger.info(response.output_text)
-
-    # === 以下是回傳圖片部分 === #
-
-    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
-                    ),
-                    TextMessage(text=response.output_text),
-                ],
-            )
-        )
-'''
+        )