Update app.py

app.py

@@ -7,7 +7,7 @@ import tempfile
 
 # Define transformation function
 
-def convert_ci(by_line_item_file, combined_file, password):
+def convert_ci(by_line_item_file, combined_file, password, max_rows):
     # Decrypt the protected "by line item" file
     decrypted = io.BytesIO()
     with open(by_line_item_file.name, "rb") as f:
@@ -143,8 +143,7 @@ def convert_ci(by_line_item_file, combined_file, password):
             merged_df[col] = ""
     merged_df = merged_df[column_order]
 
-    # Save merged output partitioned into 351-row chunks 
-    max_rows = 350
+    # Save merged output partitioned into 998-row chunks 
     outputs = []
 
     for i, start in enumerate(range(0, len(merged_df), max_rows)):
@@ -170,7 +169,8 @@ iface = gr.Interface(
     inputs=[
         gr.File(label="Upload 'By Line Item' File (.xlsx)", type="filepath"),
         gr.File(label="Upload 'Combined' File (.xlsx)", type="filepath"),
-        gr.Textbox(label="Password for 'By Line Item' File", type="password")
+        gr.Textbox(label="Password for 'By Line Item' File", type="password"),
+        gr.Number(label="Max Rows per Output File", value=350, precision=0)
     ],
     outputs=gr.File(label="Download Partitioned Files", file_types=[".xlsx"], file_count="multiple"),
     title="SHEIN CI to Magaya Format Conversion Tool",