Update gradio_tabs/animation.py

gradio_tabs/animation.py

@@ -157,8 +157,8 @@ def clear_media():
 	return None, None, *([0] * len(labels_k))
 
 
-image_output = gr.Image(label="Output Image", type='numpy', interactive=False, width=512)
-video_output = gr.Video(label="Output Video", width=512)
+image_output = gr.Image(label="Output Image", id="output_img", type='numpy', interactive=False, width=512)
+video_output = gr.Video(label="Output Video", id="output_vid", width=512)
 
 
 @torch.no_grad()
@@ -172,7 +172,7 @@ def animation():
 			with gr.Column(scale=1):
 				with gr.Row():
 					with gr.Accordion(open=True, label="Source Image"):
-						image_input = gr.Image(type="filepath", width=512)	# , height=550)
+						image_input = gr.Image(type="filepath", id="input_img", width=512)	# , height=550)
 						gr.Examples(
 							examples=[
 								["./data/source/macron.png"],
@@ -188,7 +188,7 @@ def animation():
 							)
 
 					with gr.Accordion(open=True, label="Driving Video"):
-						video_input = gr.Video(width=512)  # , height=550)
+						video_input = gr.Video(width=512,id="input_vid",)  # , height=550)
 						gr.Examples(
 							examples=[
 								["./data/driving/driving1.mp4"],
@@ -207,10 +207,10 @@ def animation():
 				with gr.Row():
 					with gr.Column(scale=1):
 						with gr.Row():	# Buttons now within a single Row
-							edit_btn = gr.Button("Edit")
-							clear_btn = gr.Button("Clear")
+							edit_btn = gr.Button("Edit", id="button_edit",)
+							clear_btn = gr.Button("Clear", id="button_clear")
 						with gr.Row():
-							animate_btn = gr.Button("Animate")
+							animate_btn = gr.Button("Animate", id="button_animate")
 
 
 
@@ -227,31 +227,31 @@ def animation():
 					with gr.Tab("Head"):
 						with gr.Row():
 							for k in labels_k[:3]:
-								slider = gr.Slider(minimum=-1.0, maximum=0.5, value=0, label=k)
+								slider = gr.Slider(minimum=-1.0, maximum=0.5, value=0, label=k, id="slider_"+str(k))
 								inputs_s.append(slider)
 						with gr.Row():
 							for k in labels_k[3:6]:
-								slider = gr.Slider(minimum=-0.5, maximum=0.5, value=0, label=k)
+								slider = gr.Slider(minimum=-0.5, maximum=0.5, value=0, label=k, id="slider_"+str(k))
 								inputs_s.append(slider)
 
 					with gr.Tab("Mouth"):
 						with gr.Row():
 							for k in labels_k[6:8]:
-								slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k)
+								slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k, id="slider_"+str(k))
 								inputs_s.append(slider)
 						with gr.Row():
 							for k in labels_k[8:10]:
-								slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k)
+								slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k, id="slider_"+str(k))
 								inputs_s.append(slider)
 
 					with gr.Tab("Eyes"):
 						with gr.Row():
 							for k in labels_k[10:12]:
-								slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k)
+								slider = gr.Slider(minimum=-0.4, maximum=0.4, value=0, label=k, id="slider_"+str(k))
 								inputs_s.append(slider)
 						with gr.Row():
 							for k in labels_k[12:14]:
-								slider = gr.Slider(minimum=-0.2, maximum=0.2, value=0, label=k)
+								slider = gr.Slider(minimum=-0.2, maximum=0.2, value=0, label=k, id="slider_"+str(k))
 								inputs_s.append(slider)