Update app.py

app.py

@@ -1,13 +1,20 @@
+import os
 import gradio as gr
-from huggingface_hub import InferenceClient
+from huggingface_hub import InferenceClient, login
 
-# Inicializar el cliente de Hugging Face
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+# Configuración inicial
+HF_TOKEN = os.getenv("HF_TOKEN", "tu_token_aqui")  # Usa variable de entorno o reemplaza
+MODEL_NAME = "HuggingFaceH4/zephyr-7b-beta"
+
+# Autenticación
+try:
+    login(token=HF_TOKEN)
+    client = InferenceClient(model=MODEL_NAME, token=HF_TOKEN, timeout=60)
+except Exception as auth_error:
+    print(f"Error de autenticación: {auth_error}")
 
 def format_prompt(message, history, system_message):
-    """
-    Formatea el prompt correctamente para el modelo Zephyr
-    """
+    """Formatea el prompt según las especificaciones de Zephyr"""
     prompt = f"<|system|>\n{system_message}</s>\n"
     for user_msg, assistant_msg in history:
         prompt += f"<|user|>\n{user_msg}</s>\n"
@@ -15,168 +22,181 @@ def format_prompt(message, history, system_message):
     prompt += f"<|user|>\n{message}</s>\n<|assistant|>\n"
     return prompt
 
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    """
-    Función principal que genera las respuestas del chatbot
-    """
+def respond(message, history, system_message, max_tokens, temperature, top_p):
+    """Genera respuestas del chatbot con manejo robusto de errores"""
     try:
-        # Formatear el prompt adecuadamente
+        # Validación de entrada
+        if not message or len(message.strip()) == 0:
+            yield "Por favor, ingresa un mensaje válido."
+            return
+
         prompt = format_prompt(message, history, system_message)
         
-        # Generar la respuesta
-        response = ""
-        for token in client.text_generation(
+        # Configuración de generación
+        generation_params = {
+            "max_new_tokens": min(max_tokens, 1024),  # Limita a 1024 tokens máximo
+            "temperature": max(0.1, min(temperature, 1.0)),
+            "top_p": max(0.1, min(top_p, 1.0)),
+            "do_sample": True,
+            "truncate": 2048
+        }
+
+        # Generación de respuesta
+        stream = client.text_generation(
             prompt,
-            max_new_tokens=max_tokens,
             stream=True,
-            temperature=temperature,
-            top_p=top_p,
-            truncate=1000,
-        ):
+            **generation_params
+        )
+
+        response = ""
+        for token in stream:
             response += token
             yield response
-            
+
     except Exception as e:
-        print(f"Error: {e}")
-        yield "Lo siento, ocurrió un error al procesar tu mensaje. Por favor, inténtalo de nuevo."
+        error_msg = f"Error en la generación: {str(e)}"
+        print(error_msg)
+        yield "🔴 Lo siento, tuve un problema al procesar tu mensaje. Inténtalo de nuevo más tarde."
 
-# CSS personalizado para la interfaz
+# CSS mejorado
 custom_css = """
-/* Estilos generales */
-body {
-    font-family: Arial, sans-serif;
-    background-color: #f5f5f5;
-    margin: 0;
-    padding: 20px;
+:root {
+    --primary: #6e48aa;
+    --secondary: #9d50bb;
+    --accent: #4776E6;
 }
 
-/* Contenedor principal */
 .gradio-container {
     max-width: 900px;
-    margin: 0 auto;
+    margin: 20px auto;
+    border-radius: 12px;
+    box-shadow: 0 6px 18px rgba(0,0,0,0.1);
     background: white;
-    border-radius: 10px;
-    box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);
-    padding: 20px;
+    padding: 25px;
 }
 
-/* Cabecera */
 .gradio-header {
     text-align: center;
-    margin-bottom: 20px;
+    margin-bottom: 25px;
+}
+
+h1 {
+    background: linear-gradient(45deg, var(--primary), var(--secondary));
+    -webkit-background-clip: text;
+    background-clip: text;
+    color: transparent;
+    font-size: 2.2rem;
+    margin-bottom: 10px;
+}
+
+.gradio-description {
+    color: #555;
+    font-size: 1rem;
 }
 
-/* Área de chat */
 .gradio-chatbot {
-    min-height: 400px;
+    min-height: 450px;
     border: 1px solid #e0e0e0;
-    border-radius: 8px;
+    border-radius: 10px;
     padding: 15px;
-    background: white;
-    margin-bottom: 15px;
+    background: #fafafa;
+    margin-bottom: 20px;
 }
 
-/* Input de texto */
 .gradio-textbox textarea {
     border-radius: 8px !important;
-    border: 1px solid #e0e0e0 !important;
-    padding: 12px !important;
-    font-size: 14px !important;
+    border: 1px solid #ddd !important;
+    padding: 12px 15px !important;
+    font-size: 15px !important;
+    min-height: 100px !important;
 }
 
-/* Botón de enviar */
 .gradio-button {
-    background: linear-gradient(90deg, #0f0653, #160979) !important;
+    background: linear-gradient(45deg, var(--primary), var(--secondary)) !important;
     color: white !important;
     border: none !important;
     border-radius: 8px !important;
-    padding: 12px 24px !important;
-    font-weight: bold !important;
+    padding: 12px 28px !important;
+    font-weight: 500 !important;
     transition: all 0.3s !important;
 }
 
 .gradio-button:hover {
     transform: translateY(-2px) !important;
-    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2) !important;
+    box-shadow: 0 4px 12px rgba(110, 72, 170, 0.3) !important;
 }
 
-/* Sliders y controles */
-.gradio-slider {
+.gradio-slider .wrap {
     margin: 15px 0 !important;
 }
 
-/* Mensajes del sistema */
-.gradio-markdown {
-    padding: 10px;
-    background: #f8f9fa;
-    border-radius: 8px;
-    margin-bottom: 15px;
+.dark .gradio-container {
+    background: #1a1a1a;
+}
+
+.dark .gradio-chatbot {
+    background: #252525;
+    border-color: #444;
 }
 """
 
-# Crear la interfaz de chat con Gradio
+# Configuración de la interfaz
 demo = gr.ChatInterface(
     fn=respond,
     additional_inputs=[
         gr.Textbox(
-            value="Eres un chatbot llamado ELISA. Eres amable, servicial y profesional. Desarrollado por Gerardo.",
-            label="Mensaje del Sistema",
-            lines=3
+            value="Eres ELISA, un asistente de IA útil, preciso y amable. Desarrollado por Gerardo.",
+            label="Configuración del Sistema",
+            lines=3,
+            max_lines=6
         ),
         gr.Slider(
-            minimum=32,
-            maximum=2048,
-            value=512,
+            minimum=64,
+            maximum=1024,
+            value=256,
             step=32,
-            label="Tokens máximos",
-            info="Controla la longitud máxima de la respuesta"
+            label="Longitud de Respuesta (tokens)",
+            info="Controla cuán extensa será la respuesta"
         ),
         gr.Slider(
             minimum=0.1,
-            maximum=1.5,
+            maximum=1.0,
             value=0.7,
-            step=0.1,
-            label="Temperatura",
-            info="Valores más altos = respuestas más creativas pero menos precisas"
+            step=0.05,
+            label="Creatividad (Temperatura)",
+            info="Valores más altos = respuestas más creativas"
         ),
         gr.Slider(
             minimum=0.1,
             maximum=1.0,
             value=0.9,
             step=0.05,
-            label="Top-p",
-            info="Controla la diversidad de las respuestas"
+            label="Enfoque (Top-p)",
+            info="Controla la diversidad de palabras"
         ),
     ],
     css=custom_css,
-    title="🤖💬 ELISA - Modelo de Inteligencia Artificial",
-    description="Chatbot desarrollado por Gerardo usando Zephyr-7b-beta de Hugging Face",
-    theme="soft",
+    title="🤖 ELISA - Asistente de IA",
+    description="Chatbot avanzado desarrollado por Gerardo usando Hugging Face",
     examples=[
-        ["Hola, ¿cómo estás?"],
-        ["Explícame qué es la inteligencia artificial"],
-        ["Dame ideas para un proyecto de IA"]
+        ["Hola, ¿qué puedes hacer?"],
+        ["Explícame el machine learning en términos simples"],
+        ["Recomiéndame libros sobre IA"]
     ],
-    retry_btn=None,
-    undo_btn=None,
-    clear_btn="Limpiar"
+    submit_btn="Enviar",
+    retry_btn="Reintentar",
+    undo_btn="Deshacer",
+    clear_btn="Limpiar",
+    theme="soft"
 )
 
-# Configuración adicional para el lanzamiento
+# Configuración de lanzamiento
 if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
         share=False,
+        debug=True,
         favicon_path=None,
-        auth=None,
-        auth_message=None,
-        prevent_thread_lock=False
+        auth=None
     )