Update main.py

main.py

@@ -1,6 +1,5 @@
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
-# import google.generativeai as genai
 from gradio_client import Client
 import json
 import os
@@ -15,11 +14,6 @@ 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')
-
 client = Client("https://alanchen1115-gemini-api.hf.space/--replicas/vboag/", hf_token=os.environ["HF_TOKEN"])
 
 app = FastAPI()
@@ -71,16 +65,6 @@ def handle_message(event):
        
     elif working_status:
         try: 
-            # messages = []
-            # message = event.message.text
-            # messages.append({
-            #                 "role": "user",
-            #                 "parts": [message],
-            #                 })
-            # # Check if response is valid
-            # response = model.generate_content(messages)
-            # if not response.parts or not response.parts[0].text: 
-            #     out = response.parts[0].text
             message = event.message.text
             out = client.predict(message, api_name="/predict")