Update app.py

app.py

@@ -21,7 +21,7 @@ st.title("GPT Chat with Optional File Context - Talk to your data!")
 # Output options sidebar menu
 st.sidebar.title("Output Options")
 menu = ["txt", "htm", "md"]
-choice = st.sidebar.selectbox("Choose an output file type to save your prompt and results", menu)
+choice = st.sidebar.selectbox("Choose output file type to save results", menu)
 choicePrefix = "Output and download file set to "
 if choice == "txt":
      st.sidebar.write(choicePrefix + "Text file.")
@@ -83,12 +83,11 @@ def get_table_download_link(file_path):
     href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
     return href
 
-def CompressXML(xml_text):
+def CompressXML(xml_text, max_length):
     root = ET.fromstring(xml_text)
     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")
     return ET.tostring(root, encoding='unicode', method="xml")[:max_length]  
     
 def read_file_content(file):
@@ -102,7 +101,7 @@ def read_file_content(file):
         tree = ET.parse(file)
         root = tree.getroot()
         #return ET.tostring(root, encoding='unicode')
-        return CompressXML(ET.tostring(root, encoding='unicode'))
+        return CompressXML(ET.tostring(root, encoding='unicode',max_length))
     elif file.type == "text/markdown" or file.type == "text/md":
         md = mistune.create_markdown()
         content = md(file.read().decode())
@@ -136,13 +135,12 @@ def main():
         st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
     
     if len(file_content) > 0:
-        st.markdown(f"**Content Added to Prompt:**\n{file_content}")
+        st.markdown(f"**File Content Added:**\n{file_content}")
     
     htm_files = glob.glob("*.txt")
     for file in htm_files:
         st.sidebar.markdown(get_table_download_link(file), unsafe_allow_html=True)
-        if st.sidebar.button(f"🗑Delete {file}"):
-        #if st.sidebar.button("🗑 Delete"):
+        if st.sidebar.button(f"🗑 Delete {file}"):
             os.remove(file)
             st.experimental_rerun()