Update app.py

app.py

@@ -132,10 +132,9 @@ def load_file(file_name):
     file_extension = os.path.splitext(file_name)[1].lower()
     try:
         if file_extension == '.pdf':
-            return convert_pdf_to_markdown(file_name)
+            return "PDF file. Use 'View' button to display the content."
         elif file_extension == '.html':
-            with open(file_name, 'r', encoding='utf-8') as file:
-                return file.read()
+            return "HTML file. Use 'View' button to display the content."
         else:  # For .md and other text files
             with open(file_name, 'r', encoding='utf-8') as file:
                 return file.read()
@@ -562,10 +561,11 @@ def compare_and_delete_files(files):
 def get_file_size(file_path):
     return os.path.getsize(file_path)
     
+
 def FileSidebar():
     # File Sidebar for files 🌐View, 📂Open, ▶️Run, and 🗑Delete per file
     all_files = glob.glob("*.md") + glob.glob("*_abstract.html") + glob.glob("*_abstract.pdf")
-    all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10]  # exclude files with short names
+    all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10 and os.path.basename(file)]  # exclude files with short names and empty names
     all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by filename length which puts similar prompts together
 
     # ⬇️ Download
@@ -593,12 +593,9 @@ def FileSidebar():
             if st.button("🌐", key=f"view_{timestamp}_{file}"):  # view emoji button
                 file_extension = os.path.splitext(file)[1].lower()
                 if file_extension == '.pdf':
-                    markdown_content = convert_pdf_to_markdown(file)
-                    st.markdown(markdown_content)
+                    display_pdf(file)
                 elif file_extension == '.html':
-                    with open(file, 'r', encoding='utf-8') as f:
-                        html_content = f.read()
-                    st.components.v1.html(html_content, height=600, scrolling=True)
+                    display_html(file)
                 else:
                     file_contents = load_file(file)
                     st.markdown(file_contents)
@@ -631,6 +628,16 @@ def FileSidebar():
                 st.session_state['next_action'] = next_action
 
     
+def display_pdf(file_path):
+    with open(file_path, "rb") as f:
+        base64_pdf = base64.b64encode(f.read()).decode('utf-8')
+    pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="800" height="800" type="application/pdf"></iframe>'
+    st.markdown(pdf_display, unsafe_allow_html=True)
+
+def display_html(file_path):
+    with open(file_path, 'r', encoding='utf-8') as f:
+        html_content = f.read()
+    st.components.v1.html(html_content, height=800, scrolling=True)
 
                 
     # 🚩File duplicate detector - useful to prune and view all.  Pruning works well by file size detection of two similar and flags the duplicate.