Update app.py

app.py

@@ -302,56 +302,35 @@ def drop_missing_rows(df, columns):
 #     workbook.remove(workbook["Sheet"])
 #     workbook.save('Log.xlsx')
 
-import openpyxl
-from openpyxl.styles import Font, PatternFill
-
 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]
             
-            color = None
-            if len(data) > 1:
-                color = data[1]
-                
-            if isinstance(lst[0], list):  # Check if the first element of lst is again a list
-                for sub_list in lst:  # Iterate over the sublists
-                    if isinstance(sub_list, list):
-                        sheet.append(sub_list)  # If sub_list is list, append as it is
-                    else:
-                        sheet.append([sub_list])  # If sub_list is a string, wrap it in a list
-                    
-                    # Fill color if provided
-                    if color:
-                        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
-            else:
-                sheet.append(lst)  # If lst[0] is not a list, write lst in a new row as before
-                
-                # Fill color if provided
+            # Extract the list values and color
+            lst, color = data[0], data[1] if len(data) > 1 else None
+            
+            # Write each item of the list in a new row and apply the color if provided
+            for item in lst:
+                sheet.append([item])
                 if color:
                     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
-
+                    cell = sheet.cell(row=sheet.max_row, column=1)
+                    cell.fill = fill
+    
     # Remove the default sheet created and save the workbook
     if workbook.active.title == "Sheet":
         workbook.remove(workbook.active)
+        
     workbook.save('Log.xlsx')
 
-
 warnings = []
 def is_found(c,text):
   global warnings