Update app.py

app.py

@@ -7,13 +7,8 @@ import gradio as gr
 # CONFIG
 # ===============================
 
-# Hardcoded GAIA API URL
-BASE_URL = "https://agents-course-unit4-scoring.hf.space"
-
-# Hugging Face username
-HF_USERNAME = "jatinror"  # your HF username
-
-# Your Space ID (updated)
+BASE_URL = "https://agents-course-unit4-scoring.hf.space"  # GAIA API URL
+HF_USERNAME = "jatinror"  # your Hugging Face username
 SPACE_ID = "jatinror/Final_Assignment"
 AGENT_CODE_URL = f"https://huggingface.co/spaces/{SPACE_ID}/tree/main"
 
@@ -54,7 +49,7 @@ def solve_question(question, task_id):
     context = ""
     if file_path and os.path.exists(file_path):
         try:
-            with open(file_path, "r", errors="ignore") as f:
+            with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
                 context = f.read()
         except:
             context = ""
@@ -63,16 +58,34 @@ def solve_question(question, task_id):
     return extract_answer(context)
 
 # ===============================
-# ANSWER EXTRACTION
+# IMPROVED ANSWER EXTRACTION
 # ===============================
 
 def extract_answer(text):
+    """
+    Improved answer extraction for GAIA Level 1:
+    - Return first number if present
+    - Otherwise, return first meaningful short phrase
+    - Strip extra spaces, punctuation
+    """
     import re
+    import string
+
+    if not text:
+        return ""
+
+    # 1️⃣ Try numbers first
     numbers = re.findall(r"\b\d+(?:\.\d+)?\b", text)
     if numbers:
         return numbers[0]
+
+    # 2️⃣ Clean text for short phrase
+    text = text.strip()
+    text = text.translate(str.maketrans('', '', string.punctuation))
     words = text.split()
-    return " ".join(words[:6]).strip()
+
+    # Return first 1–3 words for short answers
+    return " ".join(words[:3]).strip()
 
 # ===============================
 # FETCH QUESTIONS
@@ -95,7 +108,7 @@ def submit_answers(answers):
     }
     response = requests.post(f"{BASE_URL}/submit", json=payload)
     print("Server response:", response.text)
-    return response.text  # Return response to output box
+    return response.text  # Return server response to output box
 
 # ===============================
 # MAIN PIPELINE
@@ -116,7 +129,7 @@ def run_agent():
             "submitted_answer": result.strip()
         })
     server_response = submit_answers(answers)
-    return server_response  # return server response
+    return server_response
 
 # ===============================
 # GRADIO BLOCKS UI