first push

src/streamlit_app.py

@@ -9,24 +9,21 @@ from sklearn.neighbors import KNeighborsClassifier
 from sklearn.tree import DecisionTreeClassifier
 from sklearn.metrics import accuracy_score, r2_score
 
-# Define existing directories (must already be present in environment, but we'll ensure creation)
-UPLOAD_DIR = "/uploads"
-PROCESSED_DIR = "/processed"
-PKL_DIR = "/pkl"
-ENCODERS_DIR = "/encoders"
+# ====== Safe, relative folder paths ======
+UPLOAD_DIR = "uploads"
+PROCESSED_DIR = "processed"
+PKL_DIR = "pkl"
+ENCODERS_DIR = "encoders"
 
-# Create them if they don't exist
+# ====== Create directories if they don't exist ======
 for folder in [UPLOAD_DIR, PROCESSED_DIR, PKL_DIR, ENCODERS_DIR]:
     os.makedirs(folder, exist_ok=True)
 
-
-# Warn if directories don't exist
-for folder in [UPLOAD_DIR, PROCESSED_DIR, PKL_DIR, ENCODERS_DIR]:
-    if not os.path.exists(folder):
-        st.warning(f"Directory not found: {folder}. Please create it before using this app.")
-
 st.sidebar.title("Navigation")
-page = st.sidebar.selectbox("Go to", ["Upload File", "Preprocess Data", "Train & Select Model", "Create App"])
+page = st.sidebar.selectbox(
+    "Go to",
+    ["Upload File", "Preprocess Data", "Train & Select Model", "Create App"]
+)
 
 # ========================
 # Upload File Page
@@ -54,10 +51,7 @@ if page == "Upload File":
 # ========================
 elif page == "Preprocess Data":
     st.title("Preprocess Uploaded CSV")
-    if os.path.exists(UPLOAD_DIR):
-        files = [f for f in os.listdir(UPLOAD_DIR) if f.endswith(".csv")]
-    else:
-        files = []
+    files = [f for f in os.listdir(UPLOAD_DIR) if f.endswith(".csv")]
 
     if files:
         selected_file = st.selectbox("Select a CSV file to preprocess", files)
@@ -117,10 +111,7 @@ elif page == "Preprocess Data":
 elif page == "Train & Select Model":
     st.title("Train Models and Select the Best One")
 
-    if os.path.exists(PROCESSED_DIR):
-        processed_files = [f for f in os.listdir(PROCESSED_DIR) if f.endswith(".csv")]
-    else:
-        processed_files = []
+    processed_files = [f for f in os.listdir(PROCESSED_DIR) if f.endswith(".csv")]
 
     if processed_files:
         selected_file = st.selectbox("Select a processed CSV", processed_files)
@@ -190,10 +181,7 @@ elif page == "Train & Select Model":
 elif page == "Create App":
     st.title("Use Model for Prediction")
 
-    if os.path.exists(PROCESSED_DIR):
-        processed_files = [f for f in os.listdir(PROCESSED_DIR) if f.endswith(".csv")]
-    else:
-        processed_files = []
+    processed_files = [f for f in os.listdir(PROCESSED_DIR) if f.endswith(".csv")]
 
     if processed_files:
         selected_csv = st.selectbox("Select a processed CSV file", processed_files)