Update app.py

app.py

@@ -2,26 +2,24 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from langchain_core.messages import HumanMessage
-from agent import build_graph
+from transformers import pipeline
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
-cclass BasicAgent:
-    """A langgraph agent."""
+class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
-        self.graph = build_graph()
+        print("Loading FLAN-T5 model...")
+        self.model = pipeline("text2text-generation", model="google/flan-t5-base", max_length=256)
+        print("Model loaded.")
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # Wrap the question in a HumanMessage from langchain_core
-        messages = [HumanMessage(content=question)]
-        messages = self.graph.invoke({"messages": messages})
-        answer = messages['messages'][-1].content
-        return answer[14:]
+        prompt = f"Answer the following question accurately and concisely:\n{question.strip()}"
+        result = self.model(prompt)
+        answer = result[0]['generated_text'].strip()
+        print(f"Answer generated: {answer}")
+        return answer
 
 # --- Helper Functions ---
 def fetch_questions(api_url: str):