Update app.py

app.py

@@ -1,154 +1,139 @@
-import os
-import streamlit as st
+"""
+Streamlit File Extension Renamer
+--------------------------------
+A lightweight Streamlit app intended for deployment on Hugging Face Spaces.
+The app lets users upload one or more files and rename (not convert) their
+extensions to any common file-type extension the user selects. Executable
+and binary files are blocked for security reasons.
+"""
+
+from __future__ import annotations
+
+import io
 import zipfile
+from datetime import datetime
 from pathlib import Path
-import io
 
-# Fix permission errors by setting environment variables BEFORE importing streamlit
-os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"
-os.environ["STREAMLIT_SERVER_HEADLESS"] = "true"
-os.environ["STREAMLIT_SERVER_ENABLE_CORS"] = "false"
-os.environ["STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION"] = "false"
-
-def is_safe_extension(extension):
-    """Check if the extension is safe to convert to"""
-    dangerous_extensions = ['.exe', '.bin', '.bat', '.cmd', '.com', '.scr', '.pif', '.vbs', '.js']
-    return extension.lower() not in dangerous_extensions
-
-def change_file_extension(file_content, original_name, new_extension):
-    """Change file extension and return new filename"""
-    if new_extension.startswith('.'):
-        new_extension = new_extension[1:]
-    
-    base_name = Path(original_name).stem
-    new_filename = f"{base_name}.{new_extension}"
-    
-    return new_filename, file_content
-
-def create_download_zip(files_data):
-    """Create a zip file containing all converted files"""
-    zip_buffer = io.BytesIO()
-    
-    with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zipf:
-        for filename, content in files_data:
-            zipf.writestr(filename, content)
-    
-    zip_buffer.seek(0)
-    return zip_buffer.getvalue()
-
-# Configure page
-st.set_page_config(
-    page_title="File Extension Converter",
-    page_icon="🔄",
-    layout="wide"
-)
-
-st.title("🔄 File Extension Converter")
-st.markdown("Convert any file extension to another (safely)")
-
-# Sidebar with instructions
-with st.sidebar:
-    st.header("📋 Instructions")
-    st.markdown("""
-    1. Upload one or more files
-    2. Enter the target extension
-    3. Download converted files
-    
-    **Safety Note:** 
-    - .exe and .bin files are blocked for security
-    - Only the extension is changed, not the file content
-    """)
-    
-    st.header("⚠️ Blocked Extensions")
-    st.code(".exe, .bin, .bat, .cmd, .com, .scr, .pif, .vbs, .js")
-
-# Main interface
-col1, col2 = st.columns([2, 1])
-
-with col1:
-    st.header("Upload Files")
-    uploaded_files = st.file_uploader(
-        "Choose files to convert",
+import streamlit as st
+
+# -----------------------------------------------------------------------------
+# Configuration
+# -----------------------------------------------------------------------------
+# List of target extensions displayed to the user. Extend this list as needed.
+ALLOWED_TARGET_EXTS: list[str] = [
+    ".txt",
+    ".pdf",
+    ".doc",
+    ".docx",
+    ".json",
+    ".csv",
+    ".xml",
+    ".html",
+    ".css",
+    ".js",
+    ".md",
+    ".jpg",
+    ".jpeg",
+    ".png",
+    ".gif",
+    ".bmp",
+    ".tiff",
+    ".svg",
+    ".ico",
+    ".mp3",
+    ".wav",
+    ".mp4",
+    ".avi",
+    ".mkv",
+    ".mov",
+    ".zip",
+    ".tar",
+    ".gz",
+    ".7z",
+]
+
+# Source‐file extensions that are disallowed (will be skipped if uploaded).
+DISALLOWED_SOURCE_EXTS: set[str] = {".exe", ".bin"}
+
+# -----------------------------------------------------------------------------
+# UI helpers
+# -----------------------------------------------------------------------------
+
+def make_sidebar() -> str:
+    """Return the user-chosen target extension from sidebar controls."""
+    st.sidebar.header("Settings")
+    target_ext = st.sidebar.selectbox(
+        label="Target extension (applied to **all** uploaded files)",
+        options=ALLOWED_TARGET_EXTS,
+        index=ALLOWED_TARGET_EXTS.index(".pdf"),
+    )
+
+    st.sidebar.markdown(
+        """❗ **Note**: This tool only renames the file extension. It does **not**
+        convert the underlying file format. Opening a renamed file with an
+        incompatible program may fail or lead to data corruption."""
+    )
+    return target_ext
+
+
+def make_uploader():
+    """Return the list of files uploaded by the user."""
+    return st.file_uploader(
+        "Upload one or more files",
         accept_multiple_files=True,
-        help="Upload any files you want to change the extension for"
+        type=[ext.lstrip(".") for ext in ALLOWED_TARGET_EXTS],
+        help="Drag-and-drop or browse for files to rename.",
     )
 
-with col2:
-    st.header("Target Extension")
-    new_extension = st.text_input(
-        "New extension",
-        placeholder="e.g., txt, pdf, jpg",
-        help="Enter the extension you want to convert to (without the dot)"
-    ).strip()
-
-if uploaded_files and new_extension:
-    if new_extension.startswith('.'):
-        new_extension = new_extension[1:]
-    
-    full_extension = f".{new_extension}"
-    
-    if not is_safe_extension(full_extension):
-        st.error(f"❌ Extension '{full_extension}' is not allowed for security reasons")
-        st.stop()
-    
-    st.success(f"✅ Converting to: **{full_extension}**")
-    
-    converted_files = []
-    
-    st.header("📁 Conversion Results")
-    
-    for uploaded_file in uploaded_files:
-        original_name = uploaded_file.name
-        file_content = uploaded_file.read()
-        
-        new_filename, new_content = change_file_extension(
-            file_content, original_name, new_extension
-        )
-        
-        converted_files.append((new_filename, new_content))
-        
-        col_orig, col_arrow, col_new = st.columns([3, 1, 3])
-        with col_orig:
-            st.text(f"📄 {original_name}")
-        with col_arrow:
-            st.text("➡️")
-        with col_new:
-            st.text(f"📄 {new_filename}")
-    
-    st.divider()
-    
-    st.header("💾 Download")
-    
-    if len(converted_files) == 1:
-        filename, content = converted_files[0]
+
+# -----------------------------------------------------------------------------
+# Core logic
+# -----------------------------------------------------------------------------
+
+def write_zip(uploaded_files: list[st.runtime.uploaded_file_manager.UploadedFile], target_ext: str) -> io.BytesIO:
+    """Return an in-memory ZIP archive of the uploaded files with the new extension."""
+    buffer = io.BytesIO()
+    with zipfile.ZipFile(buffer, "w", zipfile.ZIP_DEFLATED) as zf:
+        for file in uploaded_files:
+            orig_path = Path(file.name)
+            if orig_path.suffix.lower() in DISALLOWED_SOURCE_EXTS:
+                st.warning(f"⏭️  Skipping disallowed file: **{orig_path.name}**")
+                continue
+
+            renamed = orig_path.with_suffix(target_ext)
+            zf.writestr(renamed.name, file.read())
+            st.success(f"✅ Renamed **{orig_path.name}** → **{renamed.name}**")
+
+    buffer.seek(0)
+    return buffer
+
+
+# -----------------------------------------------------------------------------
+# Main app
+# -----------------------------------------------------------------------------
+
+def main() -> None:
+    st.set_page_config("Extension Renamer", page_icon="📄", layout="centered")
+    st.title("📄 Universal File-Extension Renamer")
+    st.write(
+        "Upload files, choose a new extension, and download them renamed in a single ZIP archive."
+    )
+
+    target_ext = make_sidebar()
+    uploaded_files = make_uploader()
+
+    if uploaded_files and st.button("🔄 Rename & Package"):
+        zip_buffer = write_zip(uploaded_files, target_ext)
+        timestamp = datetime.utcnow().strftime("%Y%m%dT%H%M%SZ")
         st.download_button(
-            label=f"📥 Download {filename}",
-            data=content,
-            file_name=filename,
-            mime="application/octet-stream"
+            label="⬇️ Download ZIP",
+            data=zip_buffer,
+            file_name=f"renamed_{timestamp}.zip",
+            mime="application/zip",
         )
-    else:
-        try:
-            zip_content = create_download_zip(converted_files)
-            st.download_button(
-                label=f"📥 Download All Files ({len(converted_files)} files)",
-                data=zip_content,
-                file_name="converted_files.zip",
-                mime="application/zip"
-            )
-            
-            st.info(f"💡 {len(converted_files)} files will be downloaded as a ZIP archive")
-        except Exception as e:
-            st.error(f"Error creating ZIP file: {str(e)}")
-
-elif uploaded_files and not new_extension:
-    st.warning("⚠️ Please enter a target extension")
-elif new_extension and not uploaded_files:
-    st.warning("⚠️ Please upload files to convert")
-
-st.divider()
-st.markdown("""
-<div style='text-align: center; color: #666; font-size: 0.8em;'>
-    Built with Streamlit 🎈 | Safe file extension conversion
-</div>
-""", unsafe_allow_html=True)
+
+    st.caption("© 2025 File-Extension Renamer • Built with Streamlit • Deployed on Hugging Face Spaces")
+
+
+if __name__ == "__main__":
+    main()