Update main.py

main.py

@@ -1,36 +1,48 @@
 import json, os
 import gradio as gr
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
+from fastapi import FastAPI, Request, Header, BackgroundTasks, HTTPException, status
 import google.generativeai as genai
 import base64
 from collections import defaultdict
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
-from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage, ImageMessage
+from linebot.models import (
+    MessageEvent,
+    TextMessage,
+    TextSendMessage,
+    ImageSendMessage,
+    AudioMessage,
+    ImageMessage,
+)
 import PIL.Image
 import tempfile
+import httpx
+import uvicorn
 
 # 設定 Google AI API 金鑰
 genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 
 # 設定生成文字的參數
-generation_config = genai.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
+generation_config = genai.types.GenerationConfig(
+    max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16
+)
 
-# 使用 Gemini-1.5-flash 模型
-model = genai.GenerativeModel('gemini-2.0-flash-exp', system_instruction="主要用繁體中文回答，但如果用戶使用詢問英文問題，就用英文回應。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題", generation_config=generation_config)
+# 使用 Gemini 模型
+model = genai.GenerativeModel(
+    "gemini-2.0-flash-exp",
+    system_instruction="主要用繁體中文回答，但如果用戶使用詢問英文問題，就用英文回應。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題",
+    generation_config=generation_config,
+)
 
-# 設定 Line Bot 的 API 金鑰和秘密金鑰
+# 初始化 LINE Bot
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
 line_handler = WebhookHandler(os.environ["CHANNEL_SECRET"])
 
-# 設定是否正在與使用者交談
-working_status = os.getenv("DEFALUT_TALKING", default = "true").lower() == "true"
-
 # 建立 FastAPI 應用程式
 app = FastAPI()
 
-# 設定 CORS，允許跨域請求
+# 設定 CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -39,48 +51,29 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# 處理根路徑請求
 @app.get("/")
 def root():
     return {"title": "Line Bot"}
 
-# 處理 Line Webhook 請求
 @app.post("/webhook")
-async def webhook(
-    request: Request,
-    background_tasks: BackgroundTasks,
-    x_line_signature=Header(None),
-):
-    # 取得請求內容
+async def webhook(request: Request, background_tasks: BackgroundTasks, x_line_signature=Header(None)):
     body = await request.body()
     try:
-        # 將處理 Line 事件的任務加入背景工作
         background_tasks.add_task(
             line_handler.handle, body.decode("utf-8"), x_line_signature
         )
     except InvalidSignatureError:
-        # 處理無效的簽章錯誤
         raise HTTPException(status_code=400, detail="Invalid signature")
     return "ok"
-    
-#==========================
-# 使用者請求生成圖片
-#==========================
-def upload_image_to_imgur_with_token(image_path, access_token):
-    """
-    使用 Imgur OAuth2 Access Token 上傳圖片
-    """
-    try:
-        with open(image_path, "rb") as img_file:
-            image_binary = img_file.read()
 
+# Imgur 圖片上傳功能（接收 image_binary，不用暫存）
+def upload_image_to_imgur_with_token(image_binary, access_token):
+    try:
         headers = {"Authorization": f"Bearer {access_token}"}
-
         data = {
             "image": base64.b64encode(image_binary).decode("utf-8"),
             "type": "base64",
         }
-
         response = httpx.post("https://api.imgur.com/3/image", headers=headers, data=data)
         if response.status_code == 200:
             return response.json()["data"]["link"]
@@ -91,10 +84,7 @@ def upload_image_to_imgur_with_token(image_path, access_token):
         print("圖片上傳例外：", e)
         return None
 
-    
-#==========================
-# 使用者上傳圖片
-#==========================
+# 接收使用者圖片
 def get_image_url(message_id):
     try:
         message_content = line_bot_api.get_message_content(message_id)
@@ -107,48 +97,31 @@ def get_image_url(message_id):
         print(f"Error getting image: {e}")
         return None
 
