retrieve adn edit

app.py

@@ -60,6 +60,23 @@ WEBSITE = ("""<div class="embed_hidden" style="text-align: center;">
 </div>
 """)
 
+
+WEB_source =  ("""<div class="embed_hidden" style="text-align: center;">
+    <h1>Pick a motion to edit!</h1>
+    <h3>
+        Here you should pick a source motion
+    </h3>
+    <hr class="double">
+</div>
+""")
+
+WEB_target =  ("""<div class="embed_hidden" style="text-align: center;">
+    <h1>Now type the text to edit that motion!</h1>
+    <h3>
+        Here you should get the generated motion!
+    </h3>
+</div>
+""")
 @spaces.GPU
 def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
@@ -164,30 +181,38 @@ def random_number():
 
 with gr.Blocks() as demo:
     gr.Markdown(WEBSITE)
-    
+    gr.Markdown(WEB_source)
+    # TODO load TMR text-encoder
+    # load motion embddings
+    # compute scores to check most similar ones with the text
+    # edit that motion!
     with gr.Row():
-        with gr.Column(scale=8):
+        with gr.Column(scale=10):
             retrieve_text = gr.Textbox(placeholder="Type the text for the motion you want to Retrieve:",
                                        show_label=True, label="Retrieval Text", value=DEFAULT_TEXT)
-        with gr.Column(scale=1):
-            clear_button_retrieval = gr.Button("Clear Retrieval Text")
-    
-    with gr.Row():
-        with gr.Column(scale=8):
-            input_text = gr.Textbox(placeholder="Type the edit text you want:",
-                                    show_label=True, label="Input Text", value=DEFAULT_TEXT)
-        with gr.Column(scale=1):
-            clear_button_edit = gr.Button("Clear Edit Text")
+            xxx = 'https://motion-editing.s3.eu-central-1.amazonaws.com/collection_wo_walks_runs/rendered_pairs/011327_120_240-002682_120_240.mp4'
 
+        with gr.Column(scale=8):
+            retrieved_video_output = gr.Video(label="Retrieved Motion",
+                                            value=xxx,
+                                            height=360, width=480)
     with gr.Row():
-        video_output = gr.Video(label="Generated Video", height=240, width=320)
-        retrieved_video_output = gr.Video(label="Retrieved Motion", height=240, width=320)
+        clear_button_retrieval = gr.Button("Clear Retrieval Text")
+        retrieve_button = gr.Button("TMRetrieve")
+        random_button = gr.Button("Random")
 
+    gr.Markdown(WEB_target)
     with gr.Row():
-        edit_button = gr.Button("Edit")
-        retrieve_button = gr.Button("Retrieve")
+        with gr.Column(scale=8):
+            input_text = gr.Textbox(placeholder="Type the edit text you want:",
+                                    show_label=True, label="Input Text", value=DEFAULT_TEXT)
+            with gr.Row():
+                edit_button = gr.Button("Edit")
+                clear_button_edit = gr.Button("Clear Edit Text")
 
-        random_button = gr.Button("Random")
+        with gr.Column(scale=8):
+            video_output = gr.Video(label="Generated Video", height=360, 
+                                    width=480)
 
     def process_and_show_video(input_text):
         fname = show_video(input_text)
@@ -196,10 +221,10 @@ with gr.Blocks() as demo:
     def process_and_retrieve_video(input_text):
         fname = retrieve_video(input_text)
         return fname
-    
-    from gen_utils import read_config
-    from retrieval_loader import load_model_from_cfg
+
     from retrieval_loader import get_tmr_model
+    # load the dataset and splits
+    
     edit_button.click(process_and_show_video, inputs=input_text, outputs=video_output)
     retrieve_button.click(process_and_retrieve_video, inputs=retrieve_text, outputs=retrieved_video_output)
     # import ipdb;ipdb.set_trace()