Update app.py

app.py

@@ -103,6 +103,10 @@ def deidentify_doc(llm, pdftext, maxtokens, temperature, top_probability):
     if find_index != -1:
         output = output[find_index:].strip()
 
+    last_index = output.rfind(' '.join(pdftext.split()[-3:]))
+    if last_index != -1:
+        output = output[:last_index].strip()
+
     output = llm.create_chat_completion(
         messages=[
             {"from": "user", "value": prompt + ' Text: ' + output},
@@ -116,6 +120,10 @@ def deidentify_doc(llm, pdftext, maxtokens, temperature, top_probability):
     find_index = output.find(' '.join(pdftext.split()[:3]))
     if find_index != -1:
         output = output[find_index:].strip()
+
+    last_index = output.rfind(' '.join(pdftext.split()[-3:]))
+    if last_index != -1:
+        output = output[:last_index].strip()
         
     return output