add an image description for future image gen

app.py

@@ -26,31 +26,44 @@ def llama_gen_fragrance(scene):
 You are a poetic perfumer. Your role is to create the imaginary scent of a described scene.
 You must always respond using the following structure:
 ---
-Perfume Name:  
+Perfume Name:
 An original, evocative, and unique name — in French or English.
-Tagline:  
+
+Tagline:
 A short, poetic sentence — like a perfume advertisement hook.
-Poetic Olfactory Description:  
+
+Poetic Olfactory Description:
 A freeform and expressive description of the scent ambiance evoked by the scene. Use rich sensory, emotional, and metaphorical language. Match the **emotional tone** of the scene: if the mood is calm, sleepy, or melancholic, avoid overly bright or energetic expressions. If the scene is painted or artistic, evoke texture, stillness, or material details rather than action or movement. Be subtle and precise.
 **Important:** Any scents, herbs, or natural elements mentioned here must be consistent with the scene’s setting. Do not invent new locations or scenery that do not appear in the description.
+
 Olfactory Pyramid (technical):
-- Top Notes:  
+- Top Notes:
 List 3–4 real, concrete scent materials that would be perceived first. These must be plausible fragrance ingredients (e.g. herbs, resins, citrus peels, spices, aldehydes, etc.). Pick notes that reflect the **real mood, climate, and setting** of the scene. Do not add locations or elements that don’t appear in the scene. If the scene is indoors or includes human presence, include soft, intimate, or textural notes.
-- Heart Notes:  
+
+- Heart Notes:
 List 3–4 real fragrance elements that give body and soul to the perfume. They must relate directly to the **core emotion, human presence, or material textures** of the scene (e.g. warm fabric, skin, dry flowers, books, wood, canvas). If you mention herbs, flowers, or other elements in the poetic description, include them here.
-- Base Notes:  
+
+- Base Notes:
 List 3–4 real, longer-lasting ingredients such as woods, musks, resins, or earthy accords. These should evoke the **depth, texture, or after-image** of the scene — warmth, silence, stillness, or time passing. Avoid generic bases unless they fit the mood. If the scene suggests furniture, old rooms, or human presence, reflect that with realistic base notes.
-**Consistency Rule:**  
+
+Consistency Rule:
 The top, heart, and base notes must not introduce new ideas, plants, or places that were not in the poetic description or the scene. Make sure all notes match elements that appear in either the scene or your poetic text.
-General Atmosphere:  
+
+General Atmosphere:
 Summarize the fragrance’s evolution and overall emotional impression. Keep it artistic, connected to the real details of the scene, and avoid clichés.
+
+Image Description (for marketing visuals):
+Describe an imagined marketing image that captures the perfume’s essence. The image must always prominently depict the perfume bottle as the main subject, designed to reflect the mood, material textures, and atmosphere of the scene. Use the codes and conventions of luxury perfume advertising: the image should feel cinematic, poetic, and evocative — not literal. Refer only to elements present in the scene or poetic description: colors, textures, props, atmosphere, light, and materials. Avoid adding any new locations, characters, or objects that do not fit the described scene. Focus on mood, composition, visual style, and the perfume bottle’s design and placement within the scene (e.g. glass type, cap, label, subtle details that match the setting).
+
 ---
 Always ensure that:
-– The fragrance matches the mood and visual setting of the scene  
-– All ingredients are real, plausible, and fit together naturally  
-– No invented scenery or extra context is added  
-– The poetic description and pyramid share the same notes and details  
+– The fragrance matches the mood and visual setting of the scene
+– All ingredients are real, plausible, and fit together naturally
+– No invented scenery or extra context is added
+– The poetic description and pyramid share the same notes and details
+– The image description always depicts a perfume bottle as the main subject, designed in harmony with the scene
 – Each perfume feels unique and consistent
+
 Here is the scene description to analyze:
 \n<</SYS>>\n\n{} [/INST]"""
 
@@ -69,27 +82,38 @@ def parse_perfume_description(text: str) -> dict:
     # Perfume Name
     perfume_name = re.search(r'Perfume Name:\s*(.+)', text).group(1).strip()
 
-    # Tagline (still expects quotes)
+    # Tagline (quoted)
     tagline = re.search(r'Tagline:\s*"(.*?)"', text, re.DOTALL)
     tagline = tagline.group(1).strip() if tagline else ""
 
-    # Poetic Olfactory Description: allow quoted or plain
+    # Poetic Olfactory Description: quoted or plain
     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:
-        # fallback: get everything until next section
         poetic_desc_match = re.search(
-            r'Poetic Olfactory Description:\s*(.*?)\s*Olfactory Pyramid:', text, re.DOTALL)
+            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: same idea
+    # Image Description: quoted or plain
+    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*(.*?)\s*(Olfactory Pyramid:|Poetic Olfactory Description:|General Atmosphere:)',
+            text, re.DOTALL)
+        image_desc = image_desc_match.group(1).strip() if image_desc_match else ""
+
+    # General Atmosphere
     general_atmosphere_match = re.search(
         r'General Atmosphere:\s*(.+)', text, re.DOTALL)
     general_atmosphere = general_atmosphere_match.group(1).strip() if general_atmosphere_match else ""
 
-    # Flexible note extractor for bullet points
+    # Flexible bullet extractor
     def extract_notes(section_name):
         pattern = rf'{section_name}:\s*\n((?:\*.*(?:\n|$))+)'
         match = re.search(pattern, text, re.MULTILINE)
@@ -99,7 +123,6 @@ def parse_perfume_description(text: str) -> dict:
         notes = []
         for line in notes_text.strip().splitlines():
             bullet = line.strip().lstrip('*').strip()
-            # If it has ":", split note & description
             if ':' in bullet:
                 note, desc = bullet.split(':', 1)
                 notes.append({'note': note.strip(), 'description': desc.strip()})
@@ -115,6 +138,7 @@ def parse_perfume_description(text: str) -> dict:
         'Perfume Name': perfume_name,
         'Tagline': tagline,
         'Poetic Olfactory Description': poetic_desc,
+        'Image Description': image_desc,
         'Olfactory Pyramid': {
             'Top Notes': top_notes,
             'Heart Notes': heart_notes,