add code interpreter in tools

app.py

@@ -111,7 +111,7 @@ code_2 = """
 !pip install python-pptx
 """
 
-def Code_Runner(code_raw: str, lang: str='python'):
+def Code_Runner(code_raw: str):
     # interpreter = CodeInterpreter(language="python", debug_mode=True)
     if '!pip' in code_raw:
         code_raw=code_raw.replace('!pip', 'pip')
@@ -525,7 +525,8 @@ def UpdateDb():
     print(index.describe_index_stats())
 
 ListAgentWithRemoteGPT = ['Zero Short React 2','Zero Short Agent 2',
-                            'OpenAI Multi 2', 'Conversation Agent']
+                            'OpenAI Multi 2', 'Conversation Agent',
+                            'Code Interpreter']
 
 def SummarizeDoc():
     global vectordb_p
@@ -738,6 +739,13 @@ python_tool2 = Tool(
     description = "Useful when you need python script to answer questions. You should input python code."    
 )
 
+python_tool3 = Tool(
+    name = "Code Runner",
+    func = Code_Runner,
+    description = "Code Interpreter which is able to run code block in local machine.\
+                It is capable to treat any task by running the code and output the result. (i.e. analyzer data, modify/creat documents ...)\
+                You should input python code."
+)
 
 
 # tools = [DB_Search(), wikipedia_tool, duckduckgo_tool, python_tool]
@@ -785,7 +793,7 @@ tools = [DB_Search(), duckduckgo_tool, python_tool, math_tool, Text2Sound_tool]
 
 tools2 = [DB_Search2(), duckduckgo_tool2, wikipedia_tool2, python_tool2, math_tool, Text2Sound_tool2]
 
-tools_remote = [DB_Search2(), duckduckgo_tool2, wikipedia_tool2, python_tool2, math_tool_2, Text2Sound_tool_loc]
+tools_remote = [DB_Search2(), duckduckgo_tool2, wikipedia_tool2, python_tool3, math_tool_2, Text2Sound_tool_loc]
 
 # tools = load_tools(["Vector Database Search","Wikipedia Search","Python REPL","llm-math"], llm=llm)
 
@@ -1012,6 +1020,29 @@ agent_ZEROSHOT_REACT_2 = initialize_agent(tools_remote, GPTfake,
                          
                         )
 
+agent_CODE_INTERPRETER = initialize_agent(tools_remote, GPTfake, 
+                        #  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+                         agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+                         verbose = True,
+                         handle_parsing_errors = True,
+                         max_iterations = int(os.environ["max_iterations"]),
+                         early_stopping_method="generate",
+                         memory = memory2,
+                         agent_kwargs={
+                            'prefix': PREFIX_2,
+                            'format_instructions': FORMAT_INSTRUCTIONS_3,
+                            'suffix': SUFFIX2,
+                            'input_variables': input_variables,
+                         
+                         },
+                        #  input_variables = input_variables,
+                        #  agent_kwargs={
+                        #     'prompt': prompt,
+                        #  }
+                         
+                        )
+
+
 agent_CONVERSATION = initialize_agent(tools_remote, GPTfake, 
                         #  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
                          agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
@@ -1118,6 +1149,11 @@ def SetAgent(Choice):
     elif Choice =='Conversation Agent':
         agent = agent_CONVERSATION
         print("Set to:", Choice)
+    elif Choice =='Code Interpreter':
+        agent = agent_CODE_INTERPRETER
+        print("Set to:", Choice)
+
+
 
 global agent
 Choice = os.environ["agent_type"]
@@ -1636,7 +1672,8 @@ with gr.Blocks() as demo:
 
             agentchoice = gr.Dropdown(
                 choices=['Zero Short Agent','Zero Short React','OpenAI Multi',
-                         'Zero Short React 2','Zero Short Agent 2','None','Conversation Agent'],
+                         'Zero Short React 2','Zero Short Agent 2','None','Conversation Agent',
+                         'Code Interpreter'],
                 label="SELECT AI AGENT",
                 scale= 2,
                 show_label = True,