add choice

app.py

@@ -483,7 +483,7 @@ agent_OPENAI_MULTI = AgentExecutor.from_agent_and_tools(
 # agent.max_execution_time = int(os.getenv("max_iterations"))
 # agent.handle_parsing_errors = True
 # agent.early_stopping_method = "generate"
-
+global agent
 agent = agent_ZEROSHOT_AGENT
 
 print(agent.agent.llm_chain.prompt.template)
@@ -581,6 +581,15 @@ 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
+    elif Choice =='Zero Short React':
+        agent = agent_ZEROSHOT_REACT
+    elif Choice =='OpenAI Multi':
+        agent = agent_OPENAI_MULTI
+
 with gr.Blocks() as demo:
     # gr.Markdown("Start typing below and then click **SUBMIT** to see the output.")
     main = gr.ChatInterface(
@@ -589,14 +598,18 @@ with gr.Blocks() as demo:
         description= "v0.3: Powered by MECH Core Team",
     )
     with gr.Row():
-        upload_button = gr.UploadButton("Upload To DB", file_count="multiple", scale= 1)
-        upload_file_button = gr.UploadButton("Upload File", file_count="single", scale= 1)
+        upload_button = gr.UploadButton("Upload To DB", file_count="multiple", scale= 0)
+        upload_file_button = gr.UploadButton("Upload File", file_count="single", scale= 0)
+        agentchoice = gr.Dropdown(
+            value=['Zero Short Agent','Zero Short React','OpenAI Multi'],
+        )
         voice_input = gr.Audio(source="microphone", type="filepath", scale= 1)
     with gr.Accordion("LOGS"):
         # logs = gr.Textbox()
         logs = gr.Textbox()
     upload_button.upload(func_upload_file, [upload_button, main.chatbot], main.chatbot)
     demo.load(read_logs, None, logs, every=0.5)
+    agentchoice.change(set, agentchoice, None)