Update app.py

app.py

@@ -80,7 +80,7 @@ def load_pipeline(model_name):
     Tries bfloat16, falls back to float16 or float32 if unsupported.
     """
     global PIPELINES
-    if model_name in PIPELINES:
+    if model_name in PIPELINES.keys():
         return PIPELINES[model_name]
     repo = MODELS[model_name]["repo_id"]
     tokenizer = AutoTokenizer.from_pretrained(repo)
@@ -148,13 +148,6 @@ def chat_response(user_msg, chat_history, system_prompt,
         debug = 'Web search disabled.'
 
     try:
-
-        # merge any fetched search results into the system prompt
-        if search_results:
-            enriched = system_prompt.strip() + "\n\nRelevant context:\n" + "\n".join(search_results)
-        else:
-            enriched = system_prompt
-
         # wait up to 1s for snippets, then replace debug with them
         if enable_search:
             thread_search.join(timeout=float(search_timeout))
@@ -165,16 +158,18 @@ def chat_response(user_msg, chat_history, system_prompt,
             else:
                 debug = "*No web search results found.*"
 
-        # merge fetched snippets into the system prompt
-        if search_results:
-            enriched = system_prompt.strip() + "\n\nRelevant context:\n" + "\n".join(search_results)
+            # merge fetched snippets into the system prompt
+            if search_results:
+                enriched = system_prompt.strip() + "\n\nRelevant context:\n" + "\n".join(search_results)
+            else:
+                enriched = system_prompt
         else:
             enriched = system_prompt
 
         pipe = load_pipeline(model_name)
         prompt = format_conversation(history, enriched, pipe["tokenizer"])
         prompt_debug = f"\n\n--- Prompt Preview ---\n```\n{prompt}\n```"
-        streamer = TextIteratorStreamer(pipe["tokenizer"],
+        streamer = TextIterStreamer(pipe["tokenizer"],
                                         skip_prompt=True,
                                         skip_special_tokens=True)
         generation_config = dict(