Update app.py

app.py

@@ -18,10 +18,10 @@ st.set_page_config(
     layout="wide")
 
 menu = ["txt", "htm", "md", "py"]
-choice = st.sidebar.selectbox("Choose output file type to save results", menu)
-choicePrefix = "Output and download file set to "
+choice = st.sidebar.selectbox("Output file type:", menu)
+choicePrefix = "Output file type is "
 if choice == "txt":
-     st.sidebar.write(choicePrefix + "Text file.")
+     st.sidebar.write(choicePrefix + "Text File.")
 elif choice == "htm":
      st.sidebar.write(choicePrefix + "HTML5.")
 elif choice == "md":
@@ -66,14 +66,11 @@ def get_table_download_link_old(file_path):
     return href
 
 def get_table_download_link(file_path):
-    import os
-    import base64
     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':
@@ -82,7 +79,6 @@ def get_table_download_link(file_path):
         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