Use by default name of the sheet and remove colors if colorized already

app.py

@@ -1,14 +1,23 @@
 import pandas as pd
 import openpyxl
 from openpyxl.styles import PatternFill
+
 import gradio as gr
 
 def load_excel(file_path):
-    """Loads the Excel file into a pandas DataFrame."""
-    df = pd.read_excel(file_path, sheet_name="Sheet1")
-    return df
+    """Loads the Excel file into a pandas DataFrame and selects the first sheet."""
+    wb = openpyxl.load_workbook(file_path)
+    sheet_name = wb.sheetnames[0]  # Access the first sheet
+    df = pd.read_excel(file_path, sheet_name=sheet_name)
+    return df, wb, sheet_name
+
+def clear_sheet_colors(ws):
+    """Removes all colors from the given worksheet."""
+    for row in ws.iter_rows():
+        for cell in row:
+            cell.fill = PatternFill(fill_type=None)
 
-def highlight_pairs(df, threshold=2.0):
+def highlight_pairs(df, wb, sheet_name, threshold=2.0):
     """Highlights pairs in columns D and E based on the threshold."""
     colors = ["FF0000", "00FF00", "0000FF", "FFFF00", "FF00FF", "00FFFF"]  # Hex colors for RGB
     color_index = 0
@@ -17,8 +26,9 @@ def highlight_pairs(df, threshold=2.0):
     df['Low Result'] = ''
     
     colored_pairs = set()
-    wb = openpyxl.load_workbook(file_path)
-    ws = wb['Sheet1']
+    ws = wb[sheet_name]
+    
+    clear_sheet_colors(ws)  # Clear existing colors before applying new ones
     
     # Function to apply color to cells
     def apply_color(cell, color_hex):
@@ -68,8 +78,8 @@ def highlight_pairs(df, threshold=2.0):
 
 def gradio_app(file_path):
     """Main function to handle Gradio inputs and outputs."""
-    df = load_excel(file_path)
-    highlight_pairs(df)
+    df, wb, sheet_name = load_excel(file_path)
+    highlight_pairs(df, wb, sheet_name)
     return 'output.xlsx'
 
 # Gradio Interface