Update src/streamlit_app.py

src/streamlit_app.py

@@ -25,40 +25,33 @@ st.set_page_config(
 @st.cache_data
 def load_data():
     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",
+        # Load BITS file
+        df_bits = pd.read_excel("/mnt/data/BITS_INTERNS.xlsx", sheet_name="Form Responses 1")
+        df_bits["Source"] = "BITS"
+
+        # Load ICFAI file
+        df_icfai = pd.read_excel("/mnt/data/Summer of AI - ICFAI  (Responses) (3).xlsx", sheet_name=0)
+        df_icfai["Source"] = "ICFAI"
+
+        # Align columns by renaming if needed (e.g., if column names differ)
+        df_bits.columns = df_bits.columns.str.strip()
+        df_icfai.columns = df_icfai.columns.str.strip()
+
+        # Optional: Rename columns in ICFAI to match BITS (only if necessary)
+        df_icfai.rename(columns={
+            "Your Full Name": "Full Name",
+            "Laptop RAM Capacity": "Laptop RAM",
+            "Mobile RAM Capacity": "Mobile RAM",
             "Laptop OS": "Laptop Operating System",
             "Mobile OS": "Mobile Operating System"
-        })
+        }, inplace=True)
 
-        # 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
+        # Combine both
+        df_combined = pd.concat([df_bits, df_icfai], ignore_index=True)
+        return df_combined, None
 
     except Exception as e:
-        return None, f"Error loading or merging data: {str(e)}"
+        return None, f"❌ Error loading Excel files: {str(e)}"
 
 
 # Enhanced RAM extraction with better parsing