Update app.py

app.py

@@ -4,6 +4,44 @@ import pandas as pd
 import openpyxl  # Explicitly import openpyxl for Excel support
 import os
 
+directory_path = "/mnt/data/"
+
+if os.path.exists(directory_path):
+    # List files if the directory exists
+    files = os.listdir(directory_path)
+    print("Files in /mnt/data/:", files)
+else:
+    print("Directory does not exist:", directory_path)
+    # Optionally, create the directory if needed
+    os.makedirs(directory_path)
+    print("Directory created:", directory_path)
+
+
+# Primary directory
+directory_path = "/mnt/data/"
+
+# Check if primary directory exists; use current directory as fallback
+if not os.path.exists(directory_path):
+    print(f"Directory {directory_path} does not exist. Using current directory instead.")
+    directory_path = "."
+
+# List files in the confirmed directory path
+files = os.listdir(directory_path)
+print("Files in directory:", files)
+
+
+directory_path = "/mnt/data/"
+
+try:
+    files = os.listdir(directory_path)
+    print("Files in /mnt/data/:", files)
+except FileNotFoundError:
+    print(f"Error: The directory {directory_path} does not exist.")
+    # Handle the error by either using another directory or creating the directory
+    os.makedirs(directory_path)
+    print("Created directory:", directory_path)
+
+
 # List all files in the /mnt/data/ directory
 files = os.listdir("/mnt/data/")
 print("Files in /mnt/data/:", files)