Update app.py

app.py

@@ -50,23 +50,7 @@ def extract_text_from_pdf(uploaded_file):
         st.error(f"PDF extraction error: {str(e)}")
         return ""
 
-def highlight_differences_side_by_side(text1, text2):
-    differ = difflib.Differ()
-    diff = list(differ.compare(text1.splitlines(keepends=True), text2.splitlines(keepends=True)))
 
-    highlighted_text1 = ""
-    highlighted_text2 = ""
-    
-    for line in diff:
-        if line.startswith("- "):
-            highlighted_text1 += f'<span style="background-color:#ffcccc">{line[2:]}</span>'
-        elif line.startswith("+ "):
-            highlighted_text2 += f'<span style="background-color:#ccffcc">{line[2:]}</span>'
-        elif line.startswith("  "):
-            highlighted_text1 += line[2:]
-            highlighted_text2 += line[2:]
-            
-    return highlighted_text1, highlighted_text2
 
 def highlight_differences_words(text1, text2):
     differ = difflib.Differ()
@@ -82,8 +66,8 @@ def highlight_differences_words(text1, text2):
             # Check for corresponding addition to highlight as changed
             if i + 1 < len(diff) and diff[i + 1].startswith("+ "):
                 added_word = diff[i + 1][2:]
-                highlighted_text2 += f'<span style="background-color:#ffffcc; display: inline-block;">{added_word}</span>'
-                diff[i + 1] = '  '
+                highlighted_text2 += f'<span style="background-color:#ffffcc; display: inline-block;">{added_word}</span>'  # Yellow for changed in text2
+                diff[i + 1] = '  '  # Consume the addition
             else:
                 highlighted_text2 += " "
         elif word.startswith("+ "):
@@ -91,7 +75,7 @@ def highlight_differences_words(text1, text2):
             highlighted_text2 += f'<span style="background-color:#ccffcc; display: inline-block;">{added_word}</span>'
             # Check for corresponding removal
             if i - 1 >= 0 and diff[i - 1].startswith("- "):
-                highlighted_text1 += f'<span style="background-color:#ffffcc; display: inline-block;">{diff[i-1][2:]}</span>'
+                highlighted_text1 += f'<span style="background-color:#ffffcc; display: inline-block;">{diff[i-1][2:]}</span>' # Yellow for changed in text1
                 diff[i-1] = '  '
             else:
                 highlighted_text1 += " "
@@ -101,7 +85,6 @@ def highlight_differences_words(text1, text2):
             highlighted_text2 += word[2:] + " "
 
     return highlighted_text1, highlighted_text2
-
 def calculate_similarity(text1, text2):
     if not text1.strip() or not text2.strip():
         return 0.0
@@ -206,21 +189,15 @@ def main():
                 
                 similarity_score = calculate_similarity(contract_text1, contract_text2)
                 
-                if comparison_format == "Side by Side":
-                    highlighted_diff1, highlighted_diff2 = highlight_differences_side_by_side(contract_text1, contract_text2)
-                    st.session_state.comparison_results = {
-                        'similarity_score': similarity_score,
-                        'highlighted_diff1': highlighted_diff1,
-                        'highlighted_diff2': highlighted_diff2,
-                        'format': 'side_by_side'
-                    }
-                else:
-                    highlighted_diff = highlight_differences_words(contract_text1, contract_text2)
-                    st.session_state.comparison_results = {
-                        'similarity_score': similarity_score,
-                        'highlighted_diff': highlighted_diff,
-                        'format': 'combined'
-                    }
+                
+                highlighted_diff1, highlighted_diff2 = highlight_differences_words(contract_text1, contract_text2)
+                st.session_state.comparison_results = {
+                    'similarity_score': similarity_score,
+                    'highlighted_diff1': highlighted_diff1,
+                    'highlighted_diff2': highlighted_diff2,
+                    'format': 'side_by_side'
+                }
+                
 
         # Display comparison results
         if st.session_state.comparison_results:
@@ -231,19 +208,15 @@ def main():
                 st.warning("Significant differences detected")
             
             st.markdown("**Visual Difference Highlighting:**")
-            if st.session_state.comparison_results['format'] == 'side_by_side':
-                col1, col2 = st.columns(2)
-                with col1:
-                    st.markdown("### Original Document")
-                    st.markdown(f'<div style="border:1px solid #ccc; padding:10px; white-space: pre-wrap; font-family: monospace; font-size: 0.9em;">{st.session_state.comparison_results["highlighted_diff1"]}</div>', unsafe_allow_html=True)
-                with col2:
-                    st.markdown("### Modified Document")
-                    st.markdown(f'<div style="border:1px solid #ccc; padding:10px; white-space: pre-wrap; font-family: monospace; font-size: 0.9em;">{st.session_state.comparison_results["highlighted_diff2"]}</div>', unsafe_allow_html=True)
-            else:
-                st.markdown(
-                    f'<div style="border:1px solid #ddd; padding:10px; max-height:400px; overflow-y:auto;">{st.session_state.comparison_results["highlighted_diff"]}</div>',
-                    unsafe_allow_html=True
-                )
+            
+            col1, col2 = st.columns(2)
+            with col1:
+                st.markdown("### Original Document")
+                st.markdown(f'<div style="border:1px solid #ccc; padding:10px; white-space: pre-wrap; font-family: monospace; font-size: 0.9em;">{st.session_state.comparison_results["highlighted_diff1"]}</div>', unsafe_allow_html=True)
+            with col2:
+                st.markdown("### Modified Document")
+                st.markdown(f'<div style="border:1px solid #ccc; padding:10px; white-space: pre-wrap; font-family: monospace; font-size: 0.9em;">{st.session_state.comparison_results["highlighted_diff2"]}</div>', unsafe_allow_html=True)
+            
 
     # ===== QUESTION ANALYSIS SECTION =====
     st.header("3. Clause Analysis")