Update app.py

app.py

@@ -3,14 +3,16 @@ import openai
 import os
 import base64
 import glob
+
 from datetime import datetime
-from dotenv import load_dotenv
 from openai import ChatCompletion
 from xml.etree import ElementTree as ET
 from bs4 import BeautifulSoup
+
 import json
 
-load_dotenv()
+# from dotenv import load_dotenv
+# load_dotenv()
 
 openai.api_key = os.getenv('OPENAI_KEY')
 
@@ -19,7 +21,6 @@ def chat_with_model(prompts):
     #model = "gpt-4-32k"
     conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
     conversation.extend([{'role': 'user', 'content': prompt} for prompt in prompts])
-
     response = openai.ChatCompletion.create(model=model, messages=conversation)
     return response['choices'][0]['message']['content']
 
@@ -32,13 +33,34 @@ def create_file(filename, prompt, response):
     with open(filename, 'w') as file:
         file.write(f"<h1>Prompt:</h1> <p>{prompt}</p> <h1>Response:</h1> <p>{response}</p>")
 
-def get_table_download_link(file_path):
+def get_table_download_link_old(file_path):
     with open(file_path, 'r') as file:
         data = file.read()
     b64 = base64.b64encode(data.encode()).decode()  
     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 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':
+        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()):
@@ -96,10 +118,9 @@ def main():
     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(f"Delete {file}"):
+        if st.sidebar.button("🗑 Delete"):
             os.remove(file)
-            # Reload page after deletion to refresh the file list
             st.experimental_rerun()
 
 if __name__ == "__main__":