-# 使用字典模擬用戶訊息歷史存儲
-user_message_history = defaultdict(list)
-def store_user_message(user_id, message_type, message_content):
-    """
-    儲存用戶的訊息
-    """
-    user_message_history[user_id].append({
-        "type": message_type,
-        "content": message_content})
-    
-def analyze_with_gemini(image_path, user_text):
-    """
-    分析用戶問題和圖片，並返回 Gemini 的回應
-    """
+# 分析用戶圖片＋問題
+def analyze_with_gemini(image_path, user_text, chat):
     try:
-        # 確保圖片存在
         if not os.path.exists(image_path):
             raise FileNotFoundError(f"圖片路徑無效：{image_path}")
-        
         organ = PIL.Image.open(image_path)
         response = chat.send_message([user_text, organ])
-        
-        # 提取回應內容
         return response.text
-
     except Exception as e:
         return f"發生錯誤: {e}"
-    
+
+# 儲存使用者訊息
+user_message_history = defaultdict(list)
+def store_user_message(user_id, message_type, message_content):
+    user_message_history[user_id].append({
+        "type": message_type,
+        "content": message_content
+    })
 
 def get_previous_message(user_id):
-    """
-    獲取用戶的上一則訊息
-    """
     if user_id in user_message_history and len(user_message_history[user_id]) > 0:
-        # 返回最後一則訊息
         return user_message_history[user_id][-1]
     return None
 
-#==========================
-# 主程式(圖片與文字)
-#==========================
-# 建立 chat_sessions 字典
+# 主訊息處理邏輯
 chat_sessions = {}
 @line_handler.add(MessageEvent, message=(ImageMessage, TextMessage))
 def handle_image_message(event):
@@ -160,23 +133,24 @@ def handle_image_message(event):
 
     if user_text and user_text.startswith("請幫我生成圖片"):
         prompt = user_text.replace("請幫我生成圖片", "").strip()
-
         image_model = genai.GenerativeModel("gemini-2.0-flash-exp-image-generation")
         response = image_model.generate_content(prompt)
 
         if response.parts and hasattr(response.parts[0], "inline_data"):
             image_data = response.parts[0].inline_data.data
             access_token = os.environ.get("IMGUR_ACCESS_TOKEN")
-            image_url = upload_image_to_imgur_with_token(image_data,access_token)
-
-        if image_url:
-            line_bot_api.reply_message(
-                event.reply_token,
-                [
-                    TextSendMessage(text="這是我為你生成的圖片喔～ ✨"),
-                    ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
-                ]
-            )
+            image_url = upload_image_to_imgur_with_token(image_data, access_token)
+
+            if image_url:
+                line_bot_api.reply_message(
+                    event.reply_token,
+                    [
+                        TextSendMessage(text="這是我為你生成的圖片喔～ ✨"),
+                        ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
+                    ]
+                )
+            else:
+                line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片上傳失敗，請稍後再試~"))
         else:
             line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成失敗，請稍後再試~"))
         return
@@ -195,14 +169,7 @@ def handle_image_message(event):
         image_path = previous_message["content"]
         user_text = event.message.text
         store_user_message(user_id, "text", user_text)
-        try:
-            if not os.path.exists(image_path):
-                raise FileNotFoundError(f"圖片路徑無效：{image_path}")
-            organ = PIL.Image.open(image_path)
-            completion = chat.send_message([user_text, organ])
-            out = completion.text
-        except Exception as e:
-            out = f"發生錯誤: {e}"
+        out = analyze_with_gemini(image_path, user_text, chat)
     else:
         if event.message.type != "text":
             line_bot_api.reply_message(event.reply_token, TextSendMessage(text="請輸入文字或圖片~"))
@@ -221,9 +188,8 @@ def handle_image_message(event):
 
     line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
 
-
+# 啟動應用
 if __name__ == "__main__":
-    # 啟動 FastAPI 應用程式
     uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=True)
 
 # 註解說明：