import pandas as pd
import gradio as gr

file_path = 'data.xlsx'
admins_df = pd.read_excel(file_path, sheet_name='Admins')
truck_owners_df = pd.read_excel(file_path, sheet_name='Truck Owners')

def verify_kyc(admin_id, truck_owner_id):
    global admins_df, truck_owners_df
    if admins_df.loc[admins_df['Admin ID'] == admin_id, 'KYC Verification Status'].iloc[0] == 'Verified':
        truck_owners_df.loc[truck_owners_df['Truck Owner ID'] == truck_owner_id, 'KYC Status'] = 'Verified'
        truck_owners_df.to_excel(file_path, sheet_name='Truck Owners', index=False)
    return truck_owners_df

def admin_ui():
    with gr.Blocks() as admin_block:
        gr.Markdown("### Admin Functions")
        admin_id = gr.Number(label="Admin ID")
        truck_owner_id = gr.Number(label="Truck Owner ID")
        verify_btn = gr.Button("Verify KYC")
        verified_data = gr.DataFrame()
        verify_btn.click(verify_kyc, inputs=[admin_id, truck_owner_id], outputs=verified_data)

        gr.Markdown("### View Admins Data")
        view_data_btn = gr.Button("View Admins Data")
        view_data_btn.click(lambda: admins_df, outputs=verified_data)
    return admin_block