Update app.py

app.py

@@ -104,17 +104,33 @@ def load_and_train():
         """
         texts = []
         # Determinar el tamaño del lote
-        num_examples = len(next(iter(examples.values())))  # Obtener el tamaño del lote
+        batch_size = len(next(iter(examples.values())))
         
-        for i in range(num_examples):
+        for i in range(batch_size):
             text = ''
-            if 'dialog' in examples and i < len(examples['dialog']) and examples['dialog'][i]:
-                text = examples['dialog'][i]
-            elif 'whole_func_string' in examples and i < len(examples['whole_func_string']) and examples['whole_func_string'][i]:
-                text = examples['whole_func_string'][i]
-            elif 'func_documentation_string' in examples and i < len(examples['func_documentation_string']) and examples['func_documentation_string'][i]:
-                text = examples['func_documentation_string'][i]
-            # Puedes añadir más campos si es necesario
+            if 'dialog' in examples and examples['dialog'][i]:
+                # Verificar si el campo es una lista y concatenar si es necesario
+                dialog = examples['dialog'][i]
+                if isinstance(dialog, list):
+                    dialog = ' '.join(dialog)
+                text = dialog
+            elif 'whole_func_string' in examples and examples['whole_func_string'][i]:
+                whole_func = examples['whole_func_string'][i]
+                if isinstance(whole_func, list):
+                    whole_func = ' '.join(whole_func)
+                text = whole_func
+            elif 'func_documentation_string' in examples and examples['func_documentation_string'][i]:
+                func_doc = examples['func_documentation_string'][i]
+                if isinstance(func_doc, list):
+                    func_doc = ' '.join(func_doc)
+                text = func_doc
+            else:
+                text = ''  # Asignar cadena vacía si no hay texto disponible
+            
+            # Asegurar que 'text' es una cadena de texto
+            if not isinstance(text, str):
+                text = str(text)
+            
             texts.append(text)
         
         examples['text'] = texts