revert

app.py

@@ -35,9 +35,10 @@ min_transaction_overdraft = st.number_input(
 
 uploaded_file = st.file_uploader("Choose a file")
 if uploaded_file is not None:
+    diff = ""
     with st.spinner('Please wait ...'):
         try:
-            output_path = pipeline(
+            diff = pipeline(
                 uploaded_file,
                 model_name,
                 balance_type,
@@ -46,13 +47,39 @@ if uploaded_file is not None:
                 min_overdrawn_fee,
                 min_transaction_overdraft
             )
-            st.success(f"Redlined document saved to {output_path}")
-            with open(output_path, "rb") as file:
-                st.download_button(
-                    label="Download Redlined Document",
-                    data=file,
-                    file_name="Redlined_Reg_E_Notice.docx",
-                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-                )
         except Exception as e:
             st.exception(e)
+
+    diff_lines = diff.split("\n")
+
+    styled_diff = """
+    <style>
+    body {
+        font-family: 'Times New Roman', serif;
+        line-height: 1.5;
+    }
+    .diff {
+        margin: 10px 0;
+        padding: 5px;
+    }
+    .add {
+        color: green;
+    }
+    .remove {
+        color: red;
+    }
+    </style>
+    <div>
+    """
+    for line in diff_lines:
+        if line.startswith('+'):
+            styled_diff += f'<div class="diff add">{line}</div>'
+        elif line.startswith('-'):
+            styled_diff += f'<div class="diff remove">{line}</div>'
+        else:
+            styled_diff += f'<div class="diff">{line}</div>'
+    styled_diff += "</div>"
+
+    st.markdown(styled_diff, unsafe_allow_html=True)
+
+st.markdown("The key changes are:")