add new logic

README.md

@@ -21,7 +21,7 @@ An AI-powered system that generates contextually appropriate replies for convers
 - **Model 1 – LLaMA**: Trigger+move augmented dataset fine-tunes a LLaMA-style LLM to respond like a confident "anh"
 - **Model 2 – PhoBERT**: Same pipeline but specialized for the Vietnamese `vinai/phobert-base` backbone (encoder-decoder LoRA)
 - **Model 3 – Wingman LoRA**: Dedicated LoRA head (wingman persona) that still falls back to the prompt-based service if adapters are missing
-- **Wingman Prompt Editor**: The UI shows the current Model 3 prompt and lets you override it live for rapid experimentation
+- **Wingman Prompt Editor**: The UI shows the current Model 3 prompt and lets you override it live (supports `{conversation}`, `{trigger}`, `{move}` placeholders) for rapid experimentation
 - **Automatic Fine-tuning**: All artifacts can be trained automatically on Spaces using `setup_and_finetune.py`
 
 ## 📋 How It Works

app.py

@@ -282,6 +282,7 @@ with gr.Blocks(title=title) as demo:
                     label="Custom Wingman Prompt",
                     value=DEFAULT_WINGMAN_PROMPT,
                     placeholder="Adjust how the wingman persona should respond...",
+                    info="You can reference {conversation}, {trigger}, {move} inside your prompt.",
                 )
                 gr.Markdown("Leave as-is for default behavior. Edits apply to Model 3 when its LoRA is used.")
             reply_btn = gr.Button("Generate Reply Suggestion", variant="primary", size="lg")

finetuned_reply_service.py

@@ -232,12 +232,22 @@ class FinetunedReplyService:
         ]
 
         if self.persona == "wingman":
-            base_prompt.append(
-                custom_prompt
-                or "You are a charming Vietnamese wingman crafting irresistible yet respectful replies for Male. "
-                "Keep replies concise (<35 words), confident, playful, and always from Male's perspective using 'anh' for self and 'em' for partner. "
-                "Embed subtle compliments/flirtation while honoring the identified move."
-            )
+            if custom_prompt:
+                try:
+                    persona_text = custom_prompt.format(
+                        conversation=conversation,
+                        trigger=trigger,
+                        move=move,
+                    )
+                except Exception:
+                    persona_text = custom_prompt
+            else:
+                persona_text = (
+                    "You are a charming Vietnamese wingman crafting irresistible yet respectful replies for Male. "
+                    "Keep replies concise (<35 words), confident, playful, and always from Male's perspective using 'anh' for self and 'em' for partner. "
+                    "Embed subtle compliments/flirtation while honoring the identified move."
+                )
+            base_prompt.append(persona_text)
         else:
             base_prompt.append(
                 "Generate the next appropriate response from Male to Female. The reply should be from Male's perspective, "