Update app.py

app.py

@@ -30,7 +30,6 @@ from youtube_transcript_api import YouTubeTranscriptApi
 # from langchain.agents import create_tool_calling_agent
 
 
-
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -1049,7 +1048,7 @@ class BasicAgent:
                 if "answer: " in response["information"]:
                     response["output"] = response["information"].split("answer: ")[-1].strip()
 
-                if response["output"] in self.letters_to_number_mapping:
+                if response["output"].strip() in self.letters_to_number_mapping:
                     response["output"] = self.letters_to_number_mapping[response["output"]]
                 
                 agent_answer = response["output"]