Update main.py

main.py

@@ -47,14 +47,20 @@ try:
 except Exception as e:
     print(f"Failed to initialize local fallback LLM: {e}")
 
+import datetime
+
 async def evaluate_needs_search(query: str) -> bool:
     """Uses a fast, small model to determine if the query requires real-time data."""
     if not GROQ_API_KEY:
         return False
         
-    system_prompt = """You are a highly efficient classification router.
+    current_date = datetime.datetime.now().strftime("%B %d, %Y")
+    
+    system_prompt = f"""You are a highly efficient classification router.
+Today's date is {current_date}.
 Determine if the user's query requires up-to-date, real-time information or current events data from the internet to answer accurately.
-Respond ONLY with "YES" if it requires search, or "NO" if it can be answered with general knowledge up to 2023.
+If the user asks about an event, person, software, or fact that changes frequently or occurred near or after {current_date}, it requires search.
+Respond ONLY with "YES" if it requires search, or "NO" if it can be answered with general, static knowledge.
 DO NOT provide any other text."""
 
     try:
@@ -81,22 +87,32 @@ DO NOT provide any other text."""
         print(f"Routing evaluation error: {e}")
         return False # Default to no search on error to save latency
 
-def perform_search(query: str, max_results: int = 3) -> str:
-    """Performs a web search using DuckDuckGo."""
-    try:
-        with DDGS() as ddgs:
-            results = list(ddgs.text(query, max_results=max_results))
-            
-        if not results:
-            return "No recent information found."
-            
-        context = "Here is some current information I found on the web regarding the user's query:\n\n"
-        for i, r in enumerate(results):
-            context += f"Source {i+1} [{r.get('title', 'No Title')}]: {r.get('body', '')}\n"
-        return context
-    except Exception as e:
-        print(f"Search error: {e}")
-        return "Search failed or was blocked."
+import time
+
+def perform_search(query: str, max_results: int = 4) -> str:
+    """Performs a web search using DuckDuckGo with basic retry logic."""
+    for attempt in range(2):
+        try:
+            with DDGS() as ddgs:
+                results = list(ddgs.text(query, max_results=max_results))
+                
+            if not results:
+                if attempt == 0:
+                    time.sleep(1) # short backoff
+                    continue
+                return "Search returned no direct results. Please rely on your internal knowledge."
+                
+            context = "Here is real-time web search data regarding the user's query:\n\n"
+            for i, r in enumerate(results):
+                context += f"Source {i+1} [{r.get('title', 'Unknown Title')}]: {r.get('body', '')}\nURL: {r.get('href', 'N/A')}\n\n"
+            return context
+        except Exception as e:
+            print(f"Search attempt {attempt + 1} failed: {e}")
+            if attempt == 0:
+                time.sleep(1)
+            else:
+                return f"Search engine was temporarily unavailable ({e}). Please rely on your internal knowledge instead."
+    return ""
 
 @app.post("/chat/completions")
 async def situation_aware_chat(request: ChatRequest):
@@ -114,9 +130,11 @@ async def situation_aware_chat(request: ChatRequest):
         print("Search complete.")
 
     # 2. Prepare the final prompt
-    system_base = "You are 'Situation Aware AI', an advanced assistant integrated into the Edyx platform."
+    current_time_str = datetime.datetime.now().strftime("%A, %B %d, %Y %I:%M %p")
+    system_base = f"You are 'Situation Aware AI', an advanced assistant integrated into the Edyx platform. Today's current date and time is {current_time_str}. Use this date as your absolute frame of reference for 'now'."
+    
     if context_injection:
-        system_base += "\n\nThe user has asked a question that requires current knowledge. You have been provided with real-time web search results below. Incorporate this information seamlessly into your answer to provide the most up-to-date and accurate response. Do not mention that you 'searched the web' unless asked, just present the facts.\n\n" + context_injection
+        system_base += f"\n\nThe user has asked a question that requires current knowledge. You have been provided with real-time web search results below. Synthesize a comprehensive, highly accurate, state-of-the-art response using ONLY the provided facts. Cite your sources naturally in your response.\n\n--- WEB SEARCH RESULTS ---\n{context_injection}\n--- END RESULTS ---"
 
     # Construct message array preserving history
     final_messages = [{"role": "system", "content": system_base}]