big changes to the application flow - get_language

utils/oneclick.py

@@ -150,4 +150,4 @@ def generate_discharge_paper_one_click(
 
     except Exception as e:
         logger.error("Error in one-click discharge generation: %s\n%s", str(e), traceback.format_exc())
-        return None, f"Error: {str(e)}", None
+        return None, f"Error: {str(e)}", None

utils/responseparser.py

@@ -3,6 +3,14 @@ import lxml.etree as etree
 from datetime import datetime
 from typing import List, Dict, Optional, Union
 import base64
+import logging
+
+# Set up logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
 
 class PatientDataExtractor:
     """Class to extract fields from FHIR Patient Bundle (JSON) or C-CDA (XML)."""
@@ -221,7 +229,14 @@ class PatientDataExtractor:
             return lang[0] if lang else ""
         elif self.format == "json":
             comms = patient.get("communication", [])
-            return comms[0]["language"]["text"] if comms and "language" in comms[0] else ""
+            if comms and "language" in comms[0]:
+                lang = comms[0]["language"]
+                # Try 'text' first, then fall back to 'coding' if available
+                if "text" in lang:
+                    return lang["text"]
+                elif "coding" in lang and lang["coding"]:
+                    return lang["coding"][0].get("display", lang["coding"][0].get("code", ""))
+            return ""
 
     # Medications
     def get_medications(self) -> List[Dict[str, str]]:
@@ -235,7 +250,7 @@ class PatientDataExtractor:
                 start_list = med.xpath(".//hl7:effectiveTime/hl7:low/@value", namespaces=self.ns)
                 start = start_list[0] if start_list else ""
                 stop_list = med.xpath(".//hl7:effectiveTime/hl7:high/@value", namespaces=self.ns)
-                stop = stop_list[0] if stop_list else ""  # Safely handle missing <high>
+                stop = stop_list[0] if stop_list else ""
                 desc_list = med.xpath(".//hl7:manufacturedMaterial/hl7:code/@displayName", namespaces=self.ns)
                 desc = desc_list[0] if desc_list else ""
                 code_list = med.xpath(".//hl7:manufacturedMaterial/hl7:code/@code", namespaces=self.ns)
@@ -449,5 +464,4 @@ class PatientDataExtractor:
 
     def get_patient_ids(self) -> List[str]:
         """Return a list of all patient IDs."""
-        return [self.get_id() for _ in self.patients]
-
+        return [self.get_id() for _ in self.patients]