Update app.py

app.py

@@ -38,10 +38,19 @@ def get_table_download_link(file_path):
     href = f'<a href="data:file/htm;base64,{b64}" target="_blank" download="{os.path.basename(file_path)}">{os.path.basename(file_path)}</a>'
     return href
 
+def add_parent_info(elem, parent=None):
+    for child in elem:
+        child.parent = parent
+        add_parent_info(child, parent=elem)
+
 def CompressXML(xml_text):
     root = ET.fromstring(xml_text)
-    ET.strip_elements(root, ET.Comment)
-    return ET.tostring(root, encoding='unicode')
+    add_parent_info(root)
+    for elem in list(root.iter()):
+        if isinstance(elem.tag, str) and 'Comment' in elem.tag:
+            elem.parent.remove(elem)
+    return ET.tostring(root, encoding='unicode', method="xml")
+
 
 def read_file_content(file):
     if file.type == "application/json":