Update app.py

app.py

@@ -1,18 +1,16 @@
-from fastapi import FastAPI
-from fastapi.responses import FileResponse
+import gradio as gr
 import csv
 import random
 import datetime
-import uvicorn
+import os
 
-app = FastAPI()
+csv_filename = "realistic_shipping_data.csv"
 
 # Sample data for randomization
 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
-csv_filename = "realistic_shipping_data.csv"
 
 # Function to generate a random phone number
 def random_phone():
@@ -25,55 +23,68 @@ def generate_tracking(carrier):
 
 # Function to generate the shipping CSV file
 def generate_csv():
-    shipping_data = []
-    for order_id in range(1001, 1011):  # Generating 10 orders
-        order_date = datetime.datetime.now() - datetime.timedelta(days=random.randint(1, 30))
-        delivery_date = order_date + datetime.timedelta(days=random.randint(2, 10))
+    try:
+        shipping_data = []
+        for order_id in range(1001, 1011):  # Generating 10 orders
+            order_date = datetime.datetime.now() - datetime.timedelta(days=random.randint(1, 30))
+            delivery_date = order_date + datetime.timedelta(days=random.randint(2, 10))
+
+            shipper = random.choice(shipper_names)
+            shipper_city, shipper_state = random.choice(cities_states)
+            receiver_city, receiver_state = random.choice(cities_states)
+
+            shipping_method = random.choice(shipping_methods)
+            weight = round(random.uniform(*weight_range), 2)
+            tracking_number = generate_tracking(shipping_method)
 
-        shipper = random.choice(shipper_names)
-        shipper_city, shipper_state = random.choice(cities_states)
-        receiver_city, receiver_state = random.choice(cities_states)
+            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_method = random.choice(shipping_methods)
-        weight = round(random.uniform(*weight_range), 2)
-        tracking_number = generate_tracking(shipping_method)
+        # 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"]
 
-        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")
-        })
+        # Write data to CSV file
+        with open(csv_filename, mode="w", newline="") as file:
+            writer = csv.DictWriter(file, fieldnames=csv_columns)
+            writer.writeheader()
+            writer.writerows(shipping_data)
 
-    # 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"]
+        print("✅ CSV file generated successfully!")
+        return csv_filename
 
-    # Write data to CSV file
-    with open(csv_filename, mode="w", newline="") as file:
-        writer = csv.DictWriter(file, fieldnames=csv_columns)
-        writer.writeheader()
-        writer.writerows(shipping_data)
+    except Exception as e:
+        print(f"❌ Error generating CSV: {e}")
+        return None
 
-# API Route to Generate and Download CSV
-@app.get("/generate-csv")
+# Gradio function to return the CSV file
 def download_csv():
-    generate_csv()
-    return FileResponse(csv_filename, media_type="text/csv", filename=csv_filename)
+    file_path = generate_csv()
+    if file_path and os.path.exists(file_path):
+        return file_path
+    return "Error: CSV file not found."
 
-@app.get("/")
-def root():
-    return {"message": "Welcome to the Shipping CSV Generator API! Use /generate-csv to download the file."}
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("# 📦 Shipping CSV Generator")
+    gr.Markdown("Click the button below to generate a shipping CSV file.")
+    btn = gr.Button("Generate CSV")
+    output = gr.File(label="Download your CSV")
+    btn.click(fn=download_csv, outputs=output)
 
-# Ensure FastAPI starts correctly on Hugging Face
+# Launch Gradio app
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    demo.launch()