Update app.py

app.py

@@ -1,48 +1,97 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from firebase_admin import auth, firestore, initialize_app
 from huggingface_hub import login
+import stripe
 
-# Autenticación con el token de Hugging Face
-login("hf_...gxEF")  # Asegúrate de reemplazar con tu token completo
+# Configurar Firebase y Stripe
+initialize_app()  # Inicializar Firebase con credenciales
+db = firestore.client()  # Cliente de Firestore
+stripe.api_key = "TU_CLAVE_SECRETA_STRIPE"
 
-# Cargar el modelo de Hugging Face
-model_name = "meta-llama/Llama-2-7b-chat-hf"  # Cambia al modelo que prefieras
+# Autenticación de Hugging Face
+login("hf_...gxEF")
+
+# Configuración del modelo
+model_name = "meta-llama/Llama-2-7b-chat-hf"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
-
-# Configurar el pipeline de generación de texto
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-# Definir el límite de mensajes diarios
-DAILY_MESSAGE_LIMIT = 10
-user_message_count = {}
+# Límite de mensajes gratis
+FREE_MESSAGE_LIMIT = 15
+
+def sign_in_with_email(email, password):
+    user = auth.get_user_by_email(email)
+    if user:
+        # Iniciar sesión y devolver detalles del usuario
+        return {"user_id": user.uid, "message_count": user.custom_claims.get("message_count", 0)}
+    return None
+
+def sign_up_with_email(email, password, username):
+    user = auth.create_user(email=email, password=password, display_name=username)
+    db.collection("users").document(user.uid).set({"email": email, "username": username, "message_count": FREE_MESSAGE_LIMIT})
+    return {"user_id": user.uid, "message_count": FREE_MESSAGE_LIMIT}
+
+def check_subscription(user_id):
+    user_ref = db.collection("users").document(user_id)
+    user_data = user_ref.get().to_dict()
+    if user_data and user_data.get("subscribed", False):
+        return True
+    return False
 
-# Función para generar texto con un contador de mensajes
 def generate_text(prompt, history, user_id):
-    if user_id not in user_message_count:
-        user_message_count[user_id] = 0
+    user_ref = db.collection("users").document(user_id)
+    user_data = user_ref.get().to_dict()
 
-    if user_message_count[user_id] >= DAILY_MESSAGE_LIMIT:
-        return history + [[prompt, "Has alcanzado el límite de mensajes diarios. Considera suscribirte para obtener más acceso."]]
+    # 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})
 
-    # Generar la respuesta
     response = generator(prompt, max_length=100, temperature=0.7, top_k=50, top_p=0.9)[0]['generated_text']
-    user_message_count[user_id] += 1
-    history.append([prompt, response])  # Añadir el mensaje al historial de conversación
+    history.append([prompt, response])
     return history
 
-# Crear la interfaz de usuario estilo chat
+def create_subscription(user_id):
+    session = stripe.checkout.Session.create(
+        payment_method_types=['card'],
+        line_items=[{
+            'price': 'precio_ID',  # Reemplaza con el ID de tu precio en Stripe
+            'quantity': 1,
+        }],
+        mode='subscription',
+        success_url='https://tu-app/success',  # URL de éxito
+        cancel_url='https://tu-app/cancel',  # URL de cancelación
+    )
+    return session.url
+
+# Interfaz 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")
+    gr.Markdown("# Bienvenido a Amsterdam IA")
+    
+    # Registro e Inicio de Sesión
+    with gr.Tab("Iniciar Sesión"):
+        email_login = gr.Textbox(label="Correo")
+        password_login = gr.Textbox(label="Contraseña", type="password")
+        login_button = gr.Button("Iniciar Sesión")
+    
+    with gr.Tab("Crear Cuenta"):
+        email_signup = gr.Textbox(label="Correo")
+        password_signup = gr.Textbox(label="Contraseña", type="password")
+        username_signup = gr.Textbox(label="Nombre de Usuario")
+        signup_button = gr.Button("Crear Cuenta")
+        
+    # Chat
+    chatbot = gr.Chatbot(label="Chat con Amsterdam IA")
     prompt = gr.Textbox(label="Escribe tu mensaje aquí...")
-
     submit_button = gr.Button("Enviar")
-    clear_button = gr.Button("Limpiar Chat")
+    clear_button = gr.Button("Nuevo Chat")
 
-    # Actualizar la conversación en el chatbot
-    submit_button.click(fn=generate_text, inputs=[prompt, chatbot, user_id], outputs=chatbot)
-    clear_button.click(fn=lambda: None, inputs=None, outputs=chatbot, label="Limpiar")
+    login_button.click(sign_in_with_email, inputs=[email_login, password_login], outputs=[chatbot])
+    signup_button.click(sign_up_with_email, inputs=[email_signup, password_signup, username_signup], outputs=[chatbot])
+    submit_button.click(generate_text, inputs=[prompt, chatbot, email_login], outputs=chatbot)
+    clear_button.click(fn=lambda: None, inputs=None, outputs=chatbot)
 
 iface.launch()