Update app.py

app.py

@@ -35,17 +35,20 @@ def make_llm_api_call(prompt):
     )
     return message
 
+
 def get_llm_response(extractedtext1, extractedtext2):
     prompt = DIFFERENTIATE_PROMPT.format(text1=extractedtext1, text2=extractedtext2)
     
     message = make_llm_api_call(prompt)
     
     message_text = message.content[0].text
-    before_differences = message_text.split("<differences>")[0]
-    after_differences = message_text.split("</differences>")[1]
+
     try:
+        before_differences = message_text.split("<differences>")[0]
+        after_differences = message_text.split("</differences>")[1]
         differences_list = extract_differences(message_text.split("<differences>")[1].split("</differences>")[0].strip())
     except Exception as e:
+        print("Error:", e)
         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])
@@ -84,10 +87,10 @@ def main():
             extracted_text1 = extract_text_with_pypdf(uploaded_file1)
             extracted_text2 = extract_text_with_pypdf(uploaded_file2)
             with col1.expander(filename1):
-                st.write(extracted_text1)
+                st.write("\n\n".join(extracted_text1.splitlines()))
             
             with col2.expander(filename2):
-                st.write(extracted_text2)
+                st.write("\n\n".join(extracted_text2.splitlines()))
                 
             st.success(f"Content of files **{filename1}** and **{filename2}** have been extracted successfully.")
         except Exception as e:
@@ -102,18 +105,20 @@ def main():
 
                 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 else display_text1
-                    display_text2 = f" **{diff_text2}** ".join(display_text2.split(diff_text2)) if diff_text2 in display_text2 else display_text2
+                    diff_text2 = diff['text2'].strip()                        
+                    diff_text1_phrase = "\n".join([f"<span style='background-color: grey;'>{t}</span>" for t in diff_text1.splitlines()])
+                    diff_text2_phrase = "\n".join([f"<span style='background-color: grey;'>{t}</span>" for t in diff_text2.splitlines()])
+                    display_text1 = diff_text1_phrase.join(display_text1.split(diff_text1)) if diff_text1 in display_text1 else display_text1
+                    display_text2 = diff_text2_phrase.join(display_text2.split(diff_text2)) if diff_text2 in display_text2 else display_text2
                 with col1.expander(f"{filename1} Difference Highlighted"):
-                    st.write(display_text1)
+                    st.write("\n\n".join(display_text1.splitlines()), unsafe_allow_html=True)
 
                 with col2.expander(f"{filename2} Difference Highlighted"):
-                    st.write(display_text2)
+                    st.write("\n\n".join(display_text2.splitlines()), unsafe_allow_html=True)
                     
                 st.markdown(display_text)
             except Exception as e:
                 st.error(f"Error finding differences: {str(e)}")
 
 if __name__ == "__main__":
-    main()
+    main()