Update app.py

app.py

@@ -23,7 +23,7 @@ def chat_with_model(prompts):
 def generate_filename(prompt):
     safe_date_time = datetime.now().strftime("%Y_%m_%d_%H_%M_%S")
     safe_prompt = "".join(x for x in prompt if x.isalnum())[:50]
-    return f"{safe_date_time}_{safe_prompt}.txt"
+    return f"{safe_date_time}_{safe_prompt}.htm"
 
 def create_file(filename, prompt, response):
     with open(filename, 'w') as file:
@@ -33,7 +33,7 @@ def get_table_download_link(file_path):
     with open(file_path, 'r') as file:
         data = file.read()
     b64 = base64.b64encode(data.encode()).decode()  
-    href = f'<a href="data:file/txt;base64,{b64}" download="{os.path.basename(file_path)}">{os.path.basename(file_path)}</a>'
+    href = f'<a href="data:file/htm;base64,{b64}" download="{os.path.basename(file_path)}">{os.path.basename(file_path)}</a>'
     return href
     
 def main():
@@ -59,7 +59,7 @@ def main():
 
         st.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
-    htm_files = glob.glob("*.txt")
+    htm_files = glob.glob("*.htm")
     for file in htm_files:
         st.markdown(get_table_download_link(file), unsafe_allow_html=True)