Update app.py

app.py

@@ -11,35 +11,53 @@ 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):
-        print("BasicAgent initialized.")
+    def __init__(self, debug: bool = False):
         self.search = DuckDuckGoSearchTool()
+        self.debug = debug
+        if self.debug:
+            print("BasicAgent initialized.")
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question: {question}")
+        if self.debug:
+            print(f"Agent received question: {question}")
+
+        # Early validation
+        if not question or not question.strip():
+            return "Please provide a valid question."
 
         try:
-            # Perform the search; returns a list of results
             results = self.search(question)
+            
+            # Use truthiness check and early return
             if not results:
                 return "No results found for that query."
 
-            # Results is a sequence of dicts, assume it has 'title' and 'snippet'
+            # Direct access with get() method chaining
             top = results[0]
-            title = top.get("title", "No title")
+            title = top.get("title") or "No title"
             snippet = top.get("snippet", "").strip()
             link = top.get("link", "")
 
-            # Compose a clean answer
-            answer = f"**{title}**\n{snippet}"
+            # Build answer more efficiently
+            parts = [f"**{title}**"]
+            if snippet:
+                parts.append(snippet)
             if link:
-                answer += f"\nSource: {link}"
+                parts.append(f"Source: {link}")
+            
+            answer = "\n".join(parts)
 
+        except (IndexError, KeyError, AttributeError):
+            # More specific exception handling
+            answer = "Sorry, I couldn't process the search results properly."
         except Exception as e:
             answer = f"Sorry, I couldn't fetch results due to: {e}"
 
-        print(f"Agent returning answer: {answer}")
+        if self.debug:
+            print(f"Agent returning answer: {answer}")
+        
         return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):