Mejora visualización

app.py

@@ -128,6 +128,32 @@ class NgramModel:
 def create_interface():
     model = NgramModel()
     
+    def train_model(text_input, file_input):
+        """Entrena el modelo con texto o archivo."""
+        if file_input is not None:
+            training_text = file_input
+        elif text_input.strip():
+            training_text = text_input
+        else:
+            return "Por favor, proporcione texto de entrenamiento."
+        
+        model.train(training_text)
+        return "Modelo entrenado exitosamente."
+
+    def predict(input_text):
+        """Realiza la predicción y actualiza el texto de entrada."""
+        if not model.is_trained:
+            return input_text, "", "El modelo no ha sido entrenado aún."
+        
+        next_word, explanation = model.predict_next_word(input_text)
+        new_text = input_text.strip() + " " + next_word if input_text.strip() else next_word
+        
+        return new_text, next_word, explanation
+
+    # Crear interfaz con Gradio
+def create_interface():
+    model = NgramModel()
+    
     def train_model(text_input, file_input):
         """Entrena el modelo con texto o archivo."""
         if file_input is not None:
@@ -161,6 +187,17 @@ def create_interface():
             max-width: 900px;
             margin: auto;
         }
+        .centered-title {
+            text-align: center !important;
+        }
+        .centered-title h1 {
+            text-align: center !important;
+            margin-bottom: 1.5rem;
+        }
+        .centered-title h2 {
+            text-align: center !important;
+            margin-bottom: 1rem;
+        }
     """) as interface:
         with gr.Column(elem_classes="container"):
             # Logo centrado
@@ -172,12 +209,12 @@ def create_interface():
                 </div>
             """)
             
-            gr.Markdown("# ¡Construye to propio miniGPT!")
-            gr.Markdown("## En la pestaña Entrenamiento puedes copiar o subir un texto base y entrenar tu miniGPT")
-            gr.Markdown("## En la pestaña Predicción puedes comenzar a escribir un texto y usar tu miniGPT para continuarlo automáticamente")
+            with gr.Column(elem_classes="centered-title"):
+                gr.Markdown("# ¡Construye tu propio miniGPT!")
+                gr.Markdown("### En la pestaña Entrenamiento puedes copiar o subir un texto base y entrenar tu miniGPT")
+                gr.Markdown("### En la pestaña Predicción puedes comenzar a escribir un texto y usar tu miniGPT para continuarlo automáticamente")
             
             with gr.Tab("Entrenamiento"):
-                gr.Markdown("### Entrena el modelo con texto")
                 text_input = gr.Textbox(
                     label="Texto de entrenamiento",
                     lines=5,
@@ -203,7 +240,6 @@ def create_interface():
                 )
             
             with gr.Tab("Predicción"):
-                gr.Markdown("### Predice la siguiente palabra")
                 input_text = gr.Textbox(
                     label="Introduce texto",
                     placeholder="Escribe aquí el texto para predecir la siguiente palabra..."