Update app.py

app.py

@@ -29,19 +29,24 @@ def db_agent(query: str) -> str:
             return "No transactions found for today."
         return None
     except sqlite3.OperationalError as e:
-        # Handle missing table or other DB errors
         return f"Database error: {e}. Please initialize 'transactions' table in shop.db."
 
 
 def web_search_agent(query: str) -> str:
-    # Example using SerpAPI
-    resp = requests.get(
-        "https://serpapi.com/search",
-        params={"q": query, "api_key": os.getenv("SERPAPI_KEY")}  
-    )
-    data = resp.json()
-    snippet = data.get("organic_results", [{}])[0].get("snippet", "")
-    return llm_agent(f"Summarize: {snippet}")
+    # Try fetching a snippet from SerpAPI, otherwise fallback to direct LLM
+    try:
+        resp = requests.get(
+            "https://serpapi.com/search",
+            params={"q": query, "api_key": os.getenv("SERPAPI_KEY")}  
+        )
+        data = resp.json()
+        snippet = data.get("organic_results", [{}])[0].get("snippet", "").strip()
+        if snippet:
+            return llm_agent(f"Summarize: {snippet}")
+    except Exception:
+        pass
+    # Fallback for no snippet or errors
+    return llm_agent(query)
 
 
 def llm_agent(prompt: str) -> str: