Update main.py

main.py

@@ -25,8 +25,8 @@ def make_body(file):
 
 def index_to_ref_id(reference_text, reference_ids):
     reference_text = reference_text.strip('[]')
-    dash_separator = "–"
-    comma_separator = ','
+    dash_separator = "–" # long dash here
+    comma_separator = ", " # workaound
     id_max = len(reference_ids)
     ids = []
     refs = []
@@ -36,7 +36,7 @@ def index_to_ref_id(reference_text, reference_ids):
         for i in range(s,e+1):
             ids.append(i)
     elif reference_text.find(comma_separator) != -1:
-        ids = [int(x) for x in reference_text.split(comma_separator)]
+        ids = [int(x) for x in reference_text.split(comma_separator)]    
     else:
         ids.append(int(reference_text))
 
@@ -65,16 +65,16 @@ def xref_generator(reference_text, reference_ids):
     return generated_xref
 
 def make_refs(article_body, reference_ids):
-    citations_set = set()
+    citations_list = []
 
-    citations_raw = re.findall(r'(\[\d+–\d+\]|\[\d+(, \d+)*\])', contents)
+    citations_raw = re.findall(r"\[\d+(?:–\d+)?(?:,\s?\d+)*\]", contents)
 
     for cit in citations_raw:
-        citations_set.add(cit[0])
+        citations_list.append(cit) # bugfix
 
     x = contents
 
-    for cit in citations_set:
+    for cit in citations_list:
         x = x.replace(str(cit), str(xref_generator(cit, reference_ids)))
 
     return x