Update app.py

app.py

@@ -9,7 +9,7 @@ import re
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Enhanced Agent ---
-class EnhancedAgent: 
+class EnhancedAgent:  
     def __init__(self):
         print("Initializing Enhanced Agent...")
         
@@ -25,7 +25,7 @@ class EnhancedAgent:
             from duckduckgo_search import DDGS
             self.search = DDGS()
             print("✅ Search ready")
-        except:
+        except: 
             self.search = None
             print("⚠️ Search unavailable")
     
@@ -34,7 +34,7 @@ class EnhancedAgent:
         if not self.search:
             return ""
         
-        try:
+        try: 
             results = list(self.search.text(query, max_results=max_results))
             if not results:
                 return ""
@@ -58,24 +58,24 @@ class EnhancedAgent:
         patterns_to_remove = [
             r"^according to.*? [,:]",
             r"^based on.*?[,:]",
-            r"^the answer is: ? ",
-            r"^answer: ?",
-            r"^final answer:?",
-            r"^in summary:?",
-            r"^therefore:?",
-            r"^thus:?",
-            r"^so:?",
+            r"^the answer is: ?  ? ",
+            r"^answer: ? ?",
+            r"^final answer:? ?",
+            r"^in summary:? ?",
+            r"^therefore:? ?",
+            r"^thus:? ?",
+            r"^so:? ?",
         ]
         
         cleaned = text. strip()
-        for pattern in patterns_to_remove:
+        for pattern in patterns_to_remove: 
             cleaned = re.sub(pattern, "", cleaned, flags=re.IGNORECASE).strip()
         
         # If answer has multiple sentences, often the last one is the direct answer
         sentences = cleaned.split('.')
         if len(sentences) > 2:
             # Check if last sentence looks like a direct answer
-            last = sentences[-1]. strip()
+            last = sentences[-1].strip()
             if last and len(last) < 100:
                 return last
         
@@ -86,7 +86,7 @@ class EnhancedAgent:
         print(f"Q: {question[: 150]}")
         
         # Determine if we need search
-        needs_search = any(keyword in question.lower() for keyword in [
+        needs_search = any(keyword in question. lower() for keyword in [
             'current', 'latest', 'recent', 'today', 'now', '2024', '2025', '2026',
             'who is', 'what is', 'where is', 'when did', 'how many'
         ])
@@ -99,7 +99,7 @@ class EnhancedAgent:
                 print(f"✅ Search:  {len(search_context)} chars")
         
         # Enhanced system prompt with better instructions
-        system_prompt = """You are an expert AI that provides accurate, direct answers. 
+        system_prompt = """You are an expert AI that provides accurate, direct answers.  
 
 CRITICAL RULES:
 1. Give ONLY the final answer - no explanations unless asked
@@ -119,7 +119,7 @@ Q: "Who is the CEO of Tesla?" → A: "Elon Musk"
 Q: "What is 15 + 27?" → A: "42"
 Q: "How many planets in the solar system?" → A: "8"
 
-DO NOT start with "The answer is" or "According to" - just give the answer directly! """
+DO NOT start with "The answer is" or "According to" - just give the answer directly!"""
 
         # Build prompt
         messages = [{"role": "system", "content":  system_prompt}]
@@ -146,7 +146,7 @@ DO NOT start with "The answer is" or "According to" - just give the answer direc
                     temperature=0.05  # Very low for maximum accuracy
                 )
                 
-                raw_answer = response.choices[0].message.content.strip()
+                raw_answer = response.choices[0].message.content. strip()
                 
                 # Clean up the answer
                 answer = self.extract_answer(raw_answer)
@@ -160,7 +160,7 @@ DO NOT start with "The answer is" or "According to" - just give the answer direc
                             answer = line
                             break
                 
-                print(f"✅ A: {answer[: 200]}")
+                print(f"✅ A: {answer[:200]}")
                 return answer
                 
             except Exception as e:
@@ -169,14 +169,14 @@ DO NOT start with "The answer is" or "According to" - just give the answer direc
                     # Last resort: try a simpler call
                     try:
                         simple_prompt = f"Answer this question concisely:\n\n{question}\n\nAnswer:"
-                        response = self. client.text_generation(
+                        response = self.client.text_generation(
                             simple_prompt,
                             model="mistralai/Mixtral-8x7B-Instruct-v0.1",
                             max_new_tokens=200,
                             temperature=0.1
                         )
                         return response.strip()
-                    except: 
+                    except:
                         return "Unable to generate answer."
         
         return "Unable to generate answer."
@@ -229,7 +229,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         task_id = item.get("task_id")
         question_text = item. get("question")
         
-        if not task_id or not question_text:
+        if not task_id or not question_text: 
             continue
 
         print(f"[{idx}/{total}] {task_id[: 8]}...")
@@ -247,7 +247,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
                 "task_id": task_id,
                 "submitted_answer": "Error processing question"
             })
-            results_log.append((idx, question_text[: 60], f"Error: {str(e)[: 50]}"))
+            results_log.append((idx, question_text[: 60], f"Error: {str(e)[:50]}"))
     
     print(f"\n{'='*70}")
     print(f"✅ Processed all {len(answers_payload)} questions")
@@ -284,7 +284,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
 ### 🏆 Score: **{score}%**
 
-{'### 🎓 Amazing!  You completed Unit 4 of the Hugging Face Agents Course!' if passed else f'### 📈 You got {score}% - need 30% to pass.  The agent is now much better, try again! '}
+{'### 🎓 Amazing! You completed Unit 4 of the Hugging Face Agents Course!' if passed else f'### 📈 You got {score}% - need 30% to pass.  The agent is now much better, try again!'}
 
 **Details:**
 - 👤 User: `{username}`
@@ -330,12 +330,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     submit_button = gr.Button(
         "🚀 Run Evaluation & Submit", 
         variant="primary", 
-        size="lg",
-        scale=2
+        size="lg"
     )
     
     output_text = gr.Markdown()
-    output_table = gr.Dataframe(label="📝 Results Preview", wrap=True, height=400)
+    output_table = gr.Dataframe(label="📝 Results Preview", wrap=True)
 
     submit_button.click(
         run_and_submit_all,