Update main.py

main.py

@@ -1,48 +1,36 @@
 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
-)
+generation_config = genai.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
 
-# 使用 Gemini 模型
-model = genai.GenerativeModel(
-    "gemini-2.0-flash-exp",
-    system_instruction="主要用繁體中文回答，但如果用戶使用詢問英文問題，就用英文回應。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題",
-    generation_config=generation_config,
-)
+# 使用 Gemini-1.5-flash 模型
+model = genai.GenerativeModel('gemini-2.0-flash-exp', system_instruction="主要用繁體中文回答，但如果用戶使用詢問英文問題，就用英文回應。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題", generation_config=generation_config)
 
-# 初始化 LINE Bot
+# 設定 Line Bot 的 API 金鑰和秘密金鑰
 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=["*"],
@@ -51,29 +39,48 @@ 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"
-
-# Imgur 圖片上傳功能（接收 image_binary，不用暫存）
-def upload_image_to_imgur_with_token(image_binary, access_token):
+    
+#==========================
+# 使用者請求生成圖片
+#==========================
+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()
+
         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"]
@@ -84,7 +91,10 @@ def upload_image_to_imgur_with_token(image_binary, access_token):
         print("圖片上傳例外：", e)
         return None
 
-# 接收使用者圖片
+    
+#==========================
+# 使用者上傳圖片
+#==========================
 def get_image_url(message_id):
     try:
         message_content = line_bot_api.get_message_content(message_id)
@@ -97,31 +107,48 @@ def get_image_url(message_id):
         print(f"Error getting image: {e}")
         return None
 
-# 分析用戶圖片＋問題
-def analyze_with_gemini(image_path, user_text, chat):
+# 使用字典模擬用戶訊息歷史存儲
+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 的回應
+    """
     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):
@@ -133,24 +160,23 @@ 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)
-                    ]
-                )
-            else:
-                line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片上傳失敗，請稍後再試~"))
+            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="圖片生成失敗，請稍後再試~"))
         return
@@ -169,7 +195,14 @@ def handle_image_message(event):
         image_path = previous_message["content"]
         user_text = event.message.text
         store_user_message(user_id, "text", user_text)
-        out = 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)
+            completion = chat.send_message([user_text, organ])
+            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="請輸入文字或圖片~"))
@@ -188,8 +221,9 @@ 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)
 
 # 註解說明：