Update app.py

app.py

@@ -7,7 +7,7 @@ from io import BytesIO
 
 st.set_page_config(page_title="Protein Repeat Comparator", layout="centered")
 st.title("🧬 Protein Repeat Comparator")
-st.write("Upload two Excel files with protein data. Frequency values should start from the second row.")
+st.write("Upload two Excel files with protein data. Frequency values should start from the first row (header).")
 
 uploaded_file1 = st.file_uploader("Upload First Excel File", type=["xlsx"])
 uploaded_file2 = st.file_uploader("Upload Second Excel File", type=["xlsx"])
@@ -40,12 +40,17 @@ if uploaded_file1 and uploaded_file2:
                 freq1 = row1[repeat_col]
                 freq2 = row2[repeat_col]
 
-                if pd.isna(freq1) or pd.isna(freq2) or freq1 == 0:
+                if pd.isna(freq1) or pd.isna(freq2):
                     continue
 
                 if freq1 != freq2:
+                    if freq1 == 0:
+                        pct_change = "Infinity"
+                    else:
+                        pct_change = ((freq2 - freq1) / freq1) * 100
+                        pct_change = round(pct_change, 2)
+
                     diff = abs(freq1 - freq2)
-                    pct_change = ((freq2 - freq1) / freq1) * 100
                     differences.append({
                         id_col: entry_id,
                         name_col: protein_name,
@@ -60,24 +65,31 @@ if uploaded_file1 and uploaded_file2:
             result_df = pd.DataFrame(differences)
             result_df = result_df.sort_values(by="Difference", ascending=False)
 
-            # Style for Excel (green for +, red for -)
+            # Show DataFrame in Streamlit app
+            st.subheader("🔍 View Changed Repeats")
+            st.dataframe(result_df, use_container_width=True)
+
+            # Apply styling
             def color_pct(val):
-                if val > 0:
+                if isinstance(val, str) and val == "Infinity":
                     return 'color: green'
-                elif val < 0:
-                    return 'color: red'
+                elif isinstance(val, (int, float)):
+                    if val > 0:
+                        return 'color: green'
+                    elif val < 0:
+                        return 'color: red'
                 return ''
 
             styled_df = result_df.style.applymap(color_pct, subset=["%age Change"])
 
+            # Save styled output
             output = BytesIO()
             with pd.ExcelWriter(output, engine='openpyxl') as writer:
                 styled_df.to_excel(writer, index=False, sheet_name="Changed Repeats")
             output.seek(0)
 
-            st.success("✅ Comparison complete. Showing only changed repeats.")
             st.download_button(
-                label="📥 Download Excel",
+                label="📥 Download Excel File",
                 data=output,
                 file_name="changed_repeats_with_percentage.xlsx",
                 mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"