Update app.py

app.py

@@ -67,27 +67,30 @@ def is_binary(file_path):
     except:
         return True
 
+def is_text_file(file_path):
+    """Check if a file is a text file."""
+    text_file_types = ['.txt', '.py', '.md', '.html', '.css', '.js', '.json', '.xml']
+    return any(file_path.endswith(ext) for ext in text_file_types)
+
 def file_expander(file_path):
     with st.expander(os.path.basename(file_path)):
-        # 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:
-            # Text editing functionality for non-binary files
+        # Handle text files
+        if is_text_file(file_path):
             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:
+                with open(file_path, "r", encoding='utf-8') 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:
+                with open(file_path, "w", encoding='utf-8') 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)
+        else:
+            st.info("This is a binary file and cannot be edited.")
 
         # Download link
         st.markdown(get_download_link(file_path), unsafe_allow_html=True)