Update app.py

app.py

@@ -15,6 +15,8 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import datetime
 import base64
+from io import BytesIO
+from pyxlsb import open_workbook as open_xlsb
 
 # BODY
 st.image('gdmk.png', width=150)
@@ -90,8 +92,7 @@ else:
     resultados_df.columns = ['timestamp', 'nombre']
     resultados_df['arch_evento'] = caras_evento_code_df.rotulo
     resultados_df['imagenes'] = caras_evento_code_df.image
-    resultados_df.to_csv('resultados/resultados_'+timestamp+'.csv')
-    
+
     st.dataframe(resultados_df.drop(['imagenes'], axis=1))
     asistentes = resultados_df.nombre
     
@@ -104,6 +105,23 @@ else:
     asist_df = pd.DataFrame(asistencia)
     st.dataframe(asist_df)
     
+    def convierte_excel(df):
+        output = BytesIO()
+        writer = pd.ExcelWriter(output, engine='xlsxwriter')
+        df.to_excel(writer, index=False, sheet_name='data_extraida')
+        workbook = writer.book
+        worksheet = writer.sheets['data_extraida']
+        format1 = workbook.add_format({'num_format': '0.00'}) 
+        worksheet.set_column('A:A', None, format1)  
+        writer.save()
+        processed_data = output.getvalue()
+        writer.close()
+        return processed_data
+    
+    df_xlsx = convierte_excel(asist_df)
+    st.download_button(label='📩 Descargar XLSX', data=df_xlsx,
+                                file_name = 'resultados_'+timestamp+'.xlsx')
+                                
     n_fig = 10
     fig, ax = plt.subplots(1, n_fig, figsize=(21, 2))
     for i in range(n_fig):
@@ -117,14 +135,14 @@ else:
             ax[i].set_title(str(resultados_df.nombre.iloc[i]))
             ax[i].axis('off')
             
-    timestamp = str(datetime.datetime.now()).replace(':','-')
-    plt.savefig('resultados_'+timestamp+'.jpg')
-    st.image(Image.open('resultados_'+timestamp+'.jpg'), width=800)
-    #st.pyplot(fig)
+    #timestamp = str(datetime.datetime.now()).replace(':','-')
+    #plt.savefig('resultados_'+timestamp+'.jpg')
+    #st.image(Image.open('resultados_'+timestamp+'.jpg'), width=800)
+    st.pyplot(fig)
     
-    st.write('Se ha guardado los archivos en el folder resultados.')
-    resultados_df.to_csv('resultados/verificacion.csv')
-    asist_df.to_csv('resultados/asistencia.csv')
+    #st.write('Se ha guardado los archivos en el folder resultados.')
+    #resultados_df.to_csv('resultados/verificacion.csv')
+    #asist_df.to_csv('resultados/asistencia.csv')
     
     c1, c2, c3 = st.columns(3)
     with c1: