From 066478a8f4767f4ae25e41bea23cc4231598a860 Mon Sep 17 00:00:00 2001
From: lakhoune <a.b.lakhoune@gmail.com>
Date: Tue, 27 Feb 2024 12:30:09 +0100
Subject: [PATCH] fix extraction of users

---
 discovery/main.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/discovery/main.py b/discovery/main.py
index aac833a..7a5fe2c 100644
--- a/discovery/main.py
+++ b/discovery/main.py
@@ -1,4 +1,5 @@
 import pm4py
+import json
     
 def discover_petri_net(event_log,algorithm="inductive"):
     """
@@ -84,6 +85,9 @@ def bot_statistics(event_log):
     stats['numberOfStates'] = event_log['concept:name'].nunique()
     if 'user' in event_log.columns:
         stats['numberOfUsers'] = event_log['user'].nunique()
+    else:
+        users = event_log['REMAKRS'].apply(lambda x: json.loads(x)['user'])
+        stats['numberOfUsers'] = users.nunique()
     stats['averageConversationLength'] = event_log.groupby(
         'case:concept:name').size().mean()
     stats['averageConversationDuration'] = event_log.groupby(