Evaaaaa06 commited on
Commit
4307798
·
verified ·
1 Parent(s): 9187e6e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -21
main.py CHANGED
@@ -91,29 +91,13 @@ def analyze_with_gemini(image_path, user_text):
91
  分析用戶問題和圖片,並返回 Gemini 的回應
92
  """
93
  try:
94
- # # 確保圖片存在
95
- # if not os.path.exists(image_path):
96
- # raise FileNotFoundError(f"圖片路徑無效:{image_path}")
97
 
98
- # # 打開本地圖片文件並將其轉換為 Base64 編碼
99
- # with open(image_path, "rb") as image_file:
100
- # image_binary = image_file.read()
101
-
102
- # # 將二進制數據編碼為 Base64
103
- # image_base64 = base64.b64encode(image_binary).decode('utf-8')
104
  organ = PIL.Image.open(image_path)
105
- response = model.generate_content([user_text, organ])
106
 
107
- # # 構建 Gemini 的請求
108
- # prompt = f"""
109
- # 用戶的問題是:{user_text}
110
- # 圖片已上傳,Base64 編碼如下:
111
- # {image_base64}
112
- # 請根據圖片和問題進行詳細分析。
113
- # """
114
- # # 使用 Gemini API 請求
115
- # response = model.generate_content([{'mime_type':'image/png', 'data': image_base64}, prompt])
116
-
117
  # 提取回應內容
118
  return response.parts[0].text
119
 
@@ -181,7 +165,7 @@ def handle_image_message(event):
181
  prompt = event.message.text
182
  store_user_message(user_id, "text", prompt)
183
  # 使用 Gemini 模型生成文字
184
- completion = model.generate_content(prompt, generation_config=generation_config)
185
  # 檢查生成結果是否為空
186
  if (completion.parts[0].text != None):
187
  # 取得生成結果
 
91
  分析用戶問題和圖片,並返回 Gemini 的回應
92
  """
93
  try:
94
+ # 確保圖片存在
95
+ if not os.path.exists(image_path):
96
+ raise FileNotFoundError(f"圖片路徑無效:{image_path}")
97
 
 
 
 
 
 
 
98
  organ = PIL.Image.open(image_path)
99
+ response = model.start_chat([user_text, organ])
100
 
 
 
 
 
 
 
 
 
 
 
101
  # 提取回應內容
102
  return response.parts[0].text
103
 
 
165
  prompt = event.message.text
166
  store_user_message(user_id, "text", prompt)
167
  # 使用 Gemini 模型生成文字
168
+ completion = model.start_chat(prompt, generation_config=generation_config)
169
  # 檢查生成結果是否為空
170
  if (completion.parts[0].text != None):
171
  # 取得生成結果