improve query rewriter with legal examples

ui/app.py

@@ -154,7 +154,18 @@ def rewrite_query(user_input: str) -> str:
         messages=[
             {
                 "role": "system",
-                "content": "You are a Kenyan legal search query optimizer. Convert the user's legal question into a concise 5-10 word search query focused on the core legal concept, statute, or procedure. Return ONLY the search query, nothing else. No punctuation, no explanation."
+                "content": """You are a Kenyan legal search query optimizer.
+Convert the user's legal question into a concise 5-10 word search query.
+Focus on: the specific legal document type, the governing Act name, and the legal concept.
+Examples:
+- 'draft a plaint for breach of contract' → 'plaint institution of suit Civil Procedure Act Order 4'
+- 'how do I file at ELRC for dismissal' → 'unfair termination Employment Act ELRC claim'
+- 'bail application for murder charge' → 'bail application criminal charge Criminal Procedure Code'
+- 'petition for divorce' → 'divorce petition Marriage Act matrimonial cause'
+- 'landlord wants to evict my client' → 'eviction notice landlord tenant Rent Restriction Act'
+- 'how to file a succession matter' → 'grant of probate succession Law of Succession Act'
+- 'company director breach of fiduciary duty' → 'director fiduciary duty breach Companies Act'
+Return ONLY the search query, nothing else."""
             },
             {"role": "user", "content": user_input}
         ],