Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import os
 from dotenv import load_dotenv
+from datetime import datetime, timedelta
 
 # Cargar variables de entorno desde el archivo .env
 load_dotenv()
@@ -11,7 +12,30 @@ CHUNK_SIZE = 1024  # Tamaño de los chunks para leer/escribir a la vez
 XI_API_KEY = os.getenv("XI_API_KEY")  # Tu clave API para autenticación
 VOICE_ID = os.getenv("VOICE_ID")  # ID del modelo de voz a utilizar
 
-def text_to_speech(text):
+# Diccionario para almacenar la información de los usuarios
+user_data = {}
+
+# Configuración de límites
+MESSAGE_LIMIT = 2
+TIME_LIMIT = timedelta(hours=2)
+
+def text_to_speech(text, user_id):
+    current_time = datetime.now()
+
+    if user_id not in user_data:
+        user_data[user_id] = {'messages': [], 'last_reset': current_time}
+
+    user_info = user_data[user_id]
+
+    # Resetear el contador si ha pasado el tiempo límite
+    if current_time - user_info['last_reset'] > TIME_LIMIT:
+        user_info['messages'] = []
+        user_info['last_reset'] = current_time
+
+    # Verificar si el usuario ha alcanzado el límite de mensajes
+    if len(user_info['messages']) >= MESSAGE_LIMIT:
+        return f"Error: Límite de mensajes alcanzado. Intenta nuevamente en 2 horas."
+
     # URL para la solicitud de la API de Text-to-Speech
     tts_url = f"https://api.elevenlabs.io/v1/text-to-speech/{VOICE_ID}/stream"
 
@@ -44,6 +68,8 @@ def text_to_speech(text):
             # Leer la respuesta en chunks y escribir en el archivo
             for chunk in response.iter_content(chunk_size=CHUNK_SIZE):
                 f.write(chunk)
+        # Registrar el nuevo mensaje
+        user_info['messages'].append({'message': text, 'time': current_time})
         return output_path
     else:
         return f"Error: {response.text}"
@@ -51,10 +77,10 @@ def text_to_speech(text):
 # Crear una interfaz de Gradio para la entrada de texto y la generación de audio
 iface = gr.Interface(
     fn=text_to_speech,
-    inputs="text",
+    inputs=["text", "text"],  # Añadir un segundo campo de texto para el user_id
     outputs="audio",
-    title="",
-    description="Texto a generar."
+    title="Generador de Texto a Voz",
+    description="Ingrese el texto para convertir a voz y su ID de usuario."
 )
 
 # Ejecutar la interfaz