Update app.py

app.py

@@ -3,13 +3,13 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-
 from langgraph.graph import StateGraph, END
 from typing import TypedDict
 import string
 
+###
 
-
+import wikipedia
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -124,7 +124,26 @@ class SuperSmartAgent:
             return state
 
         ###################
-        
+        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)
+            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)}"
+            return state
 
 
 
@@ -140,11 +159,13 @@ 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_python_suitability", check_python_suitability)
         builder.add_node("generate_code", generate_code)
         builder.add_node("fallback", fallback)
@@ -158,17 +179,27 @@ class SuperSmartAgent:
 
         builder.add_conditional_edges(
             "check_riddle_or_trick",
-            lambda s: "solve_riddle" if s.get("is_riddle") else "check_python_suitability"
+            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)
 
+
         
   
         graph = builder.compile()