error fix

app.py

@@ -22,20 +22,23 @@ def remove_special_characters(text):
     cleaned_text = re.sub(pattern, '', text)
     return cleaned_text
 
-def extract_entities(text, labels):
-    doc = nlp(text)
-    entities = []
-    for ent in doc.ents:
-        if ent.label_ in labels:
-            entities.append((ent.text, ent.label_))
-    return entities
+def get_sentences(further_cleaned_text):
+    doc = nlp(further_cleaned_text)
+    sentences = [sent.text for sent in doc.sents]
+    return sentences
+    # doc = nlp(text)
+    # entities = []
+    # for ent in doc.ents:
+    #     if ent.label_ in labels:
+    #         entities.append((ent.text, ent.label_))
+    # return entities
 
 def present(email_content, labels):
     email = accept_mail(email_content)
     cleaned_text = clean_email(email)
     further_cleaned_text = remove_special_characters(cleaned_text)
-    entities = extract_entities(further_cleaned_text, labels)
-    entity_info = '\n'.join([f"{text}: {label}" for text, label in entities])
+    sentence_list = get_sentences(further_cleaned_text)
+    # entity_info = '\n'.join([f"{text}: {label}" for text, label in entities])
     
     email_info = {
         "Subject": email.subject,
@@ -43,11 +46,13 @@ def present(email_content, labels):
         "To": email.to,
         "Date": email.date,
         "Cleaned Body": further_cleaned_text,
-        "Extracted Entities": entity_info
+        # "Extracted Entities": entity_info
     }
     return [email_info[key] for key in email_info]
 
-labels = ["PERSON", "PRODUCT", "DEAL", "ORDER", "ORDER PAYMENT METHOD", "STORE", "LEGAL ENTITY", "MERCHANT", "FINANCIAL TRANSACTION", "UNCATEGORIZED", "DATE"]
+labels = ["PERSON", "PRODUCT", "DEAL", "ORDER", 
+          "ORDER PAYMENT METHOD", "STORE", "LEGAL ENTITY", 
+          "MERCHANT", "FINANCIAL TRANSACTION", "UNCATEGORIZED", "DATE"]
 
 demo = gr.Interface(
     fn=present, 
@@ -61,7 +66,7 @@ demo = gr.Interface(
         gr.components.Textbox(label="To"),
         gr.components.Textbox(label="Date"),
         gr.components.Textbox(label="Cleaned Body"),
-        gr.components.Textbox(label="Extracted Entities")
+        # gr.components.Textbox(label="Extracted Entities")
     ],
     title="Email Info",
     description="Enter the email content below to view its details and detected entities."