Update app.py

app.py

@@ -1,9 +1,10 @@
+###ORDER###
 import gradio as gr
 import pandas as pd
 from datetime import datetime
 import pytz  # This library is used for timezone conversions
 
-def display_csv(file, supplier, order_number):
+def order_csv(file, supplier, order_number):
     df = pd.read_csv(file.name)
 
     # Define supplier-specific filters
@@ -11,8 +12,8 @@ def display_csv(file, supplier, order_number):
         "Supplier 1": lambda df: df[~df['Vendor'].isin(["Seed", "EverColor", "Candy Magic", "OLENS", "Fairy", "Shobido", "Geo Medical", "Ann365", "Lenstown", "LensVery"])],
         "Supplier 2": lambda df: df[df['Vendor'].isin(["Candy Magic", "OLENS", "Fairy", "Shobido"])],
         "Supplier 3": lambda df: df[df['Vendor'].isin(["Seed", "EverColor"])],
-        "Supplier 4": lambda df: df[df['Vendor'] == "Geo Medical", "Lenstown"],
-        "Supplier 5": lambda df: df[df['Vendor'] == "Ann365", "LensVery"]
+        "Supplier 4": lambda df: df[df['Vendor'].isin(["Geo Medical", "Lenstown"])],
+        "Supplier 5": lambda df: df[df['Vendor'].isin(["Ann365", "LensVery"])]
     }
 
     # Apply supplier filter if selected
@@ -60,7 +61,7 @@ with gr.Blocks() as demo:
         output_file = gr.File(label="Download Filtered CSV")
 
     # Bind the function to inputs and outputs using the button
-    load_button.click(fn=display_csv, inputs=[file_input, supplier_input, order_number_input], outputs=[output_df, output_file])
+    load_button.click(fn=order_csv, inputs=[file_input, supplier_input, order_number_input], outputs=[output_df, output_file])
 
 # Run the interface
 demo.launch()