Update app.py

app.py

@@ -46,7 +46,52 @@ def chat_with_model(prompt, document_section):
     response = openai.ChatCompletion.create(model=model, messages=conversation)
     return response['choices'][0]['message']['content']
 
-#... Rest of your code...
+
+def get_table_download_link(file_path):
+    with open(file_path, 'r') as file:
+        data = file.read()
+    b64 = base64.b64encode(data.encode()).decode()  
+    file_name = os.path.basename(file_path)
+    ext = os.path.splitext(file_name)[1]  # get the file extension
+    if ext == '.txt':
+        mime_type = 'text/plain'
+    elif ext == '.htm':
+        mime_type = 'text/html'
+    elif ext == '.md':
+        mime_type = 'text/markdown'
+    else:
+        mime_type = 'application/octet-stream'  # general binary data type
+    href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
+    return href
+
+
+def CompressXML(xml_text):
+    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")
+    
+def read_file_content(file,max_length):
+    if file.type == "application/json":
+        content = json.load(file)
+        return str(content)
+    elif file.type == "text/html" or file.type == "text/htm":
+        content = BeautifulSoup(file, "html.parser")
+        return content.text
+    elif file.type == "application/xml" or file.type == "text/xml":
+        tree = ET.parse(file)
+        root = tree.getroot()
+        xml = CompressXML(ET.tostring(root, encoding='unicode'))
+        return xml
+    elif file.type == "text/markdown" or file.type == "text/md":
+        md = mistune.create_markdown()
+        content = md(file.read().decode())
+        return content
+    elif file.type == "text/plain":
+        return file.getvalue().decode()
+    else:
+        return ""
 
 def main():
     user_prompt = st.text_area("Your question:", '', height=120)