Update app.py

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 def display_csv(file, supplier):
     # Read the CSV file
     df = pd.read_csv(file.name)
-    
+
     # Filter DataFrame based on selected supplier
     if supplier == "Supplier 3":
         df = df[df['Vendor'].isin(["Seed", "EverColor"])]
@@ -16,25 +16,41 @@ def display_csv(file, supplier):
         df = df[df['Vendor'] == "Ann365"]
     elif supplier == "Supplier 1":
         df = df[~df['Vendor'].isin(["Seed", "EverColor", "Candy Magic", "OLENS", "Fairy", "Shobido", "Geo Medical", "Ann365"])]
-    # If no supplier is selected or blank option is chosen
     else:
         pass  # No filtering is applied, show all data
 
-    # Save the filtered DataFrame as a CSV file and return both the DataFrame and the CSV file path
+    # Display the full DataFrame
+    display_df = df.copy()
+
+    # Select specific columns to include in the output CSV
+    columns_to_include = [
+        'Order Number',
+        'Quantity',
+        'Product Title',
+        'Product Option Name',
+        'Product Option Value',
+        'Order Line item Properties 2 Name',
+        'Order Line item Properties 2 Value',
+        'Order Line item Properties 3 Name',
+        'Order Line item Properties 3 Value'
+    ]
+    download_df = df[columns_to_include]
+
+    # Save the filtered DataFrame with selected columns as a CSV file
     output_file = "filtered_output.csv"
-    df.to_csv(output_file, index=False)
-    return df, output_file
+    download_df.to_csv(output_file, index=False)
+    return display_df, output_file
 
 # Create a Gradio interface with dropdown for supplier selection
 interface = gr.Interface(
-    fn=display_csv, 
+    fn=display_csv,
     inputs=[
-        "file", 
+        "file",
         gr.Dropdown(choices=["", "Supplier 1", "Supplier 2", "Supplier 3", "Supplier 4", "Supplier 5"], label="Select Supplier", value="")
     ],
-    outputs=["dataframe", "file"], 
+    outputs=["dataframe", "file"],
     title="CSV Viewer",
-    description="Upload a CSV file, optionally select a supplier for specific Vendor filtering, view the contents, and download the file."
+    description="Upload a CSV file, optionally select a supplier for specific Vendor filtering, view the full contents on screen, and download a file with selected columns."
 )
 
 # Run the interface