Update app.py

app.py

@@ -186,11 +186,25 @@ async def gen_title(request: Request):
         return {"title": "New Chat"}
 
 
-def build_prompt(user_input: str, history: List[Dict[str, str]]) -> str:
-    # Qwen 2.5 chat format
+def build_prompt(user_input: str, history: List[Dict[str, str]], has_web_context: bool = False) -> str:
+    # Qwen 2.5 chat format with optional web context awareness
     system = (
-        "You are Hannah 1.0, an intelligent, fast, and helpful pilot assistant. "
-        "Answer efficiently and clearly."
+        "You are Hannah 1.0, an intelligent, fast, and helpful AI assistant. "
+        "Answer clearly and accurately. "
+    )
+    
+    # If web context is available, instruct the model to use it
+    if has_web_context:
+        system += (
+            "You have been provided with fresh web search context in the user's message. "
+            "Use this context to provide current, accurate information about recent events and dates. "
+            "Reference the sources when relevant. "
+        )
+    
+    system += (
+        "Keep responses concise but helpful. "
+        "If asked about your model or training details, simply say: 'I'm Hannah - a helpful AI assistant.' "
+        "Do not discuss GGUF files or internal implementation details."
     )
 
     parts: List[str] = ["<|im_start|>system\n" + system + "<|im_end|>\n"]
@@ -213,14 +227,18 @@ async def chat(request: Request):
     user_input = (data.get("message") or "").strip()
     model_file = data.get("model")
     history = data.get("history") or []
+    has_web = data.get("internet", False)  # Check if web search was enabled
 
     if not user_input:
         raise HTTPException(status_code=400, detail="Empty message")
 
     llm = get_model(model_file)
+    
+    # Detect if the message includes web context
+    has_web_context = has_web and "[Web context retrieved on" in user_input
 
     def iter_response():
-        prompt = build_prompt(user_input, history)
+        prompt = build_prompt(user_input, history, has_web_context=has_web_context)
 
         stream = llm(
             prompt,