Update app.py

app.py

@@ -13,9 +13,14 @@ from state import AgentState
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 SYSTEM_PROMPT = """
-You are a general AI assistant. I will ask you a question. Report your thoughts, 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, don't use comma 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.
+You are a general AI assistant. I will ask you a question. 
+Report your thoughts in brief, and finish your answer with the following template: 
+FINAL ANSWER: [YOUR FINAL ANSWER]
 
 IMPORTANT: When using tools that require file access (such as audio_transcriber_tool, excel_tool, analyze_code_tool, or image_tool), ALWAYS use the task_id parameter only. Do NOT use any file names mentioned by the user - ignore them completely and only pass the task_id.
+
+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, don't use comma 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.
+When using tools that require file access (such as audio_transcriber_tool, excel_tool, analyze_code_tool, or image_tool), ALWAYS use the task_id parameter only. Do NOT use any file names mentioned by the user - ignore them completely and only pass the task_id thats provided to you in the state.
 """
 
 
@@ -40,7 +45,8 @@ class BasicAgent:
         # IMPORTANT: invoke() returns a **new** state instance (or an AddableValuesDict),
         # not the object we pass in.  Use the returned value to fetch final_answer.
         out_state = self.graph.invoke(init_state, {"recursion_limit": 5})
-        print("out_state: ", out_state,'\n\n\n\n')
+        # print("out_state: ", out_state,'\n\n\n\n')
+        
         
         # Extract the final answer from the last message
         if out_state and "messages" in out_state:
@@ -48,6 +54,8 @@ class BasicAgent:
             if hasattr(last_message, 'content'):
                 content = last_message.content
                 # Look for FINAL ANSWER: pattern
+                print("content: ", content)
+                print("\n\n\n\n")
                 if "FINAL ANSWER:" in content:
                     final_answer = content.split("FINAL ANSWER:")[-1].strip()
                     return final_answer