Update app.py

app.py

@@ -142,19 +142,23 @@ class SuperSmartAgent:
             return state
 
         def use_web_search(state):
-            query = state["question"].replace("search the web for", "").replace("look up", "").strip()
+            question = state.get("question", "")
+            query = question.lower().replace("search the web for", "").replace("look up", "").strip()
             print(f"Searching DuckDuckGo for: {query}")
         
-            with DDGS() as ddgs:
-                results = ddgs.text(query, max_results=3)
-        
-            if results:
-                snippets = [r['title'] + ": " + r.get('body', '') for r in results]
-                state["response"] = "\n".join(snippets)
-            else:
-                state["response"] = "No results found."
+            try:
+                with DDGS() as ddgs:
+                    results = ddgs.text(query, max_results=3)
         
+                if results:
+                    snippets = [f"{r['title']}: {r.get('body', '')}" for r in results]
+                    state["response"] = "\n".join(snippets)
+                else:
+                    state["response"] = "No results found."
+            except Exception as e:
+                state["response"] = f"Web search error: {e}"
             return state
+
         
         def use_video_analysis(state):
             import re
@@ -221,11 +225,28 @@ class SuperSmartAgent:
         builder.add_node("fallback", fallback)
         
         # Edges
-        ####################################################
-        builder.set_entry_point("check_reversed")
-        builder.add_edge("check_reversed", "fix_question")
+        #####################################################
+        # Entry point: special tool detection first
+        builder.set_entry_point("check_special_tools")
+
+        # Special tool path (web/video) → END
+        builder.add_node("check_special_tools", check_special_tools)
+        builder.add_node("use_web_search", use_web_search)
+        builder.add_node("use_video_analysis", use_video_analysis)
+        
+        builder.add_conditional_edges(
+            "check_special_tools",
+            lambda s: "use_web_search" if s.get("use_tool") == "web"
+            else "use_video_analysis" if s.get("use_tool") == "video"
+            else "check_reversed"
+        )
+        builder.add_edge("use_web_search", END)
+        builder.add_edge("use_video_analysis", END)
         
+        # Reasoning pipeline (for non-tool questions)
+        builder.add_edge("check_reversed", "fix_question")
         builder.add_edge("fix_question", "check_riddle_or_trick")
+        
         builder.add_conditional_edges(
             "check_riddle_or_trick",
             lambda s: "solve_riddle" if s["is_riddle"] else "check_python_suitability"
@@ -238,21 +259,7 @@ class SuperSmartAgent:
         )
         builder.add_edge("generate_code", END)
         builder.add_edge("fallback", END)
-        #####################################################
-        builder.add_node("check_special_tools", check_special_tools)
-        builder.add_node("use_web_search", use_web_search)
-        builder.add_node("use_video_analysis", use_video_analysis)
-        
-        builder.set_entry_point("check_special_tools")
-        builder.add_conditional_edges(
-            "check_special_tools",
-            lambda s: "use_web_search" if s.get("use_tool") == "web"
-            else "use_video_analysis" if s.get("use_tool") == "video"
-            else "check_reversed"
-        )
-        
-        builder.add_edge("use_web_search", END)
-        builder.add_edge("use_video_analysis", END)
+
             
         graph = builder.compile()
         return graph