import pandas as pd
import pm4py

# csv_filename='Insurance_claims_event_log.csv'
csv_filename='Business Travel Permits.csv'
# csv_filename='Parent-Child Union Output.csv'
# csv_filename='PM4PY_Dummy_Event_Log_100_Cases_pm4py.csv'
# csv_filename='PM4Py_Dummy_Event_Log-congested-business-process.csv'


if __name__ == "__main__":
    dataframe = pd.read_csv(csv_filename, sep=',')
    dataframe = pm4py.format_dataframe(dataframe, case_id='case_id', activity_key='concept:name', timestamp_key='timestamp')
    # dataframe = pm4py.format_dataframe(dataframe, case_id='case_id', activity_key='activity_name', timestamp_key='timestamp')
    event_log = pm4py.convert_to_event_log(dataframe)

    dfg, start_activities, end_activities = pm4py.discover_dfg(event_log)
    pm4py.view_dfg(dfg, start_activities, end_activities)