Update transporter.py

transporter.py

@@ -22,6 +22,18 @@ def add_transporter(company_name, load_requirements, source, destination):
     transporters_df.to_excel(file_path, sheet_name='Transporters', index=False)
     return transporters_df
 
+def update_transporter_status(transporter_id, load_requirements_update):
+    global transporters_df
+    transporters_df.loc[transporters_df['Transporter ID'] == transporter_id, 'Load Requirements'] = load_requirements_update
+    transporters_df.to_excel(file_path, sheet_name='Transporters', index=False)
+    return transporters_df
+
+def check_booking_status(transporter_id):
+    global transporters_df
+    booking_status = transporters_df.loc[transporters_df['Transporter ID'] == transporter_id, 'Booking Status'].iloc[0]
+    current_location = transporters_df.loc[transporters_df['Transporter ID'] == transporter_id, 'Current Truck Location'].iloc[0]
+    return f"Booking Status: {booking_status}, Current Location: {current_location}"
+
 def transporter_ui():
     with gr.Blocks() as transporter_block:
         gr.Markdown("### Transporter Functions")
@@ -37,7 +49,7 @@ def transporter_ui():
         load_requirements_update = gr.Textbox(label="Load Requirements")
         update_btn = gr.Button("Update Load Requirements")
         updated_data = gr.DataFrame()
-        update_btn.click(lambda transporter_id, load_requirements_update: transporters_df.loc[transporters_df['Transporter ID'] == transporter_id, 'Load Requirements'].iloc[0] = load_requirements_update, inputs=[transporter_id, load_requirements_update], outputs=updated_data)
+        update_btn.click(update_transporter_status, inputs=[transporter_id, load_requirements_update], outputs=updated_data)
 
         gr.Markdown("### View Transporters Data")
         view_data_btn = gr.Button("View Transporters Data")
@@ -46,5 +58,5 @@ def transporter_ui():
         gr.Markdown("### Check Booking Status and Current Truck Location")
         check_status_btn = gr.Button("Check Booking Status")
         status_display = gr.Text()
-        check_status_btn.click(lambda transporter_id: f"Booking Status: {transporters_df.loc[transporters_df['Transporter ID'] == transporter_id, 'Booking Status'].iloc[0]}, Current Location: {transporters_df.loc[transporters_df['Transporter ID'] == transporter_id, 'Current Truck Location'].iloc[0]}", inputs=transporter_id, outputs=status_display)
+        check_status_btn.click(check_booking_status, inputs=transporter_id, outputs=status_display)
     return transporter_block