Create app.py

app.py

@@ -0,0 +1,116 @@
+import gradio as gr
+import pandas as pd
+import requests
+import webbrowser
+
+# Define the backend API URL (modify this to your backend's URL)
+BACKEND_API_URL = "https://do0rmamu-truck-loading-poc.hf.space/submit_data/"
+
+# Function to process the file and send data to the API
+def process_file(file, truck_type, auto_suggest):
+    try:
+        # Debugging: Print when file is being processed
+        print(f"Processing file: {file.name}")
+
+        # Read the uploaded file into a Pandas DataFrame
+        if file.name.endswith('.csv'):
+            df = pd.read_csv(file.name)
+        elif file.name.endswith('.xls') or file.name.endswith('.xlsx'):
+            df = pd.read_excel(file.name)
+        else:
+            return "Unsupported file format. Please upload a CSV or Excel file."
+        
+        print(f"File processed successfully, data:\n{df.head()}")  # Print first few rows of the data
+
+        # Prepare consignments data from the DataFrame
+        consignments_data = []
+        grouped = df.groupby('ConsignmentNo')
+        for consignment_no, group in grouped:
+            consignment_boxes = [
+                {
+                    'PieceLength': row['PieceLength'],
+                    'PieceBreadth': row['PieceBreadth'],
+                    'PieceHeight': row['PieceHeight'],
+                    'Priority': row.get('Priority', 0)
+                }
+                for _, row in group.iterrows()
+            ]
+            consignments_data.append({
+                'ConsignmentNo': str(consignment_no),  # Convert ConsignmentNo to string
+                'Priority': group['Priority'].max(),
+                'boxes': consignment_boxes
+            })
+
+        print(f"Consignment data prepared:\n{consignments_data}")
+
+        # Prepare the JSON payload to be sent
+        json_data = {
+            'truck_name': truck_type,
+            'autoSuggest': auto_suggest,
+            'consignments_data': consignments_data
+        }
+
+        print(f"Sending the following data to backend:\n{json_data}")
+
+        # Send the data as JSON to the backend
+        response = requests.post(BACKEND_API_URL, json=json_data, allow_redirects=False)
+
+        print(f"Received response from backend, status code: {response.status_code}")
+
+        # Handle redirect (302 Found)
+        if response.status_code == 302:
+            redirect_url = response.headers.get('Location')
+            if redirect_url:
+                print(f"Redirecting to: {redirect_url}")
+                webbrowser.open(redirect_url)
+                return f"Redirecting to visualization: {redirect_url}"
+        elif response.status_code != 200:
+            return f"Failed to submit data. Status code: {response.status_code}\nError message: {response.text}"
+        
+        return "Data submitted successfully."
+
+    except Exception as e:
+        print(f"Exception occurred: {str(e)}")
+        return f"An error occurred: {str(e)}"
+
+# Gradio interface
+def gradio_interface():
+    # Define available truck types
+    truck_types = [
+        "TATA ACE", "ASHOK LEYLAND DOST", "MAHINDRA BOLERO PICK UP", 
+        "ASHOK LEYLAND BADA DOST", "TATA 407", "EICHER 14 FEET",
+        "EICHER 17 FEET", "EICHER 19 FEET", "TATA 22 FEET",
+        "TATA TRUCK (6 TYRE)", "TAURUS 16 T (10 TYRE)", "TAURUS 21 T (12 TYRE)",
+        "TAURUS 25 T (14 TYRE)", "CONTAINER 20 FT", "CONTAINER 32 FT SXL",
+        "CONTAINER 32 FT MXL", "CONTAINER 32 FT SXL / MXL HQ",
+        "20 FEET OPEN ALL SIDE (ODC)", "28-32 FEET OPEN-TRAILOR JCB ODC",
+        "32 FEET OPEN-TRAILOR ODC", "40 FEET OPEN-TRAILOR ODC", "SCV", "LCV", 
+        "ICV", "MCV"
+    ]
+
+    with gr.Blocks() as demo:
+        # Title
+        gr.Markdown("## Truck Loading Data Submission")
+        
+        # File Upload Input
+        file_input = gr.File(label="Upload your consignments file (CSV or Excel)")
+
+        # Truck Type Dropdown
+        truck_type_input = gr.Dropdown(truck_types, label="Select Truck Type")
+
+        # Auto-Suggest Checkbox
+        auto_suggest_input = gr.Checkbox(label="Auto-Suggest Truck", value=False)
+
+        # Submit Button
+        submit_button = gr.Button("Submit Data")
+
+        # Output Textbox
+        output_text = gr.Markdown()
+
+        # Define interaction
+        submit_button.click(process_file, [file_input, truck_type_input, auto_suggest_input], output_text)
+
+    return demo
+
+# Run the Gradio interface
+gradio_interface().launch()