Update app.py

app.py

@@ -8,7 +8,7 @@ from xml.dom import minidom
 
 from modules.htlm_webpage import display_bpmn_xml
 from modules.utils import class_dict, rescale_boxes
-from modules.toXML import calculate_pool_bounds, add_diagram_elements, create_bpmn_object, create_flow_element, get_size_elements, definitions
+from modules.toXML import calculate_pool_bounds, add_diagram_elements, create_bpmn_object, create_flow_element, get_size_elements
 from streamlit_cropper import st_cropper
 from streamlit_image_select import image_select
 from streamlit_js_eval import streamlit_js_eval
@@ -17,6 +17,24 @@ from modules.streamlit_utils import get_memory_usage, clear_memory, get_image, l
 
 # Function to create a BPMN XML file from prediction results
 def create_XML(full_pred, text_mapping, size_scale, scale):
+    namespaces = {
+        'bpmn': 'http://www.omg.org/spec/BPMN/20100524/MODEL',
+        'bpmndi': 'http://www.omg.org/spec/BPMN/20100524/DI',
+        'di': 'http://www.omg.org/spec/DD/20100524/DI',
+        'dc': 'http://www.omg.org/spec/DD/20100524/DC',
+        'xsi': 'http://www.w3.org/2001/XMLSchema-instance'
+    }
+    
+    
+    definitions = ET.Element('bpmn:definitions', {
+        'xmlns:xsi': namespaces['xsi'],
+        'xmlns:bpmn': namespaces['bpmn'],
+        'xmlns:bpmndi': namespaces['bpmndi'],
+        'xmlns:di': namespaces['di'],
+        'xmlns:dc': namespaces['dc'],
+        'targetNamespace': "http://example.bpmn.com",
+        'id': "simpleExample"
+    })
 
     size_elements = get_size_elements(size_scale)