Update app.py

app.py

@@ -180,6 +180,33 @@ def parse_perfume_description(text: str) -> dict:
 
     return result
 
+def extract_image_description(response_json: str) -> str:
+    """
+    Extracts the Image Description value from a JSON response.
+    
+    Args:
+        response_json (str): The JSON string returned by your LLM.
+
+    Returns:
+        str: The Image Description text.
+    """
+    try:
+        data = json.loads(response_json)
+    except json.JSONDecodeError:
+        raise ValueError("Invalid JSON string")
+
+    # The key might be 'Image Description' or 'image_description'
+    image_description = (
+        data.get("Image Description")
+        or data.get("image_description")
+        or None
+    )
+
+    if not image_description:
+        raise KeyError("No 'Image Description' field found in the JSON")
+
+    return image_description.strip()
+
 def get_text_after_colon(input_text):
     # Find the first occurrence of ":"
     colon_index = input_text.find(":")
@@ -214,6 +241,9 @@ def infer(image_input):
 
     parsed = parse_perfume_description(result)
 
+    image_desc = extract_image_description(parsed)
+    print(image_desc)
+
     return result, parsed
 
 css="""