Samuel Thomas commited on
Commit
5145d77
·
1 Parent(s): 9c250a6

strip final answer

Browse files
Files changed (1) hide show
  1. app.py +20 -1
app.py CHANGED
@@ -4,9 +4,28 @@ import requests
4
  import inspect
5
  import pandas as pd
6
  import traceback
 
7
  from huggingface_hub import login
8
  from tools import create_memory_safe_workflow, get_file_type, write_bytes_to_temp_dir, AgentState, extract_final_answer, run_agent
9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  # (Keep Constants as is)
11
  # --- Constants ---
12
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -105,7 +124,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
105
  task_id = hf_questions[r]['task_id']
106
  question_text = hf_questions[r]['question']
107
  full_answer = run_agent(agent, s)
108
- submitted_answer = extract_final_answer(full_answer[-1].content)
109
  print(f"\n\nQuestion {r+1} Answer: {submitted_answer}\n\n")
110
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
111
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
 
4
  import inspect
5
  import pandas as pd
6
  import traceback
7
+ import re
8
  from huggingface_hub import login
9
  from tools import create_memory_safe_workflow, get_file_type, write_bytes_to_temp_dir, AgentState, extract_final_answer, run_agent
10
 
11
+ import re
12
+
13
+ def strip_final_answer(text):
14
+ """
15
+ Removes 'FINAL ANSWER:' (case-insensitive) and all following whitespace from the start of the string.
16
+ Returns the remainder of the string.
17
+ """
18
+ # The regex matches 'FINAL ANSWER:', optional colon, and all whitespace after it
19
+ return re.sub(r'^\s*FINAL ANSWER:\s*', '', text, flags=re.IGNORECASE)
20
+
21
+ # Example usage:
22
+ s = "FINAL ANSWER: Joe Torre"
23
+ print(strip_final_answer(s)) # Output: Joe Torre
24
+
25
+ s2 = " FINAL ANSWER: Jane Doe"
26
+ print(strip_final_answer(s2)) # Output: Jane Doe
27
+
28
+
29
  # (Keep Constants as is)
30
  # --- Constants ---
31
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
124
  task_id = hf_questions[r]['task_id']
125
  question_text = hf_questions[r]['question']
126
  full_answer = run_agent(agent, s)
127
+ submitted_answer = strip_final_answer(extract_final_answer(full_answer[-1].content))
128
  print(f"\n\nQuestion {r+1} Answer: {submitted_answer}\n\n")
129
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
130
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})