Update app.py

app.py

@@ -14,6 +14,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndB
 from smolagents import CodeAgent, VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool
 from smolagents.models import ChatMessage
 from custom_tools import WebpageStructureAnalyzerTool
+from system_prompts import CODE_AGENT_DEFAULT_SYSTEM_PROMPTS
 
 subprocess.run(["playwright", "install"], check=True)
 
@@ -197,9 +198,20 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         search_tool = CachedWebSearchTool()
         python_tool = PythonInterpreterTool()
         html_parse_tool = VisitWebpageTool()
-        #(3) Create the CodeAgent, passing the LLM wrapper and tools
+        # (3) Create the system prompt
+        my_added_prompts = """
+        When you receive output from a tool (like wiki_tool or html_parse_tool), do not include the entire raw output in your next thought if it is very long. 
+        Instead, first analyze it (possibly using another tool or by writing code to extract key parts) and only include essential snippets, summaries, or the extracted data relevant to your current plan in your thoughts and observations. 
+        ALL executable Python code that you want to run with the PythonInterpreterTool MUST be enclosed in triple backticks with 'py' like this: ```py\n# your python code here\n``` 
+        Do NOT write any explanations outside of Python comments within these code blocks. 
+        Your thoughts and reasoning should precede the code block. 
+        For web pages, always use web_structure_analyzer_tool to understand the page's layout before attempting to write detailed parsing code with bs4.
+        """
+        combined_system_prompt = CODE_AGENT_DEFAULT_SYSTEM_PROMPTS + "\n\n" + added_prompts
+        # (4) Create the CodeAgent, passing the LLM wrapper and tools
         agent = CodeAgent(model=llm_model, 
                           tools=[web_structure_analyzer_tool, wiki_tool, search_tool, python_tool, html_parse_tool], 
+                          system_prompt=combined_system_prompt,
                           max_steps=5,
                           add_base_tools=True,
                           additional_authorized_imports=["dateparser", "bs4", "regex"])