alanchen1115 commited on
Commit
91b9cb6
·
verified ·
1 Parent(s): 9f427aa

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -131,7 +131,7 @@ def handle_image_message(event):
131
  file_path = get_file_url(event.message.id, file_name, file_type="pdf")
132
 
133
  if file_path:
134
- file_generator = Image_text_Generator() # 確保這個類別已正確初始化
135
  store_user_message(user_id, "file", file_path)
136
  line_bot_api.reply_message(
137
  event.reply_token,
@@ -155,7 +155,7 @@ def handle_image_message(event):
155
  if not os.path.exists(file_path):
156
  raise FileNotFoundError(f"📂 檔案路徑無效:{file_path}")
157
 
158
- file_generator = Image_text_Generator(imgur)
159
  completion = file_generator.generate_txt_with_gemini([user_text, file_path])
160
 
161
  # 若是 Gemini 回傳的內容需取 text
 
131
  file_path = get_file_url(event.message.id, file_name, file_type="pdf")
132
 
133
  if file_path:
134
+ file_generator = Image_text_Generator(user_id) # 確保這個類別已正確初始化
135
  store_user_message(user_id, "file", file_path)
136
  line_bot_api.reply_message(
137
  event.reply_token,
 
155
  if not os.path.exists(file_path):
156
  raise FileNotFoundError(f"📂 檔案路徑無效:{file_path}")
157
 
158
+ file_generator = Image_text_Generator(user_id)
159
  completion = file_generator.generate_txt_with_gemini([user_text, file_path])
160
 
161
  # 若是 Gemini 回傳的內容需取 text