Update app.py

app.py

@@ -68,9 +68,9 @@ def txt_to_html(text):
     html_content += "</body></html>"
     return html_content
 
-def deidentify_doc(pdftext="", maxtokens=600, temperature=1.2, top_probability=0.95):
+def deidentify_doc(pdftext="", prompt="", maxtokens=600, temperature=1.2, top_probability=0.95):
 
-    prompt = "Please anonymize the following clinical note. Replace all the following information with the term '[redacted]': Redact any strings that might be a name or initials, patients’ names, doctors’ names, the names Dr., redact any medical staff names, redact any strings that might be a location or address, such as '3970 Longview Drive', redact any strings that look like 'age 37', redact any dates and registration numbers, redact professions such as 'manager', redact any contact information."
+    # prompt = "Please anonymize the following clinical note. Replace all the following information with the term '[redacted]': Redact any strings that might be a name or initials, patients’ names, doctors’ names, the names Dr., redact any medical staff names, redact any strings that might be a location or address, such as '3970 Longview Drive', redact any strings that look like 'age 37', redact any dates and registration numbers, redact professions such as 'manager', redact any contact information."
 
     print('Max Tokens is  ',maxtokens)
     print('Temperature is  ',temperature)
@@ -118,12 +118,12 @@ def deidentify_doc(pdftext="", maxtokens=600, temperature=1.2, top_probability=0
 
     return output
 
-def pdf_to_text(file, maxtokens=600, temperature=1.2, top_probability=0.95):
+def pdf_to_text(file, prompt, maxtokens=600, temperature=1.2, top_probability=0.95):
     pdftext=""
     if(file):
        page2content = process_document(file, page_ids=[0])
        pdftext = page2content[1]
-    display_text = deidentify_doc(pdftext, maxtokens, temperature, top_probability)
+    display_text = deidentify_doc(pdftext, prompt, maxtokens, temperature, top_probability)
     html = txt_to_html(display_text)
     with open('out.html', "w", encoding="utf-8") as file:
         file.write(html)
@@ -149,7 +149,7 @@ prob_slider = gr.Slider(minimum=0, maximum=1, value=0.95, label="Max Probability
 max_tokens = gr.Number(value=600, label="Max Tokens")
 iface = gr.Interface(
     fn = pdf_to_text,
-    inputs = ['file', max_tokens, temp_slider, prob_slider],
+    inputs = ['file', "textbox", max_tokens, temp_slider, prob_slider],
     outputs="html",
     title='COBIx Endoscopy Report De-Identification',
     description="This application assists to remove personal information from the uploaded clinical report",