Update loading_file.py

loading_file.py

@@ -48,12 +48,18 @@ def fetch_data():
     df.dropna(how="all", inplace=True)
     df.reset_index(drop=True, inplace=True)
 
-    # Identify header row and reformat DataFrame
-    for idx, row in df.iterrows():
-        if "Application Number" in row.values:
-            df.columns = ["Application Number", "Decision"]
-            df = df.iloc[idx + 1:]  # Skip the header row
-            break
+    # Check columns and rename appropriately
+    print("Columns before renaming:", df.columns.tolist())  # For debugging purposes
+
+    # Attempt to rename columns only if there are sufficient columns
+    if len(df.columns) >= 2:
+        df.columns = ["Application Number", "Decision"]
+    else:
+        st.error("Insufficient data columns detected.")
+        return None
+
+    # Only keep relevant columns
+    df = df[["Application Number", "Decision"]]
 
     df["Application Number"] = df["Application Number"].astype(str)
     return df