Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 import base64
 import glob
 import json
-from xml.etree import ElementTree as ET
+from xml.etree import ElementTree
 from datetime import datetime
 from dotenv import load_dotenv
 from openai import ChatCompletion
@@ -38,20 +38,6 @@ 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)
-    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":
         content = json.load(file)
@@ -60,9 +46,9 @@ def read_file_content(file):
         content = BeautifulSoup(file, "html.parser")
         return content.text
     elif file.type == "application/xml" or file.type == "text/xml":
-        xml_text = file.getvalue().decode()
-        compressed_text = CompressXML(xml_text)
-        return compressed_text
+        tree = ElementTree.parse(file)
+        root = tree.getroot()
+        return ElementTree.tostring(root, encoding='unicode')
     elif file.type == "text/plain":
         return file.getvalue().decode()
     else:
@@ -81,7 +67,6 @@ def main():
 
     if uploaded_file is not None:
         file_content = read_file_content(uploaded_file)
-        st.markdown(f"**Content Added to Prompt:**\n{file_content}")
         prompts.append(file_content)
 
     if st.button('Chat'):
@@ -100,7 +85,7 @@ def main():
         st.sidebar.markdown(get_table_download_link(file), unsafe_allow_html=True)
         if st.sidebar.button(f"Delete {file}"):
             os.remove(file)
+            st.experimental_rerun()
 
 if __name__ == "__main__":
     main()
-