viboognesh commited on
Commit
3073f86
·
verified ·
1 Parent(s): 1e69d2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -10
app.py CHANGED
@@ -35,17 +35,20 @@ def make_llm_api_call(prompt):
35
  )
36
  return message
37
 
 
38
  def get_llm_response(extractedtext1, extractedtext2):
39
  prompt = DIFFERENTIATE_PROMPT.format(text1=extractedtext1, text2=extractedtext2)
40
 
41
  message = make_llm_api_call(prompt)
42
 
43
  message_text = message.content[0].text
44
- before_differences = message_text.split("<differences>")[0]
45
- after_differences = message_text.split("</differences>")[1]
46
  try:
 
 
47
  differences_list = extract_differences(message_text.split("<differences>")[1].split("</differences>")[0].strip())
48
  except Exception as e:
 
49
  return message_text, []
50
 
51
  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():
84
  extracted_text1 = extract_text_with_pypdf(uploaded_file1)
85
  extracted_text2 = extract_text_with_pypdf(uploaded_file2)
86
  with col1.expander(filename1):
87
- st.write(extracted_text1)
88
 
89
  with col2.expander(filename2):
90
- st.write(extracted_text2)
91
 
92
  st.success(f"Content of files **{filename1}** and **{filename2}** have been extracted successfully.")
93
  except Exception as e:
@@ -102,18 +105,20 @@ def main():
102
 
103
  for diff in parsed_data:
104
  diff_text1 = diff['text1'].strip()
105
- diff_text2 = diff['text2'].strip()
106
- display_text1 = f" **{diff_text1}** ".join(display_text1.split(diff_text1)) if diff_text1 in display_text1 else display_text1
107
- display_text2 = f" **{diff_text2}** ".join(display_text2.split(diff_text2)) if diff_text2 in display_text2 else display_text2
 
 
108
  with col1.expander(f"{filename1} Difference Highlighted"):
109
- st.write(display_text1)
110
 
111
  with col2.expander(f"{filename2} Difference Highlighted"):
112
- st.write(display_text2)
113
 
114
  st.markdown(display_text)
115
  except Exception as e:
116
  st.error(f"Error finding differences: {str(e)}")
117
 
118
  if __name__ == "__main__":
119
- main()
 
35
  )
36
  return message
37
 
38
+
39
  def get_llm_response(extractedtext1, extractedtext2):
40
  prompt = DIFFERENTIATE_PROMPT.format(text1=extractedtext1, text2=extractedtext2)
41
 
42
  message = make_llm_api_call(prompt)
43
 
44
  message_text = message.content[0].text
45
+
 
46
  try:
47
+ before_differences = message_text.split("<differences>")[0]
48
+ after_differences = message_text.split("</differences>")[1]
49
  differences_list = extract_differences(message_text.split("<differences>")[1].split("</differences>")[0].strip())
50
  except Exception as e:
51
+ print("Error:", e)
52
  return message_text, []
53
 
54
  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])
 
87
  extracted_text1 = extract_text_with_pypdf(uploaded_file1)
88
  extracted_text2 = extract_text_with_pypdf(uploaded_file2)
89
  with col1.expander(filename1):
90
+ st.write("\n\n".join(extracted_text1.splitlines()))
91
 
92
  with col2.expander(filename2):
93
+ st.write("\n\n".join(extracted_text2.splitlines()))
94
 
95
  st.success(f"Content of files **{filename1}** and **{filename2}** have been extracted successfully.")
96
  except Exception as e:
 
105
 
106
  for diff in parsed_data:
107
  diff_text1 = diff['text1'].strip()
108
+ diff_text2 = diff['text2'].strip()
109
+ diff_text1_phrase = "\n".join([f"<span style='background-color: grey;'>{t}</span>" for t in diff_text1.splitlines()])
110
+ diff_text2_phrase = "\n".join([f"<span style='background-color: grey;'>{t}</span>" for t in diff_text2.splitlines()])
111
+ display_text1 = diff_text1_phrase.join(display_text1.split(diff_text1)) if diff_text1 in display_text1 else display_text1
112
+ display_text2 = diff_text2_phrase.join(display_text2.split(diff_text2)) if diff_text2 in display_text2 else display_text2
113
  with col1.expander(f"{filename1} Difference Highlighted"):
114
+ st.write("\n\n".join(display_text1.splitlines()), unsafe_allow_html=True)
115
 
116
  with col2.expander(f"{filename2} Difference Highlighted"):
117
+ st.write("\n\n".join(display_text2.splitlines()), unsafe_allow_html=True)
118
 
119
  st.markdown(display_text)
120
  except Exception as e:
121
  st.error(f"Error finding differences: {str(e)}")
122
 
123
  if __name__ == "__main__":
124
+ main()