Refine music prompt generation in app.py. Updated the ENRICH_SYSTEM to clarify JSON response structure and enhance music_prompt template with specific instrument and background noise guidelines. Adjusted temperature setting for improved output consistency and ensured background noise is always included in the music prompt.

app.py

@@ -46,19 +46,28 @@ app = Server(title="LoFinity")
 
 ENRICH_SYSTEM = """\
 You are the creative brain of LoFinity, a magical vending machine that sells
-lofi cassette tapes. The user tells you a vibe; you reply ONLY with JSON:
-{"music_prompt": "...", "title": "..."}
+lofi cassette tapes. The user gives you a vibe. Reply ONLY with JSON with
+exactly these two keys: {"music_prompt": "...", "title": "..."}
 
-music_prompt: one sentence for a music generation model. It MUST begin with
-the exact words "lofi chill". The genre is always lofi chill —
-relaxed, mellow, downtempo — never energetic, never another genre, always
-instrumental with no vocals. After the genre, mention 2-3 soft instruments
-(e.g. rhodes piano, muted guitar, warm bass), a calm mood, a slow tempo
-around 70-85 bpm, and analog textures (vinyl crackle, tape hiss, rain...)
-that fit the user's vibe.
+Build music_prompt from this template, in this order:
+"lofi chill, <instrument 1>, <instrument 2>, <instrument 3>, <background noise>, <mood>, slow tempo, 75 bpm, instrumental"
+
+- instruments: 2-3 picked to EVOKE the user's vibe, never a default set
+  (island -> ukulele, kalimba, steel pan; rainy city -> rhodes piano, soft
+  guitar; winter -> felt piano, soft strings; desert -> slide guitar, hand drums)
+- background noise: exactly one, matched to the vibe: vinyl crackle, tape
+  hiss, soft rain, ocean waves, fireplace crackle, birdsong, night crickets,
+  cafe murmur, or wind in trees
+- mood: one or two calm words; never energetic, no vocals
 
 title: a cozy cassette tape title inspired by the vibe, max 5 words,
-Title Case, no quotes or emoji."""
+Title Case, no quotes or emoji.
+
+Examples:
+user: island summer
+{"music_prompt": "lofi chill, ukulele, kalimba, steel pan, ocean waves in the background, breezy and warm, slow tempo, 75 bpm, instrumental", "title": "Coconut Daydream"}
+user: studying at midnight
+{"music_prompt": "lofi chill, rhodes piano, muted guitar, soft bass, vinyl crackle in the background, focused and calm, slow tempo, 75 bpm, instrumental", "title": "Midnight Study Session"}"""
 
 
 def enrich_prompt(prompt: str) -> tuple[str, str]:
@@ -75,7 +84,7 @@ def enrich_prompt(prompt: str) -> tuple[str, str]:
                 ],
                 "format": "json",
                 "stream": False,
-                "options": {"temperature": 0.9, "num_predict": 220},
+                "options": {"temperature": 0.8, "num_predict": 220},
             },
             timeout=45,
         )
@@ -87,6 +96,13 @@ def enrich_prompt(prompt: str) -> tuple[str, str]:
             # belt and suspenders: the genre must lead even if the LLM drifts
             if "lofi" not in music_prompt.lower():
                 music_prompt = f"lofi chill, {music_prompt}"
+            # …and there must always be a background noise layer
+            noise_words = (
+                "crackle", "hiss", "rain", "waves", "noise", "birdsong",
+                "crickets", "murmur", "wind", "fireplace", "static",
+            )
+            if not any(w in music_prompt.lower() for w in noise_words):
+                music_prompt += ", soft vinyl crackle in the background"
             return music_prompt, title
     except Exception as e:  # noqa: BLE001 — any failure means "use fallback"
         print(f"[lofinity] ollama enrichment failed ({e!r}), using fallback")