Update main.py

main.py

@@ -1,20 +1,16 @@
+import json, os
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
-from gradio_client import Client
-import json
-import os
+import google.generativeai as genai
+
+from linebot import LineBotApi, WebhookHandler
+from linebot.exceptions import InvalidSignatureError
+from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
+
+genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+generation_config = genai.types.GenerationConfig(max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16)
+model = genai.GenerativeModel('gemini-1.5-flash')
 
-from linebot import (
-    LineBotApi, WebhookHandler
-)
-from linebot.exceptions import (
-    InvalidSignatureError
-)
-from linebot.models import (
-    MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
-)
- 
-client = Client(os.environ["GEMINI_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"
@@ -63,11 +59,14 @@ def handle_message(event):
        
     elif working_status:
         try: 
-            message = event.message.text
-            out = client.predict(message, api_name="/predict")
-
+            prompt = event.message.text
+            completion = model.generate_content(prompt, generation_config=generation_config)
+            if (completion.parts[0].text != None):
+                out = completion.parts[0].text
+            else:
+                out = "Gemini沒答案!請換個說法！"
         except:
-            out = "Gemini error!" 
+            out = "Gemini執行出錯!請換個說法！" 
   
         line_bot_api.reply_message(
             event.reply_token,