bug8

app.py

@@ -2,8 +2,9 @@ import streamlit as st
 from langchain_pipeline import pipeline, model_names
 import pdfplumber
 from docx import Document
-from redlines import Redlines
+from difflib import unified_diff
 import tempfile
+from docx.shared import RGBColor
 
 def pdf_to_word(pdf_path, word_path):
     with pdfplumber.open(pdf_path) as pdf:
@@ -32,18 +33,19 @@ def redline_changes(original_path, revised_path, output_path):
     original_text = "\n".join([para.text for para in original_doc.paragraphs])
     revised_text = "\n".join([para.text for para in revised_doc.paragraphs])
 
-    redline = Redlines(original_text, revised_text)
-    diff_md = redline.output_markdown()
+    diff = unified_diff(original_text.splitlines(), revised_text.splitlines(), lineterm='')
 
-    # Create a new document and add the Markdown content as text
     diff_doc = Document()
-    for line in diff_md.split("\n"):
-        if line.startswith('- '):
+    for line in diff:
+        if line.startswith('-'):
             p = diff_doc.add_paragraph(line, style='Normal')
             p.font.color.rgb = RGBColor(255, 0, 0)  # Red
-        elif line.startswith('+ '):
+        elif line.startswith('+'):
             p = diff_doc.add_paragraph(line, style='Normal')
             p.font.color.rgb = RGBColor(0, 128, 0)  # Green
+        elif line.startswith('@@'):
+            p = diff_doc.add_paragraph(line, style='Normal')
+            p.font.color.rgb = RGBColor(0, 0, 255)  # Blue
         else:
             diff_doc.add_paragraph(line, style='Normal')