Update src/streamlit_app.py

src/streamlit_app.py

@@ -22,38 +22,44 @@ st.set_page_config(
 )
 
 # Enhanced data loading with error handling - FIXED FILE PATH
-@st.cache_data
+
+  @st.cache_data
 def load_data():
-    def try_read_excel(path: str) -> Optional[pd.DataFrame]:
-        try:
-            df = pd.read_excel(path, sheet_name=0)
-            df.columns = df.columns.str.strip()
-            return df
-        except Exception:
-            return None
-
-    possible_files = [
-        "BITS_INTERNS.xlsx",
-        "src/BITS_INTERNS.xlsx",
-        "./BITS_INTERNS.xlsx",
-        "Summer of AI - ICFAI  (Responses) (3).xlsx",
-        "src/Summer of AI - ICFAI  (Responses) (3).xlsx",
-        "./Summer of AI - ICFAI  (Responses) (3).xlsx"
-    ]
-
-    dfs = []
-    for file in possible_files:
-        df = try_read_excel(file)
-        if df is not None:
-            st.info(f"✅ Loaded data from: {file}")
-            dfs.append(df)
-
-    if not dfs:
-        return None, "❌ No valid Excel files found. Please upload at least one file."
-
-    # Merge all valid DataFrames
-    combined_df = pd.concat(dfs, ignore_index=True)
-    return combined_df, None
+    try:
+        # Load BITS data
+        bits_df = pd.read_excel("BITS_INTERNS.xlsx", sheet_name="Form Responses 1")
+        bits_df.columns = bits_df.columns.str.strip()
+
+        # Load ICFAI data
+        icfai_df = pd.read_excel("Summer of AI - ICFAI  (Responses) (3).xlsx")
+        icfai_df.columns = icfai_df.columns.str.strip()
+
+        # Rename ICFAI columns to match BITS format
+        icfai_df = icfai_df.rename(columns={
+            "Name": "Full Name",
+            "Laptop RAM (e.g. 8GB)": "Laptop RAM",
+            "Mobile RAM (e.g. 6GB)": "Mobile RAM",
+            "Laptop OS": "Laptop Operating System",
+            "Mobile OS": "Mobile Operating System"
+        })
+
+        # Align columns in both datasets
+        common_columns = ["Full Name", "Laptop RAM", "Mobile RAM", "Laptop Operating System", "Mobile Operating System"]
+        bits_df = bits_df[common_columns]
+        icfai_df = icfai_df[common_columns]
+
+        # Add source tags
+        bits_df["Institute"] = "BITS"
+        icfai_df["Institute"] = "ICFAI"
+
+        # Combine datasets
+        combined_df = pd.concat([bits_df, icfai_df], ignore_index=True)
+
+        st.info("✅ Loaded and merged data from BITS and ICFAI.")
+        return combined_df, None
+
+    except Exception as e:
+        return None, f"Error loading or merging data: {str(e)}"
 
 
 # Enhanced RAM extraction with better parsing