Update app.py

app.py

@@ -1,3 +1,4 @@
+# app.py
 import requests
 import os
 import logging
@@ -364,6 +365,8 @@ def route_model(messages, task_type, files=None, search_web=False, jurisdiction=
         try:
             # Build system prompt contextual
             system_content = build_grok_prompt(prompt, task_type, jurisdiction, rag_context)  # But since messages have it, prepend if not
+            if 'CourtListener' in rag_context:
+                system_content += "\nPrioritize CourtListener snippets for accuracy: Quote key snippets, cite cases, and polish into a structured response (e.g., IRAC format for analysis tasks)."
             if messages[0]['role'] != 'system':
                 messages = [{'role': 'system', 'content': system_content}] + messages
             stream_grok = ask_grok(messages, stream=True)