Update app.py

app.py

@@ -18,10 +18,10 @@ if uploaded_file1 and uploaded_file2:
         df1 = pd.read_excel(uploaded_file1, header=1)
         df2 = pd.read_excel(uploaded_file2, header=1)
 
-        # Column names
-        id_col = "Entry ID"
-        name_col = "Protein Name"
-        repeat_cols = [col for col in df1.columns if col not in [id_col, name_col]]
+        # Automatically detect first two columns
+        id_col = df1.columns[0]
+        name_col = df1.columns[1]
+        repeat_cols = df1.columns[2:]
 
         records = []
 
@@ -29,7 +29,7 @@ if uploaded_file1 and uploaded_file2:
             entry_id = row1[id_col]
             protein_name = row1[name_col]
 
-            # Match protein in second file
+            # Match in second file
             match = df2[(df2[id_col] == entry_id) & (df2[name_col] == protein_name)]
             if match.empty:
                 continue
@@ -41,8 +41,8 @@ if uploaded_file1 and uploaded_file2:
                 if freq1 != freq2:
                     diff = abs(freq1 - freq2)
                     records.append({
-                        "Entry ID": entry_id,
-                        "Protein Name": protein_name,
+                        id_col: entry_id,
+                        name_col: protein_name,
                         "Repeat": repeat,
                         "Frequency File 1": freq1,
                         "Frequency File 2": freq2,
@@ -52,7 +52,6 @@ if uploaded_file1 and uploaded_file2:
         result_df = pd.DataFrame(records)
         result_df = result_df.sort_values(by="Difference", ascending=False)
 
-        # In-memory Excel
         output = BytesIO()
         with pd.ExcelWriter(output, engine="openpyxl") as writer:
             result_df.to_excel(writer, index=False)
@@ -67,5 +66,4 @@ if uploaded_file1 and uploaded_file2:
         )
 
     except Exception as e:
-        st.error(f"⚠️ Error: {e}")
-        
+        st.error(f"⚠️ Error: {e}")