Create image_processor.py

image_processor.py

@@ -0,0 +1,33 @@
+# image_processor.py
+import google.generativeai as genai
+from PIL import Image
+import config
+import os
+
+# 初始化 Gemini API
+if config.GEMINI_KEY:
+    genai.configure(api_key=config.GEMINI_KEY)
+
+def extract_text(image_path):
+    """
+    眼睛模組：讀取圖片並擷取所有文字 (OCR)
+    """
+    if not config.GEMINI_KEY:
+        print("⚠️ 未設定 GEMINI_KEY，無法啟用視覺功能")
+        return None
+
+    try:
+        # 呼叫 Gemini 1.5 Flash 模型 (速度最快，適合 OCR)
+        model = genai.GenerativeModel('gemini-1.5-flash')
+        
+        # 開啟圖片
+        img = Image.open(image_path)
+        
+        # 給 Gemini 的指令：只要抓字，不要自己亂加戲或翻譯
+        prompt = "請幫我辨識這張圖片中的所有文字（包含中文、英文字母或族語羅馬拼音）。請直接回覆原文，保持原本的換行格式，不要自行翻譯、解釋或推測。"
+        
+        response = model.generate_content([prompt, img])
+        return response.text.strip()
+    except Exception as e:
+        print(f"圖片辨識失敗: {e}")
+        return None