commit 000432

app.py

@@ -52,18 +52,19 @@ parser = StrOutputParser()
 
 basic_chain = prompt | llm | parser  # This is a Runnable
 
-# Define a function to decide if we need web search
+# ✅ Enhanced keyword detection
+REAL_TIME_KEYWORDS = {"latest", "current", "news", "today", "price", "time", "live", "trending", "update", "happening"}
+
 def should_search(input_dict):
     question = input_dict.get("question", "").lower()
-    keywords = ["latest", "current", "news", "today", "price", "time", "live", "recent"]
-    return any(k in question for k in keywords)
+    return any(kw in question for kw in REAL_TIME_KEYWORDS)
 
 search_tool = DuckDuckGoSearchRun()
 
-# Chain that runs DuckDuckGo search
+# ✅ Refined search chain returns clean results
 search_chain = (
     RunnableLambda(lambda q: {"question": search_tool.run(q["question"])}) |
-    RunnableLambda(lambda d: f"Search result: {d['question']}")
+    RunnableLambda(lambda d: f"(Live info) {d['question']}")
 )
 
 # Main chain with conditional logic