Improved search..

app.py

@@ -342,6 +342,18 @@ def chat(request: ChatRequest, req: Request):
 
         # Step 6: Stream LLM response via LangGraph
         has_image = bool(request.image)
+
+        # Resolve target model dynamically to print/trace and pass to LangGraph config
+        try:
+            from graph import _classify, _pick
+            category = _classify(clean_message)
+            target_model, _ = _pick(category, has_image=has_image)
+        except Exception as e:
+            print(f"[MODEL PICK ERROR] {e}", flush=True)
+            target_model = ""
+
+        print(f"[ROUTER CHAT] Routing query to model: {target_model}", flush=True)
+
         config = {
             "configurable": {
                 "thread_id":       request.thread_id,
@@ -352,6 +364,8 @@ def chat(request: ChatRequest, req: Request):
                 "student_profile": student_profile,
                 "user_api_key":    user_api_key,
                 "has_image":       has_image,
+                "search_enabled":  (web_search_enabled or yt_search_enabled),
+                "model":           target_model,
             }
         }
 

graph.py

@@ -305,6 +305,7 @@ def chat_node(state: ChatState, config: RunnableConfig):
     profile   = cfg.get("student_profile", "")
     api_key   = cfg.get("user_api_key", "")
     override  = cfg.get("model", "")
+    search_enabled = cfg.get("search_enabled", False)
     has_image = cfg.get("has_image", False)
 
     user_text = _extract_text(state["messages"])
@@ -313,6 +314,15 @@ def chat_node(state: ChatState, config: RunnableConfig):
     messages = state["messages"]
     base_prompt = prompts.build(persona, context, language, username, profile)
 
+    if search_enabled:
+        base_prompt += (
+            "\n\nSCOPE OVERRIDE (ACTIVE SEARCH/YT DETECTED):\n"
+            "The user has enabled Web Search / YouTube Video Search. You are acting as a general information assistant. "
+            "The strict Physics, Chemistry, and Mathematics scope restriction is temporarily suspended. "
+            "Answer the user's query comprehensively using the web search or YouTube transcript context provided, "
+            "regardless of the academic or non-academic nature of the query. Do NOT output a scope refusal message.\n"
+        )
+
     if has_image:
         base_prompt += (
             "\n\nIMAGE HANDLING:\n"

static/app.js

@@ -1401,7 +1401,7 @@ function _updateInputPlaceholder() {
     if (isWebSearchEnabled) {
         placeholder = 'Search the web and ask...';
     } else if (isYtSearchEnabled) {
-        placeholder = 'Paste YouTube URL or ask about video...';
+        placeholder = 'Paste YouTube link...';
     }
 
     const inputs = [
@@ -1463,11 +1463,9 @@ function setupAttachMenu(plusBtn, menuEl, webSearchToggleEl, ytSearchToggleEl, a
     }
 
     if (docInputEl) {
-        const newDocInputEl = docInputEl.cloneNode(true);
-        docInputEl.parentNode.replaceChild(newDocInputEl, docInputEl);
-        newDocInputEl.addEventListener('change', () => {
-            if (newDocInputEl.files[0]) {
-                handleAttachedFile(newDocInputEl.files[0], previewEl);
+        docInputEl.addEventListener('change', () => {
+            if (docInputEl.files[0]) {
+                handleAttachedFile(docInputEl.files[0], previewEl);
             }
         });
     }

tools.py

@@ -2,10 +2,10 @@ from __future__ import annotations
 import re
 from langchain_community.tools import DuckDuckGoSearchResults
 from langchain_core.tools import tool
+from duckduckgo_search import DDGS
+from youtube_transcript_api import YouTubeTranscriptApi
 
 # ── Web Search ────────────────────────────────────────────────
-# Using DuckDuckGoSearchResults (structured) over DuckDuckGoSearchRun (plain string).
-# num_results=4 keeps context tight for free-tier models.
 web_search = DuckDuckGoSearchResults(
     name="web_search",
     num_results=4,
@@ -20,11 +20,9 @@ web_search = DuckDuckGoSearchResults(
 
 def run_web_search(query: str) -> str:
     """
-    Run web search programmatically using DDGS with a LangChain fallback.
-    Catches all exceptions to ensure search failures never crash the chat.
+    Run web search programmatically using DDGS.
     """
     try:
-        from duckduckgo_search import DDGS
         with DDGS(timeout=10) as ddgs:
             results = list(ddgs.text(query, max_results=4))
             if not results:
@@ -38,9 +36,8 @@ def run_web_search(query: str) -> str:
                 formatted.append(f"Title: {title}\nURL: {href}\nSnippet: {body}\n")
             return "\n".join(formatted)
     except Exception as e:
-        print(f"[SEARCH DDGS ERROR] {e}. Trying LangChain fallback...")
+        print(f"[SEARCH DDGS ERROR] {e}. Trying LangChain fallback...", flush=True)
         try:
-            # Fallback to the LangChain wrapper tool
             return web_search.invoke(query)
         except Exception as fallback_err:
             return f"Web search is temporarily unavailable. (Error: {fallback_err})"
@@ -64,13 +61,6 @@ def _extract_video_id(url_or_id: str) -> str | None:
 def yt_transcript(youtube_url: str) -> str:
     """
     Fetch the full transcript of a YouTube video.
-
-    Args:
-        youtube_url: A YouTube video URL (e.g. https://www.youtube.com/watch?v=XYZ)
-                     or a bare 11-character video ID.
-
-    Returns:
-        The full transcript text, or an error message if unavailable.
     """
     return fetch_yt_transcript(youtube_url)
 
@@ -78,21 +68,19 @@ def yt_transcript(youtube_url: str) -> str:
 def fetch_yt_transcript(youtube_url: str) -> str:
     """
     Programmatic helper to fetch the transcript of a YouTube video URL or ID.
-    Handles exceptions gracefully to prevent crashes.
     """
     video_id = _extract_video_id(youtube_url)
     if not video_id:
         return f"Could not extract a valid YouTube video ID from: {youtube_url}"
 
     try:
-        from youtube_transcript_api import YouTubeTranscriptApi
         transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
         transcript = " ".join(seg["text"] for seg in transcript_list)
         if not transcript.strip():
             return "Transcript is empty for this video."
         return transcript
-
     except Exception as exc:
+        print(f"[YT TRANSCRIPT EXCEPTION] Video ID: {video_id}, Error: {exc}", flush=True)
         err = str(exc).lower()
         if "disabled" in err or "no transcript" in err:
             return (