Update app.py

app.py

@@ -22,72 +22,85 @@ def generar_nss():
     return f"{random.randint(100, 999)}-{random.randint(10, 99)}-{random.randint(1000, 9999)}"
 
 def generar_correo_aleatorio():
-    diccionario = PyDictionary()
-    palabra = random.choice(list(diccionario.meaning("word").keys()))
+    try:
+        diccionario = PyDictionary()
+        palabra = random.choice(list(diccionario.meaning("word").keys()))
+    except Exception:
+        palabra = faker.word()
     dominio = random.choice(["gmail.com", "hotmail.com"])
     return limpiar_correo(f"{palabra}{random.randint(1, 999)}@{dominio}")
 
-def limpiar_correo(correo):
-    return unidecode(correo.replace(" ", "").lower())
-
 def generar_datos(cantidad, columnas, progreso):
-    estados = ['Ciudad de Mexico', 'Jalisco', 'Nuevo Leon', 'Veracruz', 'Yucatan', 'Puebla', 'Chiapas', 'Tamaulipas', 'Queretaro', 'Coahuila']
-    bancos, datos = ['BBVA', 'Banorte', 'Santander', 'Citibanamex', 'HSBC', 'AMEX', 'Inbursa', 'Bancoppel'], []
-    for i in range(cantidad):
-        estado = random.choice(estados)
-        nombre, apellido = faker.first_name(), faker.last_name()
-        dia, mes, anio = faker.date_of_birth().day, faker.date_of_birth().month, faker.date_of_birth().year
-        fila = {'Nombre': f"{nombre} {apellido}" if 'Nombre' in columnas else None, 
-                'RFC': generar_rfc(nombre, apellido, anio, mes, dia) if 'RFC' in columnas else None,
-                'Telefono': generar_telefono_por_estado(estado) if 'Telefono' in columnas else None,
-                'Estado': estado if 'Estado' in columnas else None,
-                'Direccion': faker.address() if 'Direccion' in columnas else None,
-                'Correo Electronico': generar_correo_aleatorio() if 'Correo Electronico' in columnas else None,
-                'Tarjeta': ''.join([str(random.randint(0, 9)) for _ in range(16)]) if 'Tarjeta' in columnas else None,
-                'Banco': random.choice(bancos) if 'Banco' in columnas else None,
-                'Score de Credito': random.randint(300, 850) if 'Score de Credito' in columnas else None,
-                'NSS': generar_nss() if 'NSS' in columnas else None}
-        datos.append({k: v for k, v in fila.items() if v is not None})
-        if i % 10000 == 0:
-            progreso.progress(i / cantidad)
-    progreso.progress(1.0)
-    return pd.DataFrame(datos)
+    try:
+        estados = ['Ciudad de Mexico', 'Jalisco', 'Nuevo Leon', 'Veracruz', 'Yucatan', 
+                   'Puebla', 'Chiapas', 'Tamaulipas', 'Queretaro', 'Coahuila']
+        bancos, datos = ['BBVA', 'Banorte', 'Santander', 'Citibanamex', 'HSBC', 
+                         'AMEX', 'Inbursa', 'Bancoppel'], []
+        for i in range(cantidad):
+            estado = random.choice(estados)
+            nombre, apellido = faker.first_name(), faker.last_name()
+            dia, mes, anio = faker.date_of_birth().day, faker.date_of_birth().month, faker.date_of_birth().year
+            fila = {'Nombre': f"{nombre} {apellido}" if 'Nombre' in columnas else None, 
+                    'RFC': generar_rfc(nombre, apellido, anio, mes, dia) if 'RFC' in columnas else None,
+                    'Telefono': generar_telefono_por_estado(estado) if 'Telefono' in columnas else None,
+                    'Estado': estado if 'Estado' in columnas else None,
+                    'Direccion': faker.address() if 'Direccion' in columnas else None,
+                    'Correo Electronico': generar_correo_aleatorio() if 'Correo Electronico' in columnas else None,
+                    'Tarjeta': ''.join([str(random.randint(0, 9)) for _ in range(16)]) if 'Tarjeta' in columnas else None,
+                    'Banco': random.choice(bancos) if 'Banco' in columnas else None,
+                    'Score de Credito': random.randint(300, 850) if 'Score de Credito' in columnas else None,
+                    'NSS': generar_nss() if 'NSS' in columnas else None}
+            datos.append({k: v for k, v in fila.items() if v is not None})
+            if i % max(1, cantidad // 100) == 0:  # Actualiza progreso proporcionalmente
+                progreso.progress(i / cantidad)
+        progreso.progress(1.0)
+        return pd.DataFrame(datos)
+    except Exception as e:
+        raise RuntimeError(f"Error al generar datos: {e}")
 
 def guardar_varios_archivos(dfs, formato):
     archivos = []
-    for i, df in enumerate(dfs):
-        timestamp = datetime.now().strftime("_%H%M%S")
-        archivo_nombre = f"leads_buro_{i+1}{timestamp}." + ("csv" if formato == "CSV" else "xlsx")
-        if formato == "CSV":
-            df.to_csv(archivo_nombre, index=False)
-        else:
-            df.to_excel(archivo_nombre, index=False, engine='openpyxl')
-        archivos.append(archivo_nombre)
-    return archivos
+    try:
+        for i, df in enumerate(dfs):
+            timestamp = datetime.now().strftime("_%H%M%S")
+            archivo_nombre = f"leads_buro_{i+1}{timestamp}." + ("csv" if formato == "CSV" else "xlsx")
+            if formato == "CSV":
+                df.to_csv(archivo_nombre, index=False)
+            else:
+                df.to_excel(archivo_nombre, index=False, engine='openpyxl')
+            archivos.append(archivo_nombre)
+        return archivos
+    except Exception as e:
+        raise RuntimeError(f"Error al guardar archivos: {e}")
 
 def crear_zip(archivos):
-    zip_nombre = "archivos_generados.zip"
-    with zipfile.ZipFile(zip_nombre, 'w') as zipf:
-        for archivo in archivos:
-            zipf.write(archivo, Path(archivo).name)
-    return zip_nombre
+    try:
+        zip_nombre = "archivos_generados.zip"
+        with zipfile.ZipFile(zip_nombre, 'w') as zipf:
+            for archivo in archivos:
+                zipf.write(archivo, Path(archivo).name)
+        for archivo in archivos:  # Limpia archivos temporales
+            Path(archivo).unlink(missing_ok=True)
+        return zip_nombre
+    except Exception as e:
+        raise RuntimeError(f"Error al crear archivo ZIP: {e}")
 
 def main():
     st.set_page_config(layout="wide")
     st.sidebar.title("Configuración")
-    opciones_columnas = ["Nombre", "Telefono", "RFC", "Estado", "Direccion", "Correo Electronico", "Tarjeta", "Banco", "Score de Credito", "NSS"]
+    opciones_columnas = ["Nombre", "Telefono", "RFC", "Estado", "Direccion", 
+                         "Correo Electronico", "Tarjeta", "Banco", "Score de Credito", "NSS"]
     columnas = [col for col in opciones_columnas if st.sidebar.checkbox(col, value=False)]
     num_archivos = st.sidebar.slider("Número de archivos (1M registros c/u):", min_value=1, max_value=25, value=1)
     formato_descarga = st.sidebar.radio("Selecciona el formato de descarga:", ["CSV", "XLS"])
     if st.sidebar.button("Generar Datos"):
         try:
+            if not columnas:
+                st.error("Debe seleccionar al menos una columna.")
+                return
             progreso = st.progress(0)
             st.write(f"Generando {num_archivos} archivo(s) de 1 millón de registros...")
-            dfs = []
-            for i in range(num_archivos):
-                st.write(f"Generando archivo {i+1} de {num_archivos}...")
-                df = generar_datos(1_000_000, columnas, progreso)
-                dfs.append(df)
+            dfs = [generar_datos(1_000_000, columnas, progreso) for _ in range(num_archivos)]
             st.subheader("Previsualización de datos (primeros 100 registros del primer archivo):")
             st.dataframe(dfs[0].head(100))
             archivos_generados = guardar_varios_archivos(dfs, formato_descarga)
@@ -98,7 +111,10 @@ def main():
                     st.download_button("Descargar todos en ZIP", data=file, file_name=zip_nombre, mime="application/zip")
             else:
                 with open(archivos_generados[0], "rb") as file:
-                    st.download_button(f"Descargar {formato_descarga}", data=file, file_name=archivos_generados[0], mime="text/csv" if formato_descarga == "CSV" else "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
+                    st.download_button(f"Descargar {formato_descarga}", data=file, 
+                                       file_name=archivos_generados[0], 
+                                       mime="text/csv" if formato_descarga == "CSV" else 
+                                            "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
             st.success("Archivos generados y listos para descargar.")
         except Exception as e:
             st.error(f"Error durante la generación: {e}")