Update main.py

main.py

@@ -1,30 +1,31 @@
+import json, os, pathlib, glob
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status, staticfiles
-from gradio_client import Client
-import json
-import os
+from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
+from google import genai
+from linebot import LineBotApi, WebhookHandler
+from linebot.exceptions import InvalidSignatureError
+from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
 
-from linebot import (
-    LineBotApi, WebhookHandler
-)
-from linebot.exceptions import (
-    InvalidSignatureError
-)
-from linebot.models import (
-    MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
-)
+# Retrieve the PDF
+files = glob.glob('docs\*.pdf')
+files_content = [client.files.upload(file=pathlib.Path(file_path)) for file_path in files]
+
+# 設定 Google AI API 金鑰
+client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
+
+# 設定生成文字的參數 + 角色扮演
+generation_config = genai.types.GenerateContentConfig(max_output_tokens=3000, temperature=0.1, top_p=0.2,
+                    system_instruction="您是一位問答助手。請僅限使用以上提供的內容來回答問題。如果您不知道答案，請說您不知道，不要捏造答案。")
 
-# 設定 GeminiRAG 的HF網址
-client = Client(os.environ["GeminiRAGapi"], hf_token=os.environ['HF_TOKEN'])
 # 設定 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()
-# app.mount("/static", staticfiles.StaticFiles(directory="static"), name="static")
 
 # 設定 CORS，允許跨域請求
 app.add_middleware(
@@ -86,8 +87,10 @@ def handle_message(event):
         try: 
             # 取得使用者輸入的文字
             prompt = event.message.text
-            # 使用 GeminiRAGapi
-            completion = client.predict(question=prompt, api_name="/predict")
+            # 使用 client.models.generate_content
+            completion = client.models.generate_content(
+                                model="gemini-2.5-flash",
+                                contents=files_content.append(prompt))
             # 檢查生成結果是否為空
             if (completion != None):
                 # 取得生成結果