more

app.py

@@ -165,7 +165,7 @@ def get_target_from_request(request_text):
         return 2
     if any(item in request_text for item in ['THREE', 'THIRD']):
         return 3
-    if any(item in request_text for item in ['FOUR', 'FOURTH']):
+    if any(item in request_text for item in ['FOUR', 'FOURTH', 'FOR']):
         return 4
     return 'No goal found.'
 
@@ -176,13 +176,13 @@ with gr.Blocks(css=custom_css) as demo:
     gr.Markdown('Say the agent where to go and what to do')
 
     with gr.Row():
-        with gr.Column():
-            request_audio = gr.Microphone(editable=False)
-            # send_btn = gr.Button(value='Send Request')
-            request_text = gr.Textbox(label="Request:", lines=2, interactive=False)
-            request_target = gr.Textbox(label='Target:', lines=2)
+        with gr.Column(scale=12):
             status = gr.Textbox(label='Status:', lines=2, elem_id="mytextbox")
-        with gr.Column():
+            request_audio = gr.Microphone(editable=False, scale=2)
+            # send_btn = gr.Button(value='Send Request')
+            request_text = gr.Textbox(label="Request:", lines=2, interactive=False, visible=False)
+            request_target = gr.Textbox(label='Target:', lines=2, visible=False)
+        with gr.Column(scale=8):
             output_env = gr.Video(label="Env:", autoplay=True)
     with gr.Accordion("TODO List", open=False):
         gr.Markdown("""
@@ -199,6 +199,7 @@ with gr.Blocks(css=custom_css) as demo:
     # gr.on(triggers=["load"], fn=load_image_on_start, outputs=output_env_image)
     # my_demo.load(fn=load_image_on_start, outputs=output_env_image)
     demo.load(fn=create_standing_animation, outputs=output_env)
+    demo.load(fn=lambda: 'Where do you want me to go?', outputs=status)
     # request_audio.stream(fn=get_text_request, inputs=request_audio, outputs=request_text)
     request_audio.stop_recording(fn=get_text_request, inputs=request_audio, outputs=request_text)
     request_text.change(fn=get_target_from_request, inputs=request_text, outputs=request_target)