Update app.py

app.py

@@ -1,23 +1,23 @@
-from pathlib import Path
 import gradio as gr
 import logging
+from pathlib import Path
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
 # Default directory to scan
-DEFAULT_DATA_PATH = Path("/data")
+DEFAULT_DATA_PATH = Path.home()  # Use home directory as a dynamic default
 
 # Function to get file details and total storage usage
 def get_storage(data_path: str):
     # Convert to Path object
     data_path = Path(data_path)
-
+    
     # Validate directory
     if not data_path.exists() or not data_path.is_dir():
         logging.error(f"Directory not found: {data_path}")
-        return [], "Error: Directory not found or inaccessible."
-
+        return [], f"Error: Directory not found or inaccessible: {data_path}"
+    
     # Collect file details and calculate total size
     files = []
     total_size = 0
@@ -33,11 +33,11 @@ def get_storage(data_path: str):
                 total_size += stats.st_size
             except Exception as e:
                 logging.warning(f"Failed to process file: {file}. Error: {e}")
-
+    
     if not files:
         logging.info(f"No files found in directory: {data_path}")
         return [], "No files found in the specified directory."
-
+    
     # Total size in GB
     usage = f"{total_size / (1024.0 ** 3):.3f} GB"
     logging.info(f"Scanned {len(files)} files. Total size: {usage}")
@@ -48,7 +48,7 @@ with gr.Blocks() as app:
     # Title and description
     gr.Markdown("# 📁 File Storage Viewer\n"
                 "Easily view files and calculate storage usage in a specified directory.")
-
+    
     with gr.Row():
         # Input field for directory path
         dir_input = gr.Textbox(
@@ -56,19 +56,18 @@ with gr.Blocks() as app:
             label="Directory Path",
             placeholder="Enter the directory path to scan.",
         )
-
         # Button to trigger file listing
         fetch_btn = gr.Button("Fetch Files")
-
+    
     # Outputs: File list and total storage usage
     with gr.Row():
         file_table = gr.Dataframe(
             headers=["Original Name", "Path", "Size (MB)"],
-            interactive=True,
+            interactive=False,
             label="Files",
         )
         storage_usage = gr.Textbox(label="Total Storage Usage", interactive=False)
-
+    
     # Click action to trigger the `get_storage` function
     fetch_btn.click(
         get_storage,
@@ -77,5 +76,6 @@ with gr.Blocks() as app:
         show_progress=True
     )
 
-# Launch the Gradio app with access to custom paths
-app.launch(allowed_paths=[str(DEFAULT_DATA_PATH)], enable_queue=True)
+# Launch the Gradio app
+app.launch(allowed_paths=[str(DEFAULT_DATA_PATH)], enable_queue=True)
+