Update main.py

main.py

@@ -1,6 +1,6 @@
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
-from gradio_client import Client
+import google.generativeai as genai
 
 import json
 import os
@@ -15,7 +15,11 @@ from linebot.models import (
     MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
 )
 
-client = Client("alanchen1115/Gemini")
+GOOGLE_API_KEY = "os.environ["GOOGLE_API_KEY"]
+genai.configure(api_key = GOOGLE_API_KEY)
+# Initialize the Gemini model
+model = genai.GenerativeModel('gemini-pro')
+
 app = FastAPI()
 app.add_middleware(
     CORSMiddleware,
@@ -59,11 +63,21 @@ def handle_message(event):
             event.reply_token,
             TextSendMessage(text="Bye!"))
         return
-
+        
+    messages = []
     if working_status:
-        out = client.predict(event.message.text, [[event.message.text, ]], api_name='/user')
-        if not out:
-            out = "不懂您的意思!"
+        message = event.message.text
+        messages.append({
+                        "role": "user",
+                        "parts": [message],
+                        })
+
+        response = model.generate_content(messages)
+        out = response.parts[0].text
+        # Check if response is valid
+        if not response.parts or not response.parts[0].text:
+            out = "Gemini: [No response or the response does not contain text]"
+
         line_bot_api.reply_message(
             event.reply_token,
             TextSendMessage(text=out))