Update app.py

app.py

@@ -7,7 +7,6 @@ import gradio as gr
 import re
 import zipfile
 import datetime
-
 import openpyxl
 from openpyxl.styles import Font, PatternFill
 
@@ -236,6 +235,32 @@ def drop_missing_rows(df, columns):
     df = df.dropna(subset=columns, how='any')
     return df,removed_rows
 
+def write_log(sheet_data_dict):
+    workbook = openpyxl.Workbook()
+    for sheet_name, data_dict in sheet_data_dict.items():
+
+        sheet = workbook.create_sheet(title=sheet_name)
+        for title, data in data_dict.items():
+            # Write the title
+            sheet.append([title])
+            title_cell = sheet.cell(row=sheet.max_row, column=1)
+            title_cell.font = Font(size=14, bold=True)  # Set the font size and bold
+
+            # Write the list values
+            lst = data[0]
+            sheet.append(lst)
+
+            # Check if color is provided
+            if len(data) > 1:
+                color = data[1]
+                fill = PatternFill(start_color=color, end_color=color, fill_type="solid")
+                row_to_color = sheet[sheet.max_row]
+                for cell in row_to_color:
+                    cell.fill = fill
+
+    workbook.remove(workbook["Sheet"])
+    workbook.save('Log.xlsx')
+
 
 warnings = []
 def is_found(c,text):
@@ -385,7 +410,9 @@ def process(files,button):
         return None, msg
 
     names = unzip_files(files.name)
-
+    
+    sheet_data = dict()
+    
     for name in names:
         #name = os.path.basename(name)
         if valid(name):
@@ -400,10 +427,10 @@ def process(files,button):
                 df = pd.read_excel(name)
                 
             old_cols = df.columns
-            print()
+            old_olds = list(old_cols)
             print("Before columns")
-            print(list(old_cols))
-            print()
+            print(old_olds)
+
             if "summ" in name:
                 df,columns = filter_premiums(df)
                 if columns == None:
@@ -427,7 +454,7 @@ def process(files,button):
                     finalnames.append(columns[ind]+" ("+altnames[ind]+")")
             columns = [x for x in columns if x is not None]
 
-            print()
+
             print("After columns")
             print(columns)
             
@@ -438,6 +465,17 @@ def process(files,button):
             
             #display(df)
             name =  os.path.basename(name)
+            #print(columns)
+            #print(warnings)
+            sheetwarning = ['None']
+            if len(warnings) > 0:
+                sheetwarnings = [warnings,'#FFA500']
+            sheet_data[name] = {
+                "Before columns": [old_olds],
+                "After columns": [columns, '00FF00'],
+                #"Warnings": [sheetwarnings]
+            }
+            
             c_name = name.split('.')[0]+'_cleaned.csv'
             df.to_csv(c_name,index=False)
 
@@ -455,10 +493,13 @@ def process(files,button):
             status.append(op_outcome(name,' Failed (Only .csv, .xlsx, .zip are allowed)'+fail))
             
     if len(cleaned_names) > 0 :
+        write_log(sheet_data)
+        cleaned_names.append('Log.xlsx')
         final_file = zip_files(cleaned_names)
     else:
         final_file = None
     msg = '\n'.join(f"{index + 1}.{value}" for index, value in enumerate(status))
+    
     return gr.File.update(value=final_file,visible=True),msg
     #return(str(files)+'fole')
 
@@ -510,4 +551,5 @@ with gr.Blocks(css=elems) as demo:
 
     bt.click(fn = process, inputs=[inp,bt], outputs=[out,log])
 
-demo.launch(debug = False)
+
+demo.launch(debug = True)