Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import base64
 import glob
 import json
+import mistune
 
 from datetime import datetime
 from openai import ChatCompletion
@@ -71,7 +72,7 @@ def read_file_content(file):
     if file.type == "application/json":
         content = json.load(file)
         return str(content)
-    elif file.type == "text/html":
+    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":
@@ -79,7 +80,10 @@ def read_file_content(file):
         root = tree.getroot()
         #return ET.tostring(root, encoding='unicode')
         return CompressXML(ET.tostring(root, encoding='unicode'))
-        
+    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:
@@ -92,7 +96,7 @@ def main():
     file_content = ""
 
     user_prompt = st.text_area("Your question:", '', height=120)
-    uploaded_file = st.file_uploader("Choose a file", type=["xml", "json", "htm", "txt"])
+    uploaded_file = st.file_uploader("Choose a file", type=["xml", "json", "html", "htm", "md", "txt"])
 
     if user_prompt:
         prompts.append(user_prompt)