Update app.py

app.py

@@ -6,6 +6,7 @@ import base64
 from bs4 import BeautifulSoup
 import hashlib
 import json
+import mimetypes
 
 EXCLUDED_FILES = ['app.py', 'requirements.txt', 'pre-requirements.txt', 'packages.txt', 'README.md','.gitattributes', "backup.py","Dockerfile"]
 
@@ -55,22 +56,38 @@ def get_download_link(file):
         href = f'<a href="data:file/octet-stream;base64,{b64}" download=\'{os.path.basename(file)}\'>Click to download {os.path.basename(file)}</a>'
     return href
 
+
+def is_binary(file_path):
+    """Determine if the given file is binary or text-based."""
+    try:
+        with open(file_path, 'rb') as f:
+            textchars = bytearray({7,8,9,10,12,13,27} | set(range(0x20, 0x100)) - {0x7f})
+            is_binary_string = lambda bytes: bool(bytes.translate(None, textchars))
+            return is_binary_string(f.read(1024))
+    except:
+        return True
+
 def file_expander(file_path):
     with st.expander(os.path.basename(file_path)):
-        # Edit functionality
-        file_content = ''
-        if 'edit_content' in st.session_state and st.session_state['edit_content'][0] == file_path:
-            file_content = st.session_state['edit_content'][1]
+        # Check if file is binary
+        if is_binary(file_path):
+            st.info("This is a binary file and cannot be edited.")
+            # Only display download and delete options for binary files
         else:
-            with open(file_path, "r") as f:
-                file_content = f.read()
-        
-        edited_content = st.text_area("Edit file content:", value=file_content, height=250, key=f"textarea_{file_path}")
-        if st.button("💾 Save", key=f"save_{file_path}"):
-            with open(file_path, "w") as f:
-                f.write(edited_content)
-            st.success(f"File {os.path.basename(file_path)} saved!")
-            st.session_state['edit_content'] = (file_path, edited_content)
+            # Text editing functionality for non-binary files
+            file_content = ''
+            if 'edit_content' in st.session_state and st.session_state['edit_content'][0] == file_path:
+                file_content = st.session_state['edit_content'][1]
+            else:
+                with open(file_path, "r") as f:
+                    file_content = f.read()
+            
+            edited_content = st.text_area("Edit file content:", value=file_content, height=250, key=f"textarea_{file_path}")
+            if st.button("💾 Save", key=f"save_{file_path}"):
+                with open(file_path, "w") as f:
+                    f.write(edited_content)
+                st.success(f"File {os.path.basename(file_path)} saved!")
+                st.session_state['edit_content'] = (file_path, edited_content)
 
         # Download link
         st.markdown(get_download_link(file_path), unsafe_allow_html=True)