Upload 2 files

gemini_ai.py

@@ -0,0 +1,139 @@
+#!pip install -q -U google-generativeai
+import google.generativeai as genai
+import PIL.Image
+import image_converter as img_converter
+import random
+#基本設定都放這邊----------------------------------------
+#
+#
+# 設定圖檔位置 (此處僅為範例，純文字查詢時可忽略)
+image_path = r'G:\Python\tools\input_images\1411032041-楊智翔ai生成.webp'
+
+
+# 要使用的模型種類，免費版一分鐘只能跑最多十筆
+gemini_model = 'gemini-2.5-flash'
+
+target_JSON = {
+    "物理環境": ["辦公室", "臥室", "工作室", "工廠"],
+    "技術應用": ["人工智慧", "虛擬實境", "大數據分析", "其他"],
+    "社交關係": ["獨立工作(1人)", "團隊合作(2人以上)", "遠程協作(遠端控制)"],
+    "職業情感": ["快樂", "睡覺", "壓力/焦慮", "成就感"],
+    "資訊設備": ["AI助手", "投影儀", "手機", "眼鏡投影", 
+                "智慧手錶", "機械手臂", "平板", 
+                "電腦", "鍵盤", "滑鼠", "其他"],
+    "物體": ["床", "椅子", "桌子", 
+            "書架", "PC", 
+            "肖像", "監視器",
+            "窗戶",  "冷氣機","其他"],
+    "角色": ["機器人","教師","學生","動物","工作人員"]
+}
+
+
+#要求AI的提示語放這邊
+# image_prompt = """您現在扮演一位圖片分類大師，擅長解讀圖片中的一些抽象涵義並加以分類。
+# 請在各大類中選最近似的一樣，輸出結果如範例:"A[開心],B[學習],C[學校]"。
+# 若您覺得，該圖片不具上列特徵，請回覆"A[NIL]",加上NIL表示該類未再提供的選項內。
+# 以下是我們要請您分辨的種類:
+# A情感類-人物表情: A[面無表情,開心,生氣,悲傷,緊張,輕視，想睡，疲憊，興奮，自信滿滿,臉部遮蔽]。
+# B動作類-B[學習,工作,飲食,遊戲,駕駛,睡覺,冥想,醫療行為,會議,團隊討論，聽音樂，看電視，畫畫，騎車，烹飪，走路]。
+# C場景類-C[辦公室等工作空間,書房,臥室,客廳,學校,網咖,超現實場景，車內，外太空]。"""
+
+# image_prompt = """您現在扮演一位圖片分類大師，擅長解讀圖片中的一些抽象涵義並加以分類。
+# 請在各大類中選最近似的一樣，輸出結果如範例:"物理環境[辦公室],技術應用[人工智慧,虛擬實境,其他],資訊設備[其他]"。
+# 若您覺得，該圖片不具上列特徵，請回覆"XXX[NIL]",XXX為該類別,加上NIL表示該類未再提供的選項內。
+# 以下是我們要請您分辨的種類，會以JSON標示:
+# 物理環境[辦公室,臥室,工作室,工廠]。
+# 技術應用[人工智慧,虛擬實境,大數據分析,其他]。
+# 社交關係[獨立工作(1人),,團隊合作(2人以上),遠程協作(遠端控制)]。
+# 職業情感[快樂,睡覺,壓力/焦慮,成就感]。
+# 資訊設備[AI助手,投影儀,手機,眼鏡投影,智慧手錶,機械手臂,平板,電腦,鍵盤,滑鼠,其他]。
+# 物體[床,椅子,桌子,書架,PC,肖像,監視器,窗戶,冷氣機,其他]。
+# 角色[機器人,教師,學生,動物,工作人員]。
+# """
+
+image_prompt = """您現在扮演一位圖片分類大師，擅長解讀圖片中的一些抽象涵義並加以分類。
+請在各大類中選最近似的一樣，輸出結果如範例:"物理環境[辦公室],技術應用[人工智慧,虛擬實境,其他],資訊設備[其他]"。
+若您覺得，該圖片不具上列特徵，請回覆"XXX[NIL]",XXX為該類別,加上NIL表示該類未再提供的選項內。
+以下是我們要請您分辨的種類，會以JSON標示:"""+ str(target_JSON)
+
+#--------------------------------------------------------
+## 替換冒號和逗號為換行符號
+def replace_colon_comma_with_newline(input_string):
+  processed_string = input_string.replace(':', '\n').replace('],', ']\n')
+  return processed_string
+
+
+
+
+
+# function，輸入是文字或是圖檔的位置
+def analyze_content_with_gemini(input_content,target_JSON=target_JSON):
+    """
+    透過 Gemini API 辨識內容，可處理純文字或圖片。
+
+    Args:
+        input_content (str or PIL.Image.Image):
+            如果輸入是字串，則代表要辨識的文字訊息或圖片路徑。
+            如果輸入是 PIL.Image.Image 物件，則直接使用該圖片。
+
+    Returns:
+        str: 辨識結果的文字描述。
+    """
+    # 請將 'YOUR_API_KEY' 替換為您的實際 API 金鑰。
+    my_api_key = ['AIzaSyC6nBDxCuiE5GzBdTRQd-roYqVCGYCRy5M','AIzaSyDKHts9C72a68x58z1ItSRxgIU65UKN_xw','AIzaSyCgUnkkgAsBpsfrKe2Lqy5WgAbP0ktxKbg']  # 如果有多個金鑰，可以放在列表中
+    my_api_key = random.choice(my_api_key)  # 隨機選擇一個金鑰避免同時間大量使用同一個金鑰會被停用API服務
+    #print(f"使用的API金鑰: {my_api_key}")
+    genai.configure(api_key=my_api_key)
+    
+    try:
+        # 判斷輸入的類型
+        if isinstance(input_content, str):
+            # 如果輸入是字串��嘗試判斷是否為圖片路徑
+            # 這是一個簡易判斷，若您確定只傳純文字，可簡化此處邏輯。
+            if input_content.lower().endswith(('.png', '.jpg', '.jpeg', '.gif','.webp')):
+                if input_content.lower().endswith(('.webp')):
+                    input_content = img_converter.convert_webp_to_jpg(input_content)  # 如果是 webp 圖片，先轉換為 jpg
+                
+                # 使用能處理圖片的模型
+                model = genai.GenerativeModel(gemini_model)
+                image_obj = PIL.Image.open(input_content)
+
+                #response = model.generate_content(["這張圖片是什麼？請詳細描述。", image_obj])
+                response = model.generate_content([image_prompt, image_obj])
+            else:
+                # 使用純文字模型
+                model = genai.GenerativeModel(gemini_model)
+                response = model.generate_content(input_content)
+        elif isinstance(input_content, PIL.Image.Image):
+            # 使用能處理圖片的模型
+            model = genai.GenerativeModel(gemini_model)
+
+            #response = model.generate_content(["這張圖片是什麼？請詳細描述。", input_content])
+            response = model.generate_content([image_prompt, input_content])
+        else:
+            return "錯誤：輸入必須是文字、圖片路徑（字串）或 PIL.Image 物件。"
+
+        return replace_colon_comma_with_newline(response.text)
+
+    except Exception as e:
+        return f"發生錯誤：{e}"
+
+
+if __name__ == '__main__':
+    # --- 程式碼使用範例 ---
+
+    # 範例 1：傳送純文字訊息
+    # print("正在處理純文字訊息...")
+    # text_message = "你好，請簡要說明一下Python是什麼？"
+    # response_text = analyze_content_with_gemini(text_message)
+    # print("回應結果：")
+    # print(response_text)
+    # print("-" * 20)
+
+    # 範例 2：傳送圖片路徑
+    # 請確保 image_path 指向有效的圖片檔案
+    print("正在處理圖片訊息...")
+    response_image = analyze_content_with_gemini(image_path)
+    print("回應結果：")
+    print(response_image)
+    print("-" * 20)

