Update main.py
Browse files
main.py
CHANGED
@@ -123,48 +123,6 @@ def handle_image_message(event):
|
|
123 |
)
|
124 |
return
|
125 |
|
126 |
-
# ========
|
127 |
-
# 傳文件
|
128 |
-
# ========
|
129 |
-
if event.message.type == "file":
|
130 |
-
file_name = event.message.file_name
|
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,
|
138 |
-
TextSendMessage(text="📄 文件已接收成功囉,幫我輸入你想詢問的問題喔~")
|
139 |
-
)
|
140 |
-
else:
|
141 |
-
line_bot_api.reply_message(
|
142 |
-
event.reply_token,
|
143 |
-
TextSendMessage(text="❌ 沒有成功接收到文件,請再試一次。")
|
144 |
-
)
|
145 |
-
return
|
146 |
-
|
147 |
-
# 使用者在傳完文件後輸入問題
|
148 |
-
previous_message = get_previous_message(user_id)
|
149 |
-
if previous_message and previous_message["type"] == "file" and event.message.type == "text":
|
150 |
-
file_path = previous_message["content"]
|
151 |
-
user_text = event.message.text
|
152 |
-
store_user_message(user_id, "text", user_text)
|
153 |
-
|
154 |
-
try:
|
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
|
162 |
-
out = completion.text if hasattr(completion, "text") else str(completion)
|
163 |
-
|
164 |
-
except Exception as e:
|
165 |
-
out = f"⚠️ 發生錯誤:{e}"
|
166 |
-
|
167 |
-
|
168 |
# ========
|
169 |
# 上傳圖片
|
170 |
# ========
|
|
|
123 |
)
|
124 |
return
|
125 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
126 |
# ========
|
127 |
# 上傳圖片
|
128 |
# ========
|