word doc

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from langchain_pipeline import pipeline, model_names
+from combined_pipeline import pipeline, model_names
 
 st.title("Canarie AI Prototype")
 st.subheader("Finding the canarie in the coal mine")
@@ -35,10 +35,9 @@ 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:
-            diff = pipeline(
+            output_path = pipeline(
                 uploaded_file,
                 model_name,
                 balance_type,
@@ -47,39 +46,13 @@ 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:")

requirements.txt

@@ -6,6 +6,8 @@ langchain_astradb
 langchain_core
 langchain_openai
 langchain_anthropic
-langchain_google_genai
+langchain_google_genai==1.0.5
 google_generativeai
-langchain-google-genai==1.0.5
+pdf2docx
+pymupdf
+python-docx