Update app.py

app.py

@@ -34,19 +34,25 @@ print(f"Model {chat.model} downloaded!")
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
-def get_file_path(task_id: str, question) -> str:
+def get_file_path(question: str) -> str:
     """Retrieves reference file path."""
-    if question['task_id'] == task_id:
+    if isinstance(question, dict):
         return question['file_path']
+    elif isinstance(question, str):
+        for q in question_dataset:
+            if q['Question'] == question:
+                return q['file_path']
 
 
 def get_ref_content(path: str) -> str | object:
     """Retrieves content from the reference path provided."""
-    with open(path, "rb") as f:
-        file = f.read()
+    if path.startswith('http:'):
+        file = requests.get(path).content
+    else:
+        with open(path, "rb") as f:
+            file = f.read()
     return file
 
-
 def search_web(topic: str) -> str:
     """Retrieves information about the topic."""
     results = DuckDuckGoSearchRun().invoke(topic)
@@ -131,23 +137,15 @@ builder.add_edge("tools", "assistant")
 alfred = builder.compile()
 system_prompt = SystemMessage(
     content="You are a general AI assistant. \
-        I will ask you a question. Report your thoughts shortly, \
-        and finish your answer with the following template: \
+        I will ask you a question. Report your thoughts shortly, and finish your answer with the following template: \
         FINAL ANSWER: [YOUR FINAL ANSWER]. \
-        YOUR FINAL ANSWER should be a number OR as few words as possible \
-        OR a comma separated list of numbers and/or strings. \
-        If you are asked for a number, use only digits in your final answer. \
-        Don't use comma nor brackets to write your number neither use units such as $ or percent sign \
-        unless specified otherwise. \
-        If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), \
-        and write the digits in plain text unless specified otherwise. \
-        If you are asked for a comma separated list, apply the above rules \
-        depending of whether the element to be put in the list is a number or a string. \
-        If the question refers to an external content and there is no reference file attached, \
-        perform a web search and retrieve relevant information from the internet. \
-        Make sure that each final answer is preceded with 'FINAL ANSWER:'. "
-)
-
+        YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. \
+        If you are asked for a number, use only digits in your final answer. Don't use comma nor brackets to write your number neither use units such as $ or percent sign unless specified otherwise. \
+        If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. \
+        If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string. \
+        If the question refers to an external content and there is no reference file attached, perform a web search and retrieve relevant information from the internet. \
+        Make sure that each final answer is preceded with 'FINAL ANSWER:' and is short: it should contain a number, a string or a comma-separated list, nothing else. "
+    )
 
 class BasicAgent:
     def __init__(self):
@@ -158,8 +156,7 @@ class BasicAgent:
         message = HumanMessage(content=question)
         print(message)
         answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
-        print(answer)
-        #answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))    
+        answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))    
         answer = answer.replace('FINAL ANSWER: ', '')
         answer = answer.replace('[', '')
         nswer = answer.replace('*', '')