Update app.py

app.py

@@ -659,7 +659,6 @@ def main():
                 with open(file, 'r') as f:
                     file_contents = f.read()
                     next_action='md'
-                    addDocumentHTML5(file_contents)
         with col2:
             st.markdown(get_table_download_link(file), unsafe_allow_html=True)
         with col3:
@@ -667,30 +666,34 @@ def main():
                 with open(file, 'r') as f:
                     file_contents = f.read()
                     next_action='open'
-                    addDocumentHTML5(file_contents)
         with col4:
             if st.button("🔍", key="read_"+file):  # search emoji button
                 with open(file, 'r') as f:
                     file_contents = f.read()
                     next_action='search'
-                    addDocumentHTML5(file_contents)
         with col5:
             if st.button("🗑", key="delete_"+file):
                 os.remove(file)
                 st.experimental_rerun()
+
+                
     if len(file_contents) > 0:
         if next_action=='open':
             file_content_area = st.text_area("File Contents:", file_contents, height=500)
+            addDocumentHTML5(file_contents)
         if next_action=='md':
             st.markdown(file_contents)
+            addDocumentHTML5(file_contents)
         if next_action=='search':
             file_content_area = st.text_area("File Contents:", file_contents, height=500)
             st.write('Reasoning with your inputs...')
 
             # new - llama
             response = StreamLLMChatResponse(file_contents)
-            filename = generate_filename(user_prompt, ".txt")
+            filename = generate_filename(user_prompt, ".md")
             create_file(filename, response, '', should_save)
+            addDocumentHTML5(file_contents)
+            addDocumentHTML5(response)
             
             # old - gpt
             #response = chat_with_model(user_prompt, file_contents, model_choice)