update backend functions

app.py

@@ -1,18 +1,16 @@
 import gradio as gr
 import os
-import time
-
-from utils import make_invisible, make_visible
+from gcp import download_credentials
+from utils import make_invisible, make_visible, create_folders
 from backend_functions import get_answer, init_greeting, export_dataframe
 
 from dotenv import load_dotenv
 load_dotenv()
 
-
+download_credentials()
+create_folders()
 
 with gr.Blocks() as main_app:
-    
-    times_functions = gr.State([[], [], []])
 
     with gr.Tab('Chatbot'):
         user_id = gr.State('')  # id used to find the chat into the database

backend_functions.py

@@ -9,15 +9,15 @@ import uuid
 import re
 import pandas as pd
 import tensorflow as tf
-from utils import create_folders
 from google.cloud import storage
 from elevenlabs.client import ElevenLabs, AsyncElevenLabs
 from elevenlabs import play, save, Voice, stream
 from pymongo.mongo_client import MongoClient
-from gcp import download_credentials
 from dotenv import load_dotenv
 load_dotenv()
 
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 MODEL_OPENAI = os.getenv("MODEL_OPENAI")
 
@@ -37,6 +37,8 @@ IMG_XAVY = os.getenv("IMG_XAVY")
 CREDENTIALS_GCP = os.getenv("GOOGLE_APPLICATION_CREDENTIALS")
 NAME_BUCKET = os.getenv("NAME_BUCKET")
 
+# Gemini
+genai.configure(api_key=GEMINI_API_KEY)
 
 # Chat
 openai_client = OpenAI(api_key=OPENAI_API_KEY)
@@ -249,8 +251,6 @@ def _create_clean_message(text: str):
 def _create_audio(clean_text: str):
 
     if tf.test.is_gpu_available():
-        download_credentials()
-        create_folders()
         STORAGE_CLIENT = storage.Client.from_service_account_json(CREDENTIALS_GCP)
 
         unique_id = str(uuid.uuid4())
@@ -289,8 +289,6 @@ def _create_audio(clean_text: str):
         except Exception as e:
             print(e)
     else:
-        download_credentials()
-        create_folders()
         STORAGE_CLIENT = storage.Client.from_service_account_json(CREDENTIALS_GCP)
 
         unique_id = str(uuid.uuid4())

gcp.py

@@ -4,6 +4,7 @@ from datetime import timedelta
 from google.cloud import storage
 from huggingface_hub import hf_hub_download
 from google.cloud.storage import transfer_manager
+
 from dotenv import load_dotenv
 load_dotenv()
 
@@ -14,15 +15,17 @@ def download_credentials() -> None:
     """
     Downloads the GCP credentials from Hugging Face Hub
     """
-    assets_dir = 'assets'
+    assets_dir = './assets'
     credentials_file = os.path.join(assets_dir, "credentials.json")
+    print(credentials_file)
 
     # Verificar si la carpeta 'assets' existe y crearla si no
     os.makedirs(assets_dir, exist_ok=True)
 
     # Verificar si el archivo 'credentials.json' ya existe en la carpeta 'assets'
     if not os.path.isfile(credentials_file):
+        print("No habian credenciales")
         hf_hub_download(
             repo_id=REPO_HUGGING, repo_type='dataset', filename="credentials.json",
             token=HUB_TOKEN, local_dir=assets_dir
-        )
+        )

utils.py

@@ -19,15 +19,12 @@ def create_folders():
     """
     Creates 'audios' and 'videos' directories if they do not exist
     """
-    audio_dir = 'audios'
-    video_dir = 'videos'
-    times_dir = 'csv_times'
+    audio_dir = './audios'
+    video_dir = './videos'
+    times_dir = './csv_times'
 
-    if not os.path.exists(audio_dir):
-        os.makedirs(audio_dir)
+    os.makedirs(audio_dir, exist_ok=True)
 
-    if not os.path.exists(video_dir):
-        os.makedirs(video_dir)
+    os.makedirs(video_dir, exist_ok=True)
 
-    if not os.path.exists(times_dir):
-        os.makedirs(times_dir)
+    os.makedirs(times_dir, exist_ok=True)