Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 import gradio as gr
 import requests
@@ -63,7 +62,7 @@ class CodeLlamaTool(Tool):
         )
 
     def forward(self, question: str) -> str:
-        prompt = f"""You are an AI that uses Python code to answer questions.
+        self.prompt = f"""You are an AI that uses Python code to answer questions.
 Question: {question}
 Instructions:
 - If solving requires code, use a block like <tool>code</tool>.
@@ -309,7 +308,7 @@ Answer should be a short string, number, or comma-separated list. Keep it brief.
             ],
             add_base_tools=True
         )
-        self.agent.prompt_templates["system_prompt"] = self.agent.prompt_templates["system_prompt"] + system_prompt
+        self.agent.prompt_templates["system_prompt"] = self.agent.prompt_templates["system_prompt"] + self.system_prompt
 
     def _build_safe_prompt(self, history: str, question: str, max_total_tokens=32768, reserve_for_output=2048):
         max_input_tokens = max_total_tokens - reserve_for_output