Update app.py

app.py

@@ -3,78 +3,29 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from agents import create_agent_flow
+from langchain_core.messages import HumanMessage
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-
-class WikipediaSearchTool:
-    def search(self, query: str) -> str:
-        if "Mercedes Sosa" in query:
-            return """Between 2000 and 2009, Mercedes Sosa released the following studio albums:
-            - Corazón Libre (2005)
-            - Cantora 1 (2009)
-            - Cantora 2 (2009)
-            """
-        return "No information found."
-
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
+    
     def __init__(self):
-        self.wikipedia_tool = WikipediaSearchTool()
         print("BasicAgent initialized.")
-
+        self.agent = create_agent_flow()
+    
     def __call__(self, question: str) -> str:
-        print(f"Agent received question: {question}")
-
-        if "studio albums" in question and "Mercedes Sosa" in question:
-            wiki_text = self.wikipedia_tool.search("Mercedes Sosa studio albums between 2000 and 2009")
-            album_list = self.extract_albums(wiki_text)
-            album_count = len(album_list)
-            return str(album_count)
-        elif "Featured Article" in question and "November 2016" in question:
-            return str("FunkMonk")
-        elif "table defining" in question:
-            return str("b,e")
-        elif "1htKBjuUWec" in question:
-            return str("Extremely")
-        elif "CK-12 license" in question:
-            return str("Louvrier")
-        elif "L1vXCYZAYYM" in question:
-            return str(3)
-        elif "tfel" in question:
-            return str("right")
-        elif "grocery list" in question:
-            return str("broccoli, celery, fresh basil, lettuce, sweet potatoes")
-        elif "Yankee " in question:
-            return str(519)
-        elif "Carolyn Collins Petersen" in question:
-            return str("80GSFC21M0002")
-        elif "Vietnamese specimens" in question:
-            return str("Saint Petersburg")
-        elif "CK-12 license" in question:
-            return str("Louvrier")
-        elif "Everybody Loves Raymond" in question:
-            return str("Wojciech")
-        elif "Homework.mp3" in question:
-            return str("132, 133, 134, 197, 245")
-        elif "fast-food chain" in question:
-            return str(89706.00)
-        elif "Olympics" in question:
-            return str("CUB")
-        elif "pitchers" in question and "Taishō Tamai" in question:
-            return str("Yoshida, Uehara")
-        elif "Malko Competition" in question:
-            return str("Dmitry")
-        else:
-            return "This is a default answer."
-
-    def extract_albums(self, wiki_text: str) -> list:
-        lines = wiki_text.split("\n")
-        albums = [line.strip() for line in lines if "-" in line]
-        return albums
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+      
+        question = [HumanMessage(content=question)]
+        question_ask = self.agent.invoke({"messages": question})
+        response = question_ask['messages'][-1].content
+        print(f"Agent returning fixed answer: {response}")
+        return response
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -248,4 +199,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)