Update app.py

app.py

@@ -4,26 +4,54 @@ from firebase_admin import auth, firestore, initialize_app, credentials
 from huggingface_hub import login
 import stripe
 
-# Cargar las credenciales de Firebase
-cred = credentials.Certificate("amsterdam-ai-firebase-adminsdk-m8kjd-95ae3fb6bb.json")  # Asegúrate de que el nombre coincide
-initialize_app(cred)
+# Inicializar las credenciales de Firebase
+cred = credentials.Certificate("amsterdam-ai-firebase-adminsdk-m8kjd-95ae3fb6bb.json")  # Ruta del archivo JSON
+initialize_app(cred)  # Inicializa la aplicación de Firebase con las credenciales
 
-# Configurar el cliente de Firestore
-db = firestore.client()
+# Configuración del cliente de Firestore
+db = firestore.client()  # Esto te permite acceder a la base de datos Firestore
 
-# Configurar Stripe con tu clave de API
+# Configuración de Stripe con tu clave secreta
 stripe.api_key = "TU_CLAVE_SECRETA_STRIPE"
 
 # Autenticación de Hugging Face
-login("hf_...gxEF")
+login("hf_...gxEF")  # Asegúrate de usar tu token de acceso
 
-# Configuración del modelo
-model_name = "meta-llama/Llama-2-7b-chat-hf"
+# Cargar el modelo de Hugging Face
+model_name = "meta-llama/Llama-2-7b-chat-hf"  # Puedes cambiarlo por cualquier modelo que prefieras
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# Límite de mensajes gratis
+# Límite de mensajes gratuitos
 FREE_MESSAGE_LIMIT = 15
 
-# Aquí sigue el resto de tu código, sin cambios
+# Función para generar texto con un contador de mensajes
+def generate_text(prompt, history, user_id):
+    user_ref = db.collection("users").document(user_id)
+    user_data = user_ref.get().to_dict()
+
+    # Verificar suscripción y límite de mensajes
+    if not user_data.get("subscribed", False):
+        if user_data["message_count"] <= 0:
+            return history + [[prompt, "Has alcanzado el límite de mensajes gratuitos. Suscríbete para continuar."]]
+        user_ref.update({"message_count": user_data["message_count"] - 1})
+
+    response = generator(prompt, max_length=100, temperature=0.7, top_k=50, top_p=0.9)[0]['generated_text']
+    history.append([prompt, response])
+    return history
+
+# Crear la interfaz de usuario en Gradio
+with gr.Blocks() as iface:
+    gr.Markdown("# Venice IA Chatbot")
+    user_id = gr.Textbox(label="ID de Usuario (email o identificador único)", placeholder="Introduce tu ID de usuario", visible=False)
+    chatbot = gr.Chatbot(label="Chat con Venice IA")
+    prompt = gr.Textbox(label="Escribe tu mensaje aquí...")
+    submit_button = gr.Button("Enviar")
+    clear_button = gr.Button("Nuevo Chat")
+
+    # Funciones para los botones
+    submit_button.click(fn=generate_text, inputs=[prompt, chatbot, user_id], outputs=chatbot)
+    clear_button.click(fn=lambda: None, inputs=None, outputs=chatbot)
+
+iface.launch()