Update main.py

main.py

@@ -16,6 +16,12 @@ import httpx
 from imgurpython import ImgurClient
 from Image_generation import generate_image_with_gemini, upload_image_to_imgur
 
+from Uploading_images import get_image_url,store_user_message,analyze_with_gemini,get_previous_message
+
+#==========================
+#  API 金鑰
+#==========================
+
 # 設定 Google AI API 金鑰
 genai_gen.configure(api_key=os.environ["GOOGLE_API_KEY"])
 
@@ -35,6 +41,7 @@ client_secret = os.environ.get("IMGUR_CLIENT_SECRET")
 access_token = os.environ.get("IMGUR_ACCESS_TOKEN")
 refresh_token = os.environ.get("IMGUR_REFRESH_TOKEN")
 
+
 # 設定是否正在與使用者交談
 working_status = os.getenv("DEFALUT_TALKING", default = "true").lower() == "true"
 
@@ -86,6 +93,9 @@ def handle_image_message(event):
     chat = chat_sessions.get(user_id) or model.start_chat(history=[])
     chat_sessions[user_id] = chat
 
+    # ========
+    # 生成圖片
+    # ========
     user_text = event.message.text if event.message.type == "text" else None
     image_url = None
     if user_text and user_text.startswith("生成圖片"):
@@ -124,6 +134,9 @@ def handle_image_message(event):
         return
 
 
+    # ========
+    # 上傳圖片
+    # ========
     if event.message.type == "image":
         image_path = get_image_url(event.message.id)
         if image_path:
@@ -146,6 +159,10 @@ def handle_image_message(event):
             out = completion.text
         except Exception as e:
             out = f"發生錯誤: {e}"
+
+    # ========
+    # 純文字
+    # ========
     else:
         if event.message.type != "text":
             line_bot_api.reply_message(event.reply_token, TextSendMessage(text="請輸入文字或圖片~"))