Update app.py

app.py

@@ -4,28 +4,25 @@ import random
 import datetime
 import os
 
-csv_filename = "realistic_shipping_data.csv"
+csv_filename = "shipping_data.csv"
 
-# Sample data for randomization
+# Sample data
 shipper_names = ["Global Express", "Fast Ship Logistics", "Secure Courier"]
 cities_states = [("New York", "NY"), ("Los Angeles", "CA"), ("Chicago", "IL"), ("Houston", "TX"), ("Miami", "FL")]
 shipping_methods = ["FedEx", "UPS", "DHL", "USPS"]
 weight_range = (0.5, 50)  # Min and max weight in kg
 
-# Function to generate a random phone number
 def random_phone():
     return f"+1-{random.randint(200, 999)}-{random.randint(100, 999)}-{random.randint(1000, 9999)}"
 
-# Function to generate a random tracking number
 def generate_tracking(carrier):
     prefix = {"FedEx": "FDX", "UPS": "UPS", "DHL": "DHL", "USPS": "USPS"}.get(carrier, "TRK")
     return f"{prefix}-{random.randint(100000, 999999)}"
 
-# Function to generate the shipping CSV file
 def generate_csv():
     try:
         shipping_data = []
-        for order_id in range(1001, 1011):  # Generating 10 orders
+        for order_id in range(1001, 1011):
             order_date = datetime.datetime.now() - datetime.timedelta(days=random.randint(1, 30))
             delivery_date = order_date + datetime.timedelta(days=random.randint(2, 10))
 
@@ -37,30 +34,28 @@ def generate_csv():
             weight = round(random.uniform(*weight_range), 2)
             tracking_number = generate_tracking(shipping_method)
 
-            shipping_data.append({
-                "Order ID": order_id,
-                "Order Date": order_date.strftime("%Y-%m-%d %H:%M:%S"),
-                "Shipper Name": shipper,
-                "Shipper Address": f"{random.randint(100, 999)} {shipper_city} St, {shipper_city}, {shipper_state}",
-                "Shipper Phone": random_phone(),
-                "Receiver Name": f"Customer {order_id}",
-                "Receiver Address": f"{random.randint(100, 999)} {receiver_city} Ave, {receiver_city}, {receiver_state}",
-                "Receiver Phone": random_phone(),
-                "Package Weight (kg)": weight,
-                "Shipping Method": shipping_method,
-                "Tracking Number": tracking_number,
-                "Estimated Delivery Date": delivery_date.strftime("%Y-%m-%d")
-            })
+            shipping_data.append([
+                order_id,
+                order_date.strftime("%Y-%m-%d %H:%M:%S"),
+                shipper,
+                f"{random.randint(100, 999)} {shipper_city} St, {shipper_city}, {shipper_state}",
+                random_phone(),
+                f"Customer {order_id}",
+                f"{random.randint(100, 999)} {receiver_city} Ave, {receiver_city}, {receiver_state}",
+                random_phone(),
+                weight,
+                shipping_method,
+                tracking_number,
+                delivery_date.strftime("%Y-%m-%d"),
+            ])
 
-        # Define CSV column headers
-        csv_columns = ["Order ID", "Order Date", "Shipper Name", "Shipper Address", "Shipper Phone", "Receiver Name",
-                       "Receiver Address", "Receiver Phone", "Package Weight (kg)", "Shipping Method", "Tracking Number",
-                       "Estimated Delivery Date"]
+        csv_columns = ["Order ID", "Order Date", "Shipper Name", "Shipper Address", "Shipper Phone", 
+                       "Receiver Name", "Receiver Address", "Receiver Phone", "Package Weight (kg)", 
+                       "Shipping Method", "Tracking Number", "Estimated Delivery Date"]
 
-        # Write data to CSV file
         with open(csv_filename, mode="w", newline="") as file:
-            writer = csv.DictWriter(file, fieldnames=csv_columns)
-            writer.writeheader()
+            writer = csv.writer(file)
+            writer.writerow(csv_columns)
             writer.writerows(shipping_data)
 
         print("✅ CSV file generated successfully!")
@@ -70,7 +65,6 @@ def generate_csv():
         print(f"❌ Error generating CSV: {e}")
         return None
 
-# Gradio function to return the CSV file
 def download_csv():
     file_path = generate_csv()
     if file_path and os.path.exists(file_path):
@@ -80,11 +74,10 @@ def download_csv():
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# 📦 Shipping CSV Generator")
-    gr.Markdown("Click the button below to generate a shipping CSV file.")
+    gr.Markdown("Click the button below to generate and download a CSV file.")
     btn = gr.Button("Generate CSV")
-    output = gr.File(label="Download your CSV")
+    output = gr.File(label="Download your CSV", type="file")  # Ensure correct file type
     btn.click(fn=download_csv, outputs=output)
 
-# Launch Gradio app
 if __name__ == "__main__":
     demo.launch()