Update app.py

app.py

@@ -23,14 +23,12 @@ def generar_telefono_por_estado(estado):
         'Queretaro': '442',
         'Coahuila': '871'
     }
-    lada = ladas.get(estado, '55')  # Default a 55 si no coincide
+    lada = ladas.get(estado, '55')
     return f"{lada}{random.randint(1000000, 9999999)}"
 
 def generar_datos(cantidad, columnas, progreso):
     estados = [
-        'Ciudad de Mexico', 'Jalisco', 'Nuevo Leon', 'Veracruz', 'Yucatan', 
-        'Puebla', 'Chiapas', 'Tamaulipas', 'Queretaro', 'Coahuila'
-    ]
+        'Ciudad de Mexico', 'Jalisco', 'Nuevo Leon', 'Veracruz', 'Yucatan', 'Puebla', 'Chiapas', 'Tamaulipas', 'Queretaro', 'Coahuila']
     bancos = ['BBVA', 'Banorte', 'Santander', 'Citibanamex', 'HSBC']
     datos = []
 
@@ -65,24 +63,24 @@ def generar_datos(cantidad, columnas, progreso):
     progreso.progress(1.0)
     return pd.DataFrame(datos)
 
-def guardar_en_archivos_csv(df, registros_por_archivo, cantidad_archivos):
-    archivos_generados = []
-    for i in range(cantidad_archivos):
-        archivo_nombre = f"datos_parte_{i + 1}.csv"
-        df.to_csv(archivo_nombre, index=False)
-        archivos_generados.append(archivo_nombre)
-    return archivos_generados
-
 def comprimir_archivos(archivos, nombre_zip):
     with zipfile.ZipFile(nombre_zip, 'w') as zipf:
         for archivo in archivos:
             zipf.write(archivo, Path(archivo).name)
     return nombre_zip
 
+def guardar_en_archivo(df, formato):
+    if formato == "CSV":
+        archivo_nombre = "datos_generados.csv"
+        df.to_csv(archivo_nombre, index=False)
+    elif formato == "XLS":
+        archivo_nombre = "datos_generados.xlsx"
+        df.to_excel(archivo_nombre, index=False, engine='openpyxl')
+    return archivo_nombre
+
 def main():
-    st.title("Generador de CSV con Datos Aleatorios")
-    
-    # Checkbox para seleccionar columnas
+    st.title("Generador de Datos")
+
     columnas = []
     if st.checkbox("Nombre", value=True):
         columnas.append("Nombre")
@@ -101,37 +99,31 @@ def main():
     if st.checkbox("Banco", value=True):
         columnas.append("Banco")
     
-    # Slider para seleccionar la cantidad de archivos
-    cantidad_archivos = st.slider("Selecciona la cantidad de archivos a generar:", min_value=1, max_value=25, value=5)
-    cantidad_registros_por_archivo = 1_000_000
+    cantidad_registros = st.number_input("Cantidad de registros a generar:", min_value=1, max_value=1_000_000, value=10_000)
+    formato_descarga = st.radio("Selecciona el formato de descarga:", options=["CSV", "XLS"])
 
-    if 'archivos_generados' not in st.session_state:
-        st.session_state.archivos_generados = []
-
-    # Botón para generar los datos
-    if st.button("Generar CSV"):
+    if st.button("Generar Datos"):
         try:
             progreso = st.progress(0)
-            st.write(f"Generando {cantidad_archivos} archivos con {cantidad_registros_por_archivo} registros cada uno...")
-            df = generar_datos(cantidad_registros_por_archivo, columnas, progreso)
-            st.session_state.archivos_generados = guardar_en_archivos_csv(df, cantidad_registros_por_archivo, cantidad_archivos)
-            st.success("Archivos CSV generados correctamente.")
-        except Exception as e:
-            st.error(f"Error durante la generación: {e}")
-
-    # Botón para descargar los datos generados como ZIP
-    if st.session_state.archivos_generados:
-        if st.button("Descargar CSV"):
-            nombre_zip = "datos_generados.zip"
-            comprimir_archivos(st.session_state.archivos_generados, nombre_zip)
-            with open(nombre_zip, "rb") as file:
+            st.write(f"Generando {cantidad_registros} registros...")
+            df = generar_datos(cantidad_registros, columnas, progreso)
+            
+            # Mostrar previsualización
+            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(
-                    "Descargar ZIP con datos",
+                    f"Descargar {formato_descarga}",
                     data=file,
-                    file_name=nombre_zip,
-                    mime="application/zip"
+                    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}")
 
-# Asegurarse de que Streamlit ejecute el main al inicio
 if __name__ == "__main__":
     main()