Update app.py

app.py

@@ -3,13 +3,22 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from agent import BasicAgent
-
+from agent import build_graph
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-
+# --- Basic Agent Definition ---
+# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+class BasicAgent:
+    def __init__(self):
+        # graph factory lives here
+        self.graph = build_graph()
+        
+    def __call__(self, question: str) -> str:
+        # runs graph on each question
+        answer = self.graph.run(question)
+        return enswer.replace("FINAL ANSWER:","",1).strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """