Update app.py

app.py

@@ -29,12 +29,12 @@ bart_model = BartForConditionalGeneration.from_pretrained(bart_model_name)
 bart_model.to(device)
 bart_model.eval()
 
-def autocomplete_text(input_text, max_tokens=20):
+def autocomplete_text(input_text, max_tokens=5):
     """
-    Autocompleta el texto/código de entrada usando code-autocomplete-gpt2-base
+    Autocompleta texto común en español basado en el inicio de palabras
     
     Args:
-        input_text (str): Texto/código inicial a completar
+        input_text (str): Texto inicial a completar
         max_tokens (int): Número máximo de tokens a generar
     
     Returns:
@@ -43,6 +43,113 @@ def autocomplete_text(input_text, max_tokens=20):
     if not input_text.strip():
         return "Por favor, ingresa algún texto para completar."
     
+    # Diccionario de autocompletado común en español
+    common_completions = {
+        'h': ['hola', 'hacer', 'historia', 'hermano', 'hospital', 'habitación'],
+        'ho': ['hola', 'hombre', 'hospital', 'hoy', 'hogar'],
+        'hol': ['hola'],
+        'c': ['casa', 'comer', 'coche', 'computadora', 'ciudad', 'cambiar'],
+        'ca': ['casa', 'carro', 'cambiar', 'caminar', 'calor'],
+        'cas': ['casa', 'casi'],
+        'com': ['comer', 'computadora', 'comprar', 'completar'],
+        'comp': ['computadora', 'comprar', 'completar', 'comprimir'],
+        'compu': ['computadora'],
+        'g': ['gracias', 'grande', 'gente', 'guitarra', 'ganar'],
+        'gr': ['gracias', 'grande', 'grupo'],
+        'gra': ['gracias', 'grande'],
+        'graci': ['gracias'],
+        'b': ['bien', 'bueno', 'beber', 'buscar', 'bailar'],
+        'bi': ['bien', 'bicicleta'],
+        'bu': ['bueno', 'buscar'],
+        'bue': ['bueno', 'buenos'],
+        'buen': ['bueno', 'buenos'],
+        'm': ['muy', 'madre', 'mesa', 'música', 'mañana'],
+        'mu': ['muy', 'música', 'mujer', 'mundo'],
+        'muy': ['muy'],
+        't': ['trabajo', 'tiempo', 'teléfono', 'también', 'tener'],
+        'tr': ['trabajo', 'tres', 'transporte'],
+        'tra': ['trabajo', 'transporte'],
+        'trab': ['trabajo'],
+        'trabaj': ['trabajo'],
+        'p': ['por', 'para', 'persona', 'programa', 'problema'],
+        'po': ['por', 'poder'],
+        'pr': ['programa', 'problema', 'primero'],
+        'pro': ['programa', 'problema', 'proceso'],
+        'prog': ['programa'],
+        'progr': ['programa'],
+        'progra': ['programa'],
+        'program': ['programa'],
+        'e': ['escribir', 'estudiar', 'escuela', 'entender', 'español'],
+        'es': ['escribir', 'estudiar', 'escuela', 'español', 'estar'],
+        'esc': ['escribir', 'escuela'],
+        'escr': ['escribir'],
+        'escri': ['escribir'],
+        'escrib': ['escribir'],
+        'escribi': ['escribir'],
+        'a': ['aplicación', 'agua', 'amigo', 'aprender', 'ayuda'],
+        'ap': ['aplicación', 'aprender', 'ayuda'],
+        'apl': ['aplicación'],
+        'apli': ['aplicación'],
+        'aplic': ['aplicación'],
+        'aplica': ['aplicación'],
+        'aplicaci': ['aplicación'],
+        'aplicacio': ['aplicación'],
+        'aplicacion': ['aplicación'],
+        'v': ['ver', 'venir', 'vida', 'viaje', 'ventana'],
+        've': ['ver', 'venir', 'ventana'],
+        'ven': ['venir', 'ventana'],
+        'veni': ['venir'],
+        'd': ['de', 'día', 'decir', 'dinero', 'diferente'],
+        'di': ['día', 'dinero', 'diferente'],
+        'dia': ['día'],
+        'n': ['no', 'necesitar', 'noche', 'número', 'nuevo'],
+        'nu': ['número', 'nuevo'],
+        'nue': ['nuevo'],
+        'nuev': ['nuevo'],
+        's': ['sí', 'ser', 'sistema', 'salir', 'semana'],
+        'si': ['sí', 'sistema', 'siempre'],
+        'sis': ['sistema'],
+        'sist': ['sistema'],
+        'siste': ['sistema'],
+        'sistem': ['sistema'],
+        'l': ['la', 'lo', 'lugar', 'llamar', 'leer'],
+        'lu': ['lugar'],
+        'lug': ['lugar'],
+        'luga': ['lugar'],
+        'r': ['rápido', 'recordar', 'respuesta', 'resultado', 'realizar'],
+        'ra': ['rápido'],
+        'rap': ['rápido'],
+        'rapi': ['rápido'],
+        'rapid': ['rápido'],
+        'rapido': ['rápido']
+    }
+    
+    input_lower = input_text.lower().strip()
+    
+    # Buscar coincidencias exactas primero
+    if input_lower in common_completions:
+        suggestions = common_completions[input_lower]
+        # Filtrar para mostrar solo las que son diferentes al input
+        filtered_suggestions = [word for word in suggestions if word != input_lower]
+        if filtered_suggestions:
+            # Retornar la primera sugerencia sin el input original
+            best_match = filtered_suggestions[0]
+            return best_match[len(input_lower):]
+    
+    # Si no hay coincidencia exacta, buscar palabras que empiecen con el input
+    matching_words = []
+    for word_list in common_completions.values():
+        for word in word_list:
+            if word.startswith(input_lower) and word != input_lower:
+                matching_words.append(word)
+    
+    if matching_words:
+        # Ordenar por longitud (más cortas primero) y tomar la primera
+        matching_words.sort(key=len)
+        best_match = matching_words[0]
+        return best_match[len(input_lower):]
+    
+    # Si no encuentra coincidencias en el diccionario, usar el modelo GPT-2
     try:
         # Tokenizar el texto de entrada
         inputs = tokenizer.encode(input_text, return_tensors="pt", padding=True)
@@ -68,12 +175,12 @@ def autocomplete_text(input_text, max_tokens=20):
         new_text = generated_text[len(input_text):].strip()
         
         if not new_text:
-            return "No se pudo generar texto adicional."
+            return "No se encontraron sugerencias."
         
         return new_text
         
     except Exception as e:
-        return f"Error al generar texto: {str(e)}"
+        return f"Error al generar sugerencias: {str(e)}"
 
 def simplify_text(input_text, max_length=150):
     """
