Update app.py

app.py

@@ -14,6 +14,13 @@ from io import BytesIO
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# load the system prompt from the file
+with open("prompt2.txt", "r", encoding="utf-8") as f:
+    system_prompt = f.read()
+
+# System message
+sys_msg = SystemMessage(content=system_prompt)
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
@@ -66,8 +73,19 @@ class BasicAgent:
                 messages = [HumanMessage(content=question)]
     
             response = self.graph.invoke({"messages": messages})
-            answer = response['messages'][-1].content
+            answer = response['messages'][-1].content # First pass 
+            #return answer.split("FINAL ANSWER:")[-1].strip() # Return for single pass
+
+            # Structure the messages
+            messages = [
+                {"role": "system", "content": sys_msg},
+                {"role": "user", "content": response}
+            ]
+            
+            # Call the agent again
+            response = self.graph.invoke({"messages": messages})
             return answer.split("FINAL ANSWER:")[-1].strip()
+
     
         except Exception as e:
             return f"Error during processing: {e}"