Update app.py

app.py

@@ -14,22 +14,24 @@ uploaded_file2 = st.file_uploader("Upload Second Excel File", type=["xlsx"])
 
 if uploaded_file1 and uploaded_file2:
     try:
-        # Read both Excel files, assuming header is in second row
+        # Read Excel files (header starts from second row)
         df1 = pd.read_excel(uploaded_file1, header=1)
         df2 = pd.read_excel(uploaded_file2, header=1)
 
-        # Automatically detect first two columns
+        # Automatically get ID and Name column
         id_col = df1.columns[0]
         name_col = df1.columns[1]
         repeat_cols = df1.columns[2:]
 
+        # Ensure all column names are strings
+        repeat_cols = [str(col) for col in repeat_cols]
+
         records = []
 
         for _, row1 in df1.iterrows():
             entry_id = row1[id_col]
             protein_name = row1[name_col]
 
-            # Match in second file
             match = df2[(df2[id_col] == entry_id) & (df2[name_col] == protein_name)]
             if match.empty:
                 continue
@@ -43,27 +45,31 @@ if uploaded_file1 and uploaded_file2:
                     records.append({
                         id_col: entry_id,
                         name_col: protein_name,
-                        "Repeat": repeat,
+                        "Repeat": str(repeat),
                         "Frequency File 1": freq1,
                         "Frequency File 2": freq2,
                         "Difference": diff
                     })
 
-        result_df = pd.DataFrame(records)
-        result_df = result_df.sort_values(by="Difference", ascending=False)
+        if records:
+            result_df = pd.DataFrame(records)
+            result_df = result_df.sort_values(by="Difference", ascending=False)
 
-        output = BytesIO()
-        with pd.ExcelWriter(output, engine="openpyxl") as writer:
-            result_df.to_excel(writer, index=False)
-        output.seek(0)
+            output = BytesIO()
+            with pd.ExcelWriter(output, engine="openpyxl") as writer:
+                result_df.to_excel(writer, index=False)
+            output.seek(0)
 
-        st.success("✅ Comparison complete! Showing only changed repeats.")
-        st.download_button(
-            label="📥 Download Changed Repeats Excel",
-            data=output,
-            file_name="changed_protein_repeats.xlsx",
-            mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
-        )
+            st.success("✅ Comparison complete! Only changed repeats are included.")
+            st.download_button(
+                label="📥 Download Changed Repeats Excel",
+                data=output,
+                file_name="changed_protein_repeats.xlsx",
+                mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+            )
+        else:
+            st.info("No changes in repeat frequencies were found.")
 
     except Exception as e:
-        st.error(f"⚠️ Error: {e}")
+        st.error(f"⚠️ Error: {e}")
+