Upload app.py

app.py

@@ -0,0 +1,99 @@
+import gradio as gr
+import os
+import random
+import numpy as np
+
+def video_identity(video):
+    return video
+
+def nan():
+    return None
+
+# demo = gr.Interface(video_identity, 
+#                     gr.Video(), 
+#                     "playable_video", 
+#                     examples=[
+#                         os.path.join(os.path.dirname(__file__), 
+#                                      "videos/rl-video-episode-0.mp4")], 
+#                     cache_examples=True)
+
+FORMAT = ['mp4', 'gif'][1]
+
+def update(left, right, data_folder='videos'):
+    # data_folder='videos'
+    envs = parse_envs()   
+    env_name = envs[random.randint(0, len(envs)-1)]
+    # choose video
+    videos = os.listdir(os.path.join(data_folder, env_name))
+    video_files = []
+    for f in videos:
+        if f.endswith(f'.{FORMAT}'):
+            video_files.append(os.path.join(data_folder, env_name, f))
+    # choose two videos
+    selected_video_ids = np.random.choice(len(video_files), 2, replace=False)
+    left = video_files[selected_video_ids[0]]
+    right = video_files[selected_video_ids[1]]
+    print(env_name, left, right)
+    return left, right
+
+# def update(left, right):
+#     if FORMAT == 'mp4':
+#         left = os.path.join(os.path.dirname(__file__), 
+#                              "videos/rl-video-episode-2.mp4")
+#         right = os.path.join(os.path.dirname(__file__), 
+#                              "videos/rl-video-episode-3.mp4")
+#     else:
+#         left = os.path.join(os.path.dirname(__file__), 
+#                              "videos/rl-video-episode-2.gif")
+#         right = os.path.join(os.path.dirname(__file__), 
+#                              "videos/rl-video-episode-3.gif")  
+#     print(left, right)     
+#     return left, right
+
+def replay(left, right):  
+    return left, right
+
+def parse_envs(folder='./videos'):
+    envs = []
+    for f in os.listdir(folder):
+        if os.path.isdir(os.path.join(folder, f)):
+            envs.append(f)
+    return envs
+
+def build_interface(iter=3, data_folder='./videos'):
+    # build gradio interface
+    with gr.Blocks() as demo:
+        gr.Markdown("Here is RoboTinder!")
+        gr.Markdown("Select the best robot behaviour in your choice!")
+        with gr.Row():
+            # some initial videos
+            if FORMAT == 'mp4':
+                left_video_path = os.path.join(os.path.dirname(__file__), 
+                                    "videos/rl-video-episode-0.mp4")
+                right_video_path = os.path.join(os.path.dirname(__file__), 
+                                    "videos/rl-video-episode-1.mp4")
+                left = gr.PlayableVideo(left_video_path)
+                right = gr.PlayableVideo(right_video_path)
+            else:
+                left_video_path = os.path.join(os.path.dirname(__file__), 
+                                    "videos/rl-video-episode-0.gif")
+                right_video_path = os.path.join(os.path.dirname(__file__), 
+                                    "videos/rl-video-episode-1.gif")  
+                left = gr.Image(left_video_path, shape=(1024, 768))
+                # right = gr.Image(right_video_path).style(height=768, width=1024)
+                right = gr.Image(right_video_path)
+
+        btn1 = gr.Button("Replay")
+        btn1.click(fn=replay, inputs=[left, right], outputs=[left, right])
+
+        gr.Radio(["Left", "Right", "Not Sure"], label="Which one is your favorite?")
+        btn2 = gr.Button("Next")
+        # choose env
+        btn2.click(fn=update, inputs=[left, right], outputs=[left, right])
+
+    return demo
+
+if __name__ == "__main__":
+    demo = build_interface()
+    # demo.launch(share=True)
+    demo.launch(share=False)