@@ -132,29 +239,29 @@ def create_autocomplete_interface():
     Crea la interfaz con autocompletar y simplificación dentro de gr.Blocks()
     """
     
-    with gr.Blocks(title="Asistente de Texto y Código") as demo:
+    with gr.Blocks(title="Asistente de Texto") as demo:
         
-        gr.Markdown("# 🤖 Asistente de Texto y Código")
-        gr.Markdown("Herramientas para autocompletar código y simplificar textos complejos.")
+        gr.Markdown("# 🤖 Asistente de Texto")
+        gr.Markdown("Herramientas para autocompletar palabras y simplificar textos complejos.")
         
         with gr.Tab("Autocompletar"):
             with gr.Row():
                 with gr.Column():
                     input_textbox = gr.Textbox(
-                        label="Código a completar",
-                        placeholder="def fibonacci(n):",
-                        lines=5,
-                        max_lines=10
+                        label="Texto a completar",
+                        placeholder="Escribe 'h' para ver 'hola'...",
+                        lines=3,
+                        max_lines=5
                     )
                     
-                    generate_btn = gr.Button("Completar Código", variant="primary")
+                    generate_btn = gr.Button("Completar Texto", variant="primary")
                     
                 with gr.Column():
                     output_textbox = gr.Textbox(
-                        label="Código generado",
-                        placeholder="Aquí aparecerá la continuación del código...",
-                        lines=5,
-                        max_lines=10,
+                        label="Sugerencia de completado",
+                        placeholder="Aquí aparecerá la sugerencia...",
+                        lines=3,
+                        max_lines=5,
                         interactive=False
                     )
             
@@ -211,16 +318,22 @@ def create_autocomplete_interface():
         # Pestaña adicional con ejemplos
         with gr.Tab("Ejemplos"):
             gr.Markdown("""
-            ### Ejemplos de Autocompletado de Código:
+            ### Ejemplos de Autocompletado de Palabras:
+            
+            **Entrada:** "h"
+            **Salida:** "ola" (completa "hola")
+            
+            **Entrada:** "gra"
+            **Salida:** "cias" (completa "gracias")
             
-            **Entrada:** "def fibonacci(n):"
-            **Salida:** "\\n    if n <= 1:\\n        return n\\n    return fibonacci(n-1) + fibonacci(n-2)"
+            **Entrada:** "compu"
+            **Salida:** "tadora" (completa "computadora")
             
-            **Entrada:** "for i in range("
-            **Salida:** "10):\\n    print(i)"
+            **Entrada:** "prog"
+            **Salida:** "rama" (completa "programa")
             
-            **Entrada:** "import pandas as pd\\ndf = pd.read_csv("
-            **Salida:** "'data.csv')\\nprint(df.head())"
+            **Entrada:** "escrib"
+            **Salida:** "ir" (completa "escribir")
             
             ---