safe parsing llm output

app.py

@@ -216,47 +216,44 @@ def extract_notes(text, section_name):
     return []
 
 def parse_perfume_description(text: str) -> dict:
+    def safe_search(pattern, text, flags=0):
+        match = re.search(pattern, text, flags)
+        return match.group(1).strip() if match else None
+
     # Perfume Name
-    perfume_name = re.search(r'Perfume Name:\s*(.+)', text).group(1).strip()
+    perfume_name = safe_search(r'Perfume Name:\s*(.+)', text) or ""
+
+    # Concentration Type (same style: expects `Concentration Type:`)
+    concentration_type = safe_search(r'Concentration Type:\s*(.+)', text) or ""
 
-    # Tagline (quoted)
-    tagline = re.search(r'Tagline:\s*"(.*?)"', text, re.DOTALL)
-    tagline = tagline.group(1).strip() if tagline else ""
+    # Tagline
+    tagline = safe_search(r'Tagline:\s*"(.*?)"', text, re.DOTALL) or ""
 
     # Poetic Olfactory Description
-    poetic_desc_match = re.search(
-        r'Poetic Olfactory Description:\s*"(.*?)"', text, re.DOTALL)
-    if poetic_desc_match:
-        poetic_desc = poetic_desc_match.group(1).strip()
-    else:
-        poetic_desc_match = re.search(
-            r'Poetic Olfactory Description:\s*(.*?)\s*(Olfactory Pyramid:|Image Description:|General Atmosphere:)', 
-            text, re.DOTALL)
-        poetic_desc = poetic_desc_match.group(1).strip() if poetic_desc_match else ""
-
-    # General Atmosphere: stop at Image Description if present
-    general_atmosphere_match = re.search(
-        r'General Atmosphere:\s*(.*?)(?:\s*Image Description:|$)', text, re.DOTALL)
-    general_atmosphere = general_atmosphere_match.group(1).strip() if general_atmosphere_match else ""
+    poetic_desc = safe_search(r'Poetic Olfactory Description:\s*"(.*?)"', text, re.DOTALL)
+    if poetic_desc is None:
+        poetic_desc = safe_search(
+            r'Poetic Olfactory Description:\s*(.*?)\s*(Olfactory Pyramid:|Image Description:|General Atmosphere:|Concentration Type:)', 
+            text, re.DOTALL) or ""
+
+    # General Atmosphere
+    general_atmosphere = safe_search(
+        r'General Atmosphere:\s*(.*?)(?:\s*Image Description:|$)', 
+        text, re.DOTALL) or ""
 
     # Image Description
-    image_desc_match = re.search(
-        r'Image Description:\s*"(.*?)"', text, re.DOTALL)
-    if image_desc_match:
-        image_desc = image_desc_match.group(1).strip()
-    else:
-        image_desc_match = re.search(
-            r'Image Description:\s*(.*?)$', text, re.DOTALL)
-        image_desc = image_desc_match.group(1).strip() if image_desc_match else ""
-
-    # 🗂️ Smart bullet extractor
-    
+    image_desc = safe_search(r'Image Description:\s*"(.*?)"', text, re.DOTALL)
+    if image_desc is None:
+        image_desc = safe_search(r'Image Description:\s*(.*?)$', text, re.DOTALL) or ""
+
+    # Notes
     top_notes = extract_notes(text, 'Top Notes')
     heart_notes = extract_notes(text, 'Heart Notes')
     base_notes = extract_notes(text, 'Base Notes')
 
     result = {
         'Perfume Name': perfume_name,
+        'Concentration Type': concentration_type,
         'Tagline': tagline,
         'Poetic Olfactory Description': poetic_desc,
         'Image Description': image_desc,