Clarify prompt

app.py

@@ -23,41 +23,6 @@ llm_engine = HfEngine(model="meta-llama/Meta-Llama-3-70B-Instruct")
 
 authorized_imports = ["numpy"]
 
-
-class PythonInterpreterTool(Tool):
-    name = "python_interpreter"
-    description = "This is a tool that evaluates python code. It can be used to perform calculations."
-
-    output_type = "text"
-    available_tools = BASE_PYTHON_TOOLS.copy()
-
-    def __init__(self, *args, authorized_imports, **kwargs):
-        self.authorized_imports = list(set(LIST_SAFE_MODULES) | set(authorized_imports))
-        self.inputs = {
-            "code": {
-                "type": "text",
-                "description": (
-                    "The code snippet to evaluate. All variables used in this snippet must be defined in this same snippet, "
-                    f"else you will get an error. This code can only import the following python libraries: {authorized_imports}."
-                    "You should MAKE ALL NECESSARY IMPORTS WITHIN THIS CODE SNIPPET, else you will get an import error!"
-                ),
-            }
-        }
-        super().__init__(*args, **kwargs)
-
-    def forward(self, code):
-        for module in authorized_imports:
-            code = "import " + module + "\n" + code
-        output = str(
-            evaluate_python_code(
-                code,
-                tools=self.available_tools,
-                authorized_imports=self.authorized_imports,
-            )
-        )
-        return output
-
-
 agent = ReactCodeAgent(
     llm_engine=llm_engine,
     tools=[],
@@ -65,7 +30,6 @@ agent = ReactCodeAgent(
     max_iterations=10,
 )
 
-
 class FinalAnswerToolWithVerification(Tool):
     name = "final_answer"
     description = "Provides a final answer to the given problem"
@@ -94,7 +58,7 @@ task = """I will give you a basic function that I've created.
 Now I want you to generate a set of unit tests for these functions, check that they run, and give them to me.
 
 Please follow these steps in order:
-1. Understand what this function does and how to use it
+1. Define and run the function given o you, so that it gets defined in your interpreter.
 2. Generate one test function as a python blob, with assert statements
 3. Run the test function in a code snippet and make sure the tests pass
 4. Return to me the complete TEST function (not the original function) as a string code snippet.
@@ -167,6 +131,9 @@ def interact_with_agent(prompt):
 
 
 with gr.Blocks(theme="soft") as demo:
+    gr.Markdown("""### Python test generator
+
+Write your function in the left textbox, and the agent on the right will generate tests for you!""")
     with gr.Row():
         with gr.Column():
             text_input = gr.Textbox(