Update app.py

app.py

@@ -91,17 +91,26 @@ def format_description(raw_description, do_format=True):
     if not do_format:
         return raw_description
         
+    # Extract defaults from the raw description or use fallbacks
+    defaults = {
+        "gender": "woman" if "woman" in raw_description.lower() else "man",
+        "pitch": "moderate pitch",
+        "speed": "slowly",
+        "environment": "close-sounding and clear",
+        "delivery": "with monotone delivery"
+    }
+    
     messages = [{
         "role": "user", 
-        "content": f"""Format this voice description exactly as:
-"a [gender] with a [pitch] voice speaks [speed] in a [environment], [delivery style]"
+        "content": f"""Format this voice description and fill in any missing parameters with defaults:
+"a [gender] with a [pitch] voice speaks [speed] in a [environment], [delivery]"
 
-Required format:
-- gender must be: man/woman
-- pitch must be: slightly low-pitched/moderate pitch/high-pitched  
-- speed must be: slowly/moderately/quickly
-- environment must be: close-sounding and clear/distant-sounding and noisy
-- delivery style must be: with monotone delivery/with animated delivery
+Required parameters (use these exact terms):
+- gender: {defaults['gender']} if not specified
+- pitch: {defaults['pitch']} if not specified
+- speed: {defaults['speed']} if not specified
+- environment: {defaults['environment']} if not specified
+- delivery: {defaults['delivery']} if not specified
 
 Input: {raw_description}
 
@@ -111,7 +120,6 @@ Return only the formatted description, nothing else."""
     outputs = pipe(messages, max_new_tokens=100)
     formatted = outputs[0]["generated_text"][-1]["content"].strip()
     
-    # Validate and extract formatted description
     if "a woman" in formatted.lower() or "a man" in formatted.lower():
         return formatted
     return raw_description