From 70ea7c39b30b4d35bde645a05340fa73f67429ac Mon Sep 17 00:00:00 2001
From: "Bobby Abellana (aider)" <bobby@abellana.work>
Date: Tue, 11 Feb 2025 11:32:43 -0800
Subject: [PATCH] refactor: Update directory path handling for headless mode
 with validation

---
 src/streamlit_app.py | 34 ++++++++++++++++------------------
 1 file changed, 16 insertions(+), 18 deletions(-)

diff --git a/src/streamlit_app.py b/src/streamlit_app.py
index 7be922a..d130316 100644
--- a/src/streamlit_app.py
+++ b/src/streamlit_app.py
@@ -187,15 +187,12 @@ with col1:
                           key='source_dir_input',
                           help="Enter the full path to the directory containing files to process")
 
-        if st.button("Browse Source Directory", key='source_browse'):
-            try:
-                path = select_directory_with_dialog("Select Source Directory")
-                if path:
-                    st.session_state['source_dir'] = path
-                    st.session_state['source_dir_input'] = path
-                    st.experimental_rerun()
-            except Exception as e:
-                st.error(f"Error selecting directory: {str(e)}")
+        # Add a check and display for source directory status
+        if source_dir:
+            if os.path.exists(source_dir):
+                st.success(f"✅ Source directory exists: {source_dir}")
+            else:
+                st.error(f"❌ Source directory not found: {source_dir}")
 
         # Destination Directory        
         dest_dir = st.text_input("Destination Directory Path",
@@ -203,15 +200,16 @@ with col1:
                         key='dest_dir_input',
                         help="Enter the full path where processed files will be saved")
 
-        if st.button("Browse Destination Directory", key='dest_browse'):
-            try:
-                path = select_directory_with_dialog("Select Destination Directory")
-                if path:
-                    st.session_state['dest_dir'] = path
-                    st.session_state['dest_dir_input'] = path
-                    st.experimental_rerun()
-            except Exception as e:
-                st.error(f"Error selecting directory: {str(e)}")
+        # Add a check and display for destination directory status
+        if dest_dir:
+            dest_parent = os.path.dirname(dest_dir)
+            if os.path.exists(dest_parent):
+                if os.path.exists(dest_dir):
+                    st.success(f"✅ Destination directory exists: {dest_dir}")
+                else:
+                    st.info(f"i️ Destination directory will be created: {dest_dir}")
+            else:
+                st.error(f"❌ Parent directory not found: {dest_parent}")
 
 with col2:
     if file_type == "Excel":