Update app.py

app.py

@@ -1,9 +1,12 @@
 import streamlit as st
 import pandas as pd
 import random
+import zipfile
 from faker import Faker
 from unidecode import unidecode
 from pathlib import Path
+from datetime import datetime
+from PyDictionary import PyDictionary
 
 faker = Faker('es_MX')
 
@@ -19,16 +22,17 @@ def generar_nss():
     return f"{random.randint(100, 999)}-{random.randint(10, 99)}-{random.randint(1000, 9999)}"
 
 def generar_correo_aleatorio():
-    palabras_aleatorias = ["alpha", "beta", "gamma", "delta", "omega", "neutron", "photon", "comet", "nova", "quasar"]
+    diccionario = PyDictionary()
+    palabra = random.choice(list(diccionario.meaning("word").keys()))
     dominio = random.choice(["gmail.com", "hotmail.com"])
-    return limpiar_correo(f"{random.choice(palabras_aleatorias)}{random.randint(1, 999)}@{dominio}")
+    return limpiar_correo(f"{palabra}{random.randint(1, 999)}@{dominio}")
 
 def limpiar_correo(correo):
-    return unidecode(correo.replace(" ", ""))
+    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'], []
+    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()
@@ -44,33 +48,60 @@ def generar_datos(cantidad, columnas, progreso):
                 '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)
+        if i % 10000 == 0:
+            progreso.progress(i / cantidad)
     progreso.progress(1.0)
     return pd.DataFrame(datos)
 
-def guardar_en_archivo(df, formato):
-    archivo_nombre = "leads_buro." + ("csv" if formato == "CSV" else "xlsx")
-    (df.to_csv(archivo_nombre, index=False) if formato == "CSV" else df.to_excel(archivo_nombre, index=False, engine='openpyxl'))
-    return archivo_nombre
+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
+
+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
 
 def main():
     st.set_page_config(layout="wide")
     st.sidebar.title("Configuración")
-    columnas = [col for col, chk in zip(["Nombre", "Telefono", "RFC", "Estado", "Direccion", "Correo Electronico", "Tarjeta", "Banco", "Score de Credito", "NSS"], 
-                [st.sidebar.checkbox(c, value=c in ["Nombre", "Telefono", "RFC", "Estado", "Score de Credito", "NSS"]) for c in ["Nombre", "Telefono", "RFC", "Estado", "Direccion", "Correo Electronico", "Tarjeta", "Banco", "Score de Credito", "NSS"]]) if chk]
-    cantidad_registros = st.sidebar.number_input("Cantidad de registros a generar:", min_value=1, max_value=1_000_000, value=500_000)
+    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:
             progreso = st.progress(0)
-            st.write(f"Generando {cantidad_registros} registros...")
-            df = generar_datos(cantidad_registros, columnas, progreso)
-            st.subheader("Previsualización de datos (primeros 100 registros):")
-            st.dataframe(df.head(100))
-            archivo_generado = guardar_en_archivo(df, formato_descarga)
-            with open(archivo_generado, "rb") as file:
-                st.download_button(f"Descargar {formato_descarga}", data=file, file_name=archivo_generado, mime="text/csv" if formato_descarga == "CSV" else "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
-            st.success(f"Archivo {formato_descarga} generado y listo para descargar.")
-        except Exception as e: st.error(f"Error durante la generación: {e}")
+            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)
+            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)
+            
+            if len(archivos_generados) > 1:
+                zip_nombre = crear_zip(archivos_generados)
+                with open(zip_nombre, "rb") as file:
+                    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.success("Archivos generados y listos para descargar.")
+        except Exception as e:
+            st.error(f"Error durante la generación: {e}")
 
-if __name__ == "__main__": main()
+if __name__ == "__main__":
+    main()