Update main.py

main.py

@@ -13,12 +13,10 @@ from linebot.models import (
     AudioMessage,
     TemplateSendMessage,
     ButtonsTemplate,
-    MessageTemplateAction,
-    URIAction
+    MessageTemplateAction
 )
 from dotenv import load_dotenv
 import asyncio
-import requests
 
 # 從 .env 檔案載入環境變數
 load_dotenv()
@@ -170,53 +168,20 @@ def handle_text_message(event):
     try:
         prompt = event.message.text
         logging.info(f"收到使用者 {user_id} 的訊息: {prompt}")
-        # call Bio-Medical-Llama-3-8B API
-        response_text = call_llama_api(prompt)
-        if response_text:
-            out = response_text
-            logging.info(f"Bio-Medical-Llama-3-8B 回覆使用者 {user_id}: {out}")
+        chat_session = get_or_create_chat_session(user_id)
+        response = chat_session.send_message(prompt)
+        if response.text:
+            out = response.text
+            logging.info(f"Gemini 回覆使用者 {user_id}: {out}")
         else:
-            out = "Bio-Medical-Llama-3-8B 沒有回覆！請換個說法！"
-            logging.warning(f"Bio-Medical-Llama-3-8B 沒有回覆文字內容給使用者 {user_id}")
+            out = "Gemini 沒答案！請換個說法！"
+            logging.warning(f"Gemini 沒有回覆文字內容給使用者 {user_id}")
     except Exception as e:
-        out = "Bio-Medical-Llama-3-8B 執行出錯！請換個說法！"
-        logging.error(f"Bio-Medical-Llama-3-8B 發生錯誤 (使用者: {user_id}): {e}")
+        out = "Gemini 執行出錯！請換個說法！"
+        logging.error(f"Gemini 發生錯誤 (使用者: {user_id}): {e}")
     
     line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
 
-# --- 5.  呼叫 Hugging Face API ---
-def call_llama_api(prompt: str):
-    """
-    呼叫 Hugging Face API，取得 Bio-Medical-Llama-3-8B 的回覆
-    """
-    api_url = "https://api-inference.huggingface.co/models/ContactDoctor/Bio-Medical-Llama-3-8B"
-    headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_API_KEY')}"} # 請在.env新增HUGGINGFACE_API_KEY
-    payload = {
-        "inputs": prompt,
-        "parameters": {
-            "max_length": 512, # 調整輸出長度
-            "temperature": 0.7,
-            "top_p": 0.9,
-            "top_k": 50
-        }
-    }
-
-    try:
-        response = requests.post(api_url, headers=headers, json=payload)
-        response.raise_for_status()  # 如果發生錯誤就拋出 exception
-        response_json = response.json()
-        if isinstance(response_json, list) and response_json:
-          return response_json[0].get("generated_text") # 返回模型產生的文字
-        else:
-          return None # 如果返回格式錯誤，則返回 None
-    except requests.exceptions.RequestException as e:
-        logging.error(f"呼叫 Hugging Face API 發生錯誤：{e}")
-        return None
-    except Exception as e:
-      logging.error(f"處理 Hugging Face API 回應時發生錯誤：{e}")
-      return None
-
-
 @line_handler.add(MessageEvent, message=TextMessage)
 def handle_message(event):
     """
@@ -246,9 +211,10 @@ def handle_message(event):
         line_bot_api.reply_message(event.reply_token, buttons_template_message)
         return
 
+
     handle_text_message(event)
 
-# --- 6. 啟動應用程式 ---
+# --- 5. 啟動應用程式 ---
 if __name__ == "__main__":
     import uvicorn