change body_element

app.py

@@ -371,7 +371,7 @@ def get_body_text(xml_input):
         body = root.xpath("//tei:body", namespaces={"tei": "http://www.tei-c.org/ns/1.0"})
 
         if body:
-            body_element = body[0] if len(body) == 1 else None
+            body_element = body[0]
 
             if len(body_element) > 0:
                 # Extract the text content
@@ -452,7 +452,7 @@ def entities_to_xml(xml_content, ner):
         body = root.xpath("//tei:body", namespaces={"tei": "http://www.tei-c.org/ns/1.0"})
 
         if body:
-            body_element = body[0] if len(body) == 1 else None
+            body_element = body[0]
 
             if len(body_element) > 0: