Update main.py

main.py

@@ -1,7 +1,7 @@
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
-import google.generativeai as genai
-
+# import google.generativeai as genai
+from gradio_client import Client
 import json
 import os
 
@@ -15,10 +15,16 @@ from linebot.models import (
     MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
 )
 
-GOOGLE_API_KEY = os.environ["GOOGLE_API_KEY"]
-genai.configure(api_key = GOOGLE_API_KEY)
-# Initialize the Gemini model
-model = genai.GenerativeModel('gemini-pro')
+# GOOGLE_API_KEY = os.environ["GOOGLE_API_KEY"]
+# genai.configure(api_key = GOOGLE_API_KEY)
+# # Initialize the Gemini model
+# model = genai.GenerativeModel('gemini-pro')
+
+client = Client("https://alanchen1115-gemini-api.hf.space/--replicas/mgfss/")
+result = client.predict(
+		"Hello!!",	# str  in 'Prompt' Textbox component
+		api_name="/predict"
+)
 
 app = FastAPI()
 app.add_middleware(
@@ -75,11 +81,15 @@ def handle_message(event):
                             "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:
+            
+            # response = model.generate_content(messages)
+            # out = response.parts[0].text
+            # if not response.parts or not response.parts[0].text: 
+            # out = response.parts[0].text
+
+            out = client.predict(messages, api_name="/predict")
+            if not out:
                 out = "Gemini: [No response or the response does not contain text]"
         except:
             out = "Gemini error!"