Update app.py

app.py

@@ -9,7 +9,8 @@ import string
 
 ###
 
-import wikipedia
+import requests
+from bs4 import BeautifulSoup
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -124,29 +125,38 @@ class SuperSmartAgent:
             return state
 
         ###################
+        def query_wikipedia(question):
+            try:
+                # Try to extract keywords (naively here)
+                search_term = question.strip().replace(" ", "_")
+                url = f"https://en.wikipedia.org/wiki/{search_term}"
+                print(f"Fetching Wikipedia page: {url}")
+                resp = requests.get(url, timeout=10)
+                if resp.status_code != 200:
+                    return f"No Wikipedia page found for '{question}'."
+        
+                soup = BeautifulSoup(resp.text, "html.parser")
+                paragraphs = soup.select("p")
+                for p in paragraphs:
+                    if p.text.strip():
+                        return p.text.strip()
+                return "Wikipedia page found, but no readable content available."
+            except Exception as e:
+                return f"Error querying Wikipedia: {e}"
+
         def check_wikipedia_suitability(state):
             q = state["question"].lower()
-            keywords = ["who is", "what is", "where is", "tell me about", "wikipedia", "when was"]
-            state["is_wiki"] = any(kw in q for kw in keywords)
+            # Simple heuristic: if it's a "who is", "what is", or named entity
+            triggers = ["who is", "what is", "when did", "where is", "tell me about"]
+            state["is_wiki"] = any(trigger in q for trigger in triggers)
+            print(f"is_wiki: {state['is_wiki']}")
             return state
-            
+        
         def search_wikipedia(state):
-            query = state["question"]
-            try:
-                # Strip prompt like "Who is" to get better titles
-                cleaned_query = query.replace("who is", "").replace("what is", "").replace("tell me about", "").strip()
-                result = wikipedia.summary(cleaned_query, sentences=2)
-                state["response"] = result
-            except wikipedia.exceptions.DisambiguationError as e:
-                state["response"] = f"Too many results for '{query}', be more specific. Example options: {e.options[:3]}"
-            except wikipedia.exceptions.PageError:
-                state["response"] = f"Sorry, I couldn't find any info on '{query}'."
-            except Exception as e:
-                state["response"] = f"Wikipedia lookup failed: {str(e)}"
+            question = state["question"]
+            state["response"] = query_wikipedia(question)
             return state
 
-
-
         ##################
         
         class AgentState(TypedDict, total=False):
@@ -159,43 +169,43 @@ class SuperSmartAgent:
         
         builder = StateGraph(AgentState)
 
-         # --- Nodes ---
+        # --- Nodes ---
         builder.add_node("check_reversed", check_reversed)
         builder.add_node("fix_question", fix_question)
         builder.add_node("check_riddle_or_trick", check_riddle_or_trick)
         builder.add_node("solve_riddle", solve_riddle)
-        builder.add_node("check_wikipedia_suitability", check_wikipedia_suitability)  # NEW
-        builder.add_node("search_wikipedia", search_wikipedia)  # NEW
+        builder.add_node("check_wikipedia_suitability", check_wikipedia_suitability)
+        builder.add_node("search_wikipedia", search_wikipedia)
         builder.add_node("check_python_suitability", check_python_suitability)
         builder.add_node("generate_code", generate_code)
         builder.add_node("fallback", fallback)
 
         # --- Entry point ---
         builder.set_entry_point("check_reversed")
-
+        
         # --- Reasoning pipeline ---
         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.get("is_riddle") else "check_wikipedia_suitability"
         )
-
+        
         builder.add_edge("solve_riddle", END)
-
+        
         builder.add_conditional_edges(
             "check_wikipedia_suitability",
             lambda s: "search_wikipedia" if s.get("is_wiki") else "check_python_suitability"
         )
-
+        
         builder.add_edge("search_wikipedia", END)
-
+        
         builder.add_conditional_edges(
             "check_python_suitability",
             lambda s: "generate_code" if s.get("is_python") else "fallback"
         )
-
+        
         builder.add_edge("generate_code", END)
         builder.add_edge("fallback", END)