image_converter.py

@@ -0,0 +1,62 @@
+# 這個程式將 webp 圖片轉換為 jpg 格式， 
+# 並儲存到指定的資料夾或與原檔案相同的資料夾中。
+# 使用 PIL 庫來處理圖片格式轉換。
+
+from PIL import Image
+import os
+
+def convert_webp_to_jpg(webp_path, output_folder=None):
+    """
+    將 webp 檔案轉換為 jpg 檔案。
+
+    :param webp_path: 輸入的 webp 檔案路徑。
+    :param output_folder: 輸出的資料夾路徑。如果為 None，則輸出到與輸入檔案相同的資料夾。
+    :return: 輸出的 jpg 檔案路徑。
+    """
+    try:
+        # 開啟 webp 圖片
+        img = Image.open(webp_path).convert("RGB")
+        
+        # 決定輸出的檔案名稱與路徑
+        file_name = os.path.splitext(os.path.basename(webp_path))[0]
+        if output_folder:
+            if not os.path.exists(output_folder):
+                os.makedirs(output_folder)
+            output_path = os.path.join(output_folder, f"{file_name}.jpg")
+        else:
+            output_path = os.path.join(os.path.dirname(webp_path), f"{file_name}.jpg")
+            
+        # 儲存為 jpg
+        img.save(output_path, "jpeg")
+        
+        print(f"成功將 {webp_path} 轉換為 {output_path}")
+        return output_path
+    except Exception as e:
+        print(f"轉換失敗：{e}")
+        return None
+
+if __name__ == '__main__':
+    
+    # 建立一個假的 webp 檔案以供測試
+    if not os.path.exists("input_images"):
+        os.makedirs("input_images")
+    
+    try:
+        # 建立一個簡單的白色圖片
+        #G:\Python\tools\input_images\1411032040-楊宗祥.webp
+        dummy_webp_path = r"G:\Python\tools\input_images\1411032040-楊宗祥.webp"
+
+        # 測試轉換函數
+        # 範例 1: 轉換並儲存在相同資料夾
+        print("\n--- 範例 1: 轉換並儲存在相同資料夾 ---")
+        output_path = convert_webp_to_jpg(dummy_webp_path)
+
+        # 範例 2: 轉換並儲存在指定資料夾
+        print("\n--- 範例 2: 轉換並儲存在指定資料夾 ---")
+        if not os.path.exists("output_images"):
+            os.makedirs("output_images")
+        output_path = convert_webp_to_jpg(dummy_webp_path, "output_images")
+
+        print(output_path)
+    except Exception as e:
+        print(f"執行範例時發生錯誤: {e}")