Update main.py

main.py

@@ -90,9 +90,9 @@ def store_user_message(user_id, message_type, message_content):
         "content": message_content
     })
     
-def analyze_with_openai(image_path, user_text):
+def analyze_with_gemini(image_path, user_text):
     """
-    分析用戶問題和圖片，並返回 OpenAI 的回應
+    分析用戶問題和圖片，並返回 Gemini 的回應
     """
     try:
         # 確保圖片存在
@@ -104,7 +104,7 @@ def analyze_with_openai(image_path, user_text):
             image_binary = image_file.read()
             image_base64 = base64.b64encode(image_binary).decode("utf-8")
 
-        # 構建分析請求
+        # 構建 Gemini 的請求
         prompt = f"""
         用戶的問題是：{user_text}
         圖片已上傳，Base64 編碼如下（部分截取）：
@@ -112,27 +112,14 @@ def analyze_with_openai(image_path, user_text):
         請根據圖片和問題進行詳細分析。
         """
 
-        # 使用 OpenAI API 請求
-        response = openai.ChatCompletion.create(
-            model="gpt-4",
-            messages=[
-                {"role": "system", "content": "請用繁體中文回答。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題"},
-                {"role": "user", "content": prompt}
-            ]
-        )
+        # 使用 Gemini API 請求
+        response = model.generate_content(prompt, generation_config=generation_config)
 
         # 提取回應內容
-        return response["choices"][0]["message"]["content"]
-
-    except FileNotFoundError as fnfe:
-        print(f"[ERROR] 文件錯誤：{fnfe}")
-        return "圖片路徑無效，請確認後重新上傳！"
-    except openai.error.OpenAIError as oe:
-        print(f"[ERROR] OpenAI API 錯誤：{oe}")
-        return "OpenAI 分析失敗，請稍後再試！"
+        return response.parts[0].text
+
     except Exception as e:
-        print(f"[ERROR] 未知錯誤：{e}")
-        return "發生未知錯誤，請稍後再試！"
+        return f"發生錯誤: {e}"
     
 
 def get_previous_message(user_id):
@@ -171,7 +158,7 @@ def handle_image_message(event):
         store_user_message(user_id, "text", user_text)
 
         # 結合圖片與文字分析
-        out = analyze_with_openai(image_path, user_text)
+        out = analyze_with_gemini(image_path, user_text)
 
     else:
         global working_status