Update app.py

app.py

@@ -112,28 +112,7 @@ def create_masked_text(input_text, entities, mask_dict):
             masked_text = masked_text[:entity['start']] + mask_dict[entity['word']] + masked_text[entity['end']:]
     return masked_text
 
-def export_masked_text(masked_text, file_type):
-    if file_type == "txt":
-        return masked_text.encode("utf-8")
-    elif file_type == "pdf":
-        pdf_buffer = io.BytesIO()
-        pdf = FPDF()
-        pdf.add_page()
-        pdf.set_font("Arial", size=12)
-        pdf.multi_cell(0, 10, masked_text)
-        pdf.output(pdf_buffer)
-        pdf_buffer.seek(0)
-        return pdf_buffer.getvalue()
-    elif file_type == "docx":
-        doc = docx.Document()
-        doc.add_paragraph(masked_text)
-        buffer = io.BytesIO()
-        doc.save(buffer)
-        buffer.seek(0)
-        return buffer.getvalue()
-    else:
-        st.error("Unsupported file type for export")
-        return None
+
 Run_Button = st.button("Run")
 
 if Run_Button and input_text:
@@ -155,6 +134,7 @@ if Run_Button and input_text:
                 entity['end'] += offset
         
         all_outputs.extend(output)
+        
     
     # Combine entities
     
@@ -171,7 +151,7 @@ if Run_Button and input_text:
             entity['masked_word'] = mask_dict.get(entity['word'], entity['word'])
         else:
             entity['masked_word'] = entity['word']
-    
+    print("output_comb", output_comb)
     #df = pd.DataFrame.from_dict(output_comb)
     #cols_to_keep = ['word', 'entity_group', 'score', 'start', 'end']
     #df_final = df[cols_to_keep].loc[:,~df.columns.duplicated()].copy()
@@ -189,21 +169,8 @@ if Run_Button and input_text:
         else:
             label = entity['entity_group']
         spacy_display["ents"].append({"start": entity["start"], "end": entity["end"], "label": label})
-        # Custom CSS to prevent label overlap
-    custom_css = """
-    <style>
-    .entity-label {
-        font-size: 0.7em;
-        line-height: 1;
-        padding: 0.25em;
-        border-radius: 0.25em;
-        top: -1.5em;
-        position: relative;
-    }
-    </style>
-    """
     
-    html = custom_css + spacy.displacy.render(spacy_display, style="ent", minify=True, manual=True)
+    html = spacy.displacy.render(spacy_display, style="ent", minify=True, manual=True)
     st.write(html, unsafe_allow_html=True)
 
     st.subheader("Masking Dictionary")