Update app.py

app.py

@@ -451,6 +451,7 @@ with tabs[1]:
 # Diplomatic Edition Tab
 # -------------------------------
 
+# Function to remove diacritics from text
 # Function to remove diacritics from text
 def remove_diacritics(text):
     """
@@ -466,34 +467,49 @@ def remove_diacritics(text):
 def render_diplomatic(text_elem):
     """
     Transforms the XML Text element into uppercase Greek text without diacritics and spaces,
-    with line breaks at <lb> tags.
+    with line breaks at <lb> tags. Handles <expan> tags by including only the <abbr> text.
     """
     lines = []
     current_line = []
-
-    for elem in text_elem.iter():
+    
+    # Define a helper function to process elements recursively
+    def process_element(elem):
         if elem.tag == 'lb':
-            # Finalize the current line
-            line_text = ''.join(current_line).strip()
-            if line_text:
-                # Remove diacritics and spaces, then convert to uppercase
-                line_text = remove_diacritics(line_text).replace(' ', '').upper()
-                lines.append(line_text)
-            current_line = []
-        elif elem.text:
-            # Append text, handling any nested tags
-            current_line.append(elem.text)
-        if elem.tail:
-            # Append tail text after a nested tag
-            current_line.append(elem.tail)
-
-    # Add the last line if exists
-    if current_line:
+            finalize_current_line()
+        elif elem.tag == 'expan':
+            abbr_elem = elem.find('abbr')
+            if abbr_elem is not None and abbr_elem.text:
+                current_line.append(abbr_elem.text)
+            # Do not process <ex> or any other children within <expan>
+        else:
+            if elem.text:
+                current_line.append(elem.text)
+            # Recursively process child elements
+            for child in elem:
+                process_element(child)
+            if elem.tail:
+                current_line.append(elem.tail)
+    
+    def finalize_current_line():
+        """
+        Finalizes the current line by removing diacritics, spaces, converting to uppercase,
+        and appending it to the lines list.
+        """
+        nonlocal current_line
         line_text = ''.join(current_line).strip()
         if line_text:
+            # Remove diacritics and spaces, then convert to uppercase
             line_text = remove_diacritics(line_text).replace(' ', '').upper()
             lines.append(line_text)
-
+        current_line = []
+    
+    # Start processing from the root text element
+    process_element(text_elem)
+    
+    # Finalize the last line if any
+    if current_line:
+        finalize_current_line()
+    
     # Join all lines with newline characters
     return '\n'.join(lines)
 
@@ -518,7 +534,6 @@ with tabs[2]:
     else:
         st.warning("No text found for the selected inscription.")
 
-
 # -------------------------------
 # Editor Edition Tab
 # -------------------------------