Update app.py

app.py

@@ -46,11 +46,11 @@ def get_llm_response(extractedtext1, extractedtext2):
     try:
         differences_list = extract_differences(message_text.split("<differences>")[1].split("</differences>")[0].strip())
     except Exception as e:
-        return message_text
+        return message_text, []
     
     difference_content = "\n\n\n".join([f"**Text1:**\n\n{d['text1']}\n\n**Text2:**\n\n{d['text2']}\n\n**Explanation:**\n\n{d['explanation']}\n\n----------------------" for d in differences_list])
     display_text = f"{before_differences}\n\n{difference_content}\n\n{after_differences}"
-    return display_text
+    return display_text, differences_list
     
 
 def extract_text_with_pypdf(pdf_path):
@@ -73,7 +73,7 @@ def main():
     
     # Add upload button to right column
     uploaded_file2 = col2.file_uploader("**Text 2**", type="pdf")
-
+    
     # Check if both files are uploaded
     if uploaded_file1 and uploaded_file2:
         # Get filenames from uploaded files
@@ -96,7 +96,21 @@ def main():
         # Add button at the bottom to run Find Differences function
         if st.button("Find Differences"):
             try:
-                display_text = get_llm_response(extracted_text1, extracted_text2)
+                display_text, parsed_data = get_llm_response(extracted_text1, extracted_text2)
+                display_text1 = extracted_text1
+                display_text2 = extracted_text2
+
+                for diff in parsed_data:
+                    diff_text1 = diff['text1'].strip()
+                    diff_text2 = diff['text2'].strip()
+                    display_text1 = f"**{diff_text1}**".join(display_text1.split(diff_text1)) if diff_text1 in display_text1
+                    display_text2 = f"**{diff_text2}**".join(display_text2.split(diff_text2)) if diff_text2 in display_text2
+                with col1.expander(filename1):
+                    st.write(display_text1)
+
+                with col2.expander(filename2):
+                    st.write(display_text2)
+                    
                 st.markdown(display_text)
             except Exception as e:
                 st.error(f"Error finding differences: {str(e)}")