update direct pass type

app.py

@@ -745,25 +745,30 @@ agent_OPENAI_MULTI = AgentExecutor.from_agent_and_tools(
 # agent.handle_parsing_errors = True
 # agent.early_stopping_method = "generate"
 
+def SetAgent(Choice):
+    global agent
+    if Choice =='Zero Short Agent':
+        agent = agent_ZEROSHOT_AGENT
+        print("Set to:", Choice)
+    elif Choice =='Zero Short React':
+        agent = agent_ZEROSHOT_REACT
+        print("Set to:", Choice)
+    elif Choice =='OpenAI Multi':
+        agent = agent_OPENAI_MULTI
+        print("Set to:", Choice)
+    elif Choice =='Zero Short React 2':
+        agent = agent_ZEROSHOT_REACT_2
+        print("Set to:", Choice)
+    elif Choice =='Zero Short Agent 2':
+        agent = agent_ZEROSHOT_AGENT_2
+        print("Set to:", Choice)
+    elif Choice == "None":
+        agent = None
+        print("Set to:", Choice)
+
 global agent
 Choice = os.environ["agent_type"]
-
-if Choice =='Zero Short Agent':
-    agent = agent_ZEROSHOT_AGENT
-    print("Set to:", Choice)
-elif Choice =='Zero Short React':
-    agent = agent_ZEROSHOT_REACT
-    print("Set to:", Choice)
-elif Choice =='OpenAI Multi':
-    agent = agent_OPENAI_MULTI
-    print("Set to:", Choice)
-elif Choice =='Zero Short React 2':
-    agent = agent_ZEROSHOT_REACT_2
-    print("Set to:", Choice)
-elif Choice =='Zero Short Agent 2':
-    agent = agent_ZEROSHOT_AGENT_2
-    print("Set to:", Choice)
-
+SetAgent(Choice)
 
 
 # agent = agent_ZEROSHOT_AGENT
@@ -902,23 +907,8 @@ def read_logs():
     with open("output.log", "r") as f:
         return f.read()
 
-def SetAgent(Choice):
-    global agent
-    if Choice =='Zero Short Agent':
-        agent = agent_ZEROSHOT_AGENT
-        print("Set to:", Choice)
-    elif Choice =='Zero Short React':
-        agent = agent_ZEROSHOT_REACT
-        print("Set to:", Choice)
-    elif Choice =='OpenAI Multi':
-        agent = agent_OPENAI_MULTI
-        print("Set to:", Choice)
-    elif Choice =='Zero Short React 2':
-        agent = agent_ZEROSHOT_REACT_2
-        print("Set to:", Choice)
-    elif Choice =='Zero Short Agent 2':
-        agent = agent_ZEROSHOT_AGENT_2
-        print("Set to:", Choice)
+
+
 
 global record
 record = []
@@ -941,12 +931,16 @@ def LinkElement(chatbot_history):
 def chathmi3(message, history2):
     global last_request
     global Filename_Chatbot
+    global agent
     print("Input Message:", message)
     last_request = message
     history2 = history2 + [(message, None)]
     yield ["", history2]
     try:
-        response = agent.run(message)
+        if agent is not None:
+            response = agent.run(message)
+        elif agent is None:
+            response = asyncio.run(start_playwright(message))
         time.sleep(0.1)
         history2 = history2 + [(None, response)]
         yield ["", history2]
@@ -975,13 +969,19 @@ def chathmi3(message, history2):
 def chathmi4(message, history2):
     global last_request
     global Filename_Chatbot
+    global agent
     print("Input Message:", message)
+    
     last_request = message
     history2 = history2 + [(message, None)]
     yield ["", history2, gr.update(visible = False), gr.update(visible = True)]
     # yield ["", history2, "SUBMIT", "STOP"]
     try:
-        response = agent.run(message)
+        if agent is not None:
+            response = agent.run(message)
+        elif agent is None:
+            response = asyncio.run(start_playwright(message))
+
         time.sleep(0.1)
         history2 = history2 + [(None, response)]
         yield ["", history2, gr.update(visible = True), gr.update(visible = False)]
@@ -994,8 +994,8 @@ def chathmi4(message, history2):
             file_name = temp.split(")")[0]
             print("file_name:", file_name)
             history2 = history2 + [(None, (file_name,))]
-            Filename_Chatbot = file_name
-            yield ["", history2, "SUBMIT", "STOP"]
+            Filename_Chatbot = file_name    
+            yield ["", history2, "SUBMIT", "STOP"]        
         except:
             print("No need to add file in chatbot")        
 
@@ -1003,7 +1003,7 @@ def chathmi4(message, history2):
         print("chathmi4 error:", e)
 
     # history = history + [(message, None)]
-    
+        
     print("History2: ", history2)
     print("-" * 20)
     print("-" * 20)
@@ -1239,7 +1239,7 @@ with gr.Blocks() as demo:
         with gr.Row():
 
             agentchoice = gr.Dropdown(
-                choices=['Zero Short Agent','Zero Short React','OpenAI Multi','Zero Short React 2','Zero Short Agent 2'],
+                choices=['Zero Short Agent','Zero Short React','OpenAI Multi','Zero Short React 2','Zero Short Agent 2','None'],
                 label="SELECT AI AGENT",
                 scale= 2,
                 show_label = True,