Update app.py

app.py

@@ -5,15 +5,15 @@ import os
 import tempfile
 from pathlib import Path
 
-from gemini_client import Chatbot, Model
+from gemini_client import AsyncChatbot, Model, load_cookies
 
 app = FastAPI(title="Gemini Chat API", description="API para interagir com Google Gemini")
 
 # Inicializar chatbot globalmente
 chatbot = None
 
-def init_chatbot():
-    """Inicializa o chatbot com os cookies"""
+async def init_chatbot():
+    """Inicializa o chatbot com os cookies de forma assíncrona"""
     global chatbot
     cookie_path = os.getenv("COOKIE_PATH", "cookies.json")
     
@@ -21,10 +21,16 @@ def init_chatbot():
         raise FileNotFoundError(f"Arquivo de cookies não encontrado: {cookie_path}")
     
     try:
-        # Usar modelo PRO por padrão
-        chatbot = Chatbot(
-            cookie_path=cookie_path,
-            model=Model.G_2_5_PRO
+        # Carregar cookies
+        secure_1psid, secure_1psidts, additional_cookies = load_cookies(cookie_path)
+        
+        # Criar AsyncChatbot diretamente
+        chatbot = await AsyncChatbot.create(
+            secure_1psid=secure_1psid,
+            secure_1psidts=secure_1psidts,
+            model=Model.G_2_5_PRO,
+            additional_cookies=additional_cookies,
+            cookie_path=cookie_path
         )
         print(f"Chatbot inicializado com sucesso usando {cookie_path}")
     except Exception as e:
@@ -34,7 +40,7 @@ def init_chatbot():
 # Inicializar na startup
 @app.on_event("startup")
 async def startup_event():
-    init_chatbot()
+    await init_chatbot()
 
 @app.get("/")
 def root():
@@ -78,8 +84,8 @@ async def get_response(
                     }
                 )
         
-        # Enviar mensagem
-        response = chatbot.ask(message)
+        # Enviar mensagem (assíncrono)
+        response = await chatbot.ask(message)
         
         # Retornar resposta formatada
         return {
@@ -163,7 +169,7 @@ async def post_response(
             temp_file.close()
             media_path = temp_file.name
             
-            response = chatbot.ask(message, image=media_path)
+            response = await chatbot.ask(message, image=media_path)
         elif video:
             # Salvar vídeo temporariamente
             temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=Path(video.filename).suffix if video.filename else ".mp4")
@@ -172,7 +178,7 @@ async def post_response(
             temp_file.close()
             media_path = temp_file.name
             
-            response = chatbot.ask(message, video=media_path)
+            response = await chatbot.ask(message, video=media_path)
         elif audio:
             # Salvar áudio temporariamente
             temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=Path(audio.filename).suffix if audio.filename else ".mp3")
@@ -181,10 +187,10 @@ async def post_response(
             temp_file.close()
             media_path = temp_file.name
             
-            response = chatbot.ask(message, audio=media_path)
+            response = await chatbot.ask(message, audio=media_path)
         else:
             # Apenas texto
-            response = chatbot.ask(message)
+            response = await chatbot.ask(message)
         
         # Retornar resposta formatada
         return {