Update main.py

main.py

@@ -13,7 +13,7 @@ import PIL.Image
 
 from imgurpython import ImgurClient
 from Image_text_generation import Image_text_Generator
-from Uploading_images import get_image_url, store_user_message, analyze_with_gemini, get_previous_message
+from Uploading_images_file import get_image_url,get_file_url, store_user_message, analyze_with_gemini, get_previous_message
 
 #==========================
 #  API 金鑰
@@ -129,6 +129,47 @@ def handle_image_message(event):
                 TextSendMessage(text="⚠️ 圖片生成失敗，請稍後再試~")
             )
         return
+        
+    # ========
+    # 傳文件
+    # ========
+    if event.message.type == "file":
+        file_name = event.message.file_name
+        file_path = get_file_url(event.message.id, file_name, file_type="pdf")
+    
+        if file_path:
+            file_generator = Image_text_Generator()  # 確保這個類別已正確初始化
+            store_user_message(user_id, "file", file_path)
+            line_bot_api.reply_message(
+                event.reply_token,
+                TextSendMessage(text="📄 文件已接收成功囉，幫我輸入你想詢問的問題喔～")
+            )
+        else:
+            line_bot_api.reply_message(
+                event.reply_token,
+                TextSendMessage(text="❌ 沒有成功接收到文件，請再試一次。")
+            )
+            return
+
+    # 使用者在傳完文件後輸入問題
+    previous_message = get_previous_message(user_id)
+    if previous_message and previous_message["type"] == "file" and event.message.type == "text":
+        file_path = previous_message["content"]
+        user_text = event.message.text
+        store_user_message(user_id, "text", user_text)
+
+        try:
+            if not os.path.exists(file_path):
+                raise FileNotFoundError(f"📂 檔案路徑無效：{file_path}")
+
+            file_generator = Image_text_Generator(imgur)
+            completion = file_generator.generate_txt_with_gemini([user_text, file_path])
+
+            # 若是 Gemini 回傳的內容需取 text
+            out = completion.text if hasattr(completion, "text") else str(completion)
+
+        except Exception as e:
+            out = f"⚠️ 發生錯誤：{e}"
 
 
     # ========