Update app.py

app.py

@@ -32,6 +32,18 @@ def set_tts():
 def create_kd_talker():
     return Inferencer() 
 
+example_folder = "example"
+example_choices = ["Example 1", "Example 2", "Example 3"]
+example_mapping = {
+    "Example 1": {"audio": os.path.join(example_folder, "example1.wav"), "image": os.path.join(example_folder, "example1.png")},
+    "Example 2": {"audio": os.path.join(example_folder, "example2.wav"), "image": os.path.join(example_folder, "example2.png")},
+    "Example 3": {"audio": os.path.join(example_folder, "example3.wav"), "image": os.path.join(example_folder, "example3.png")},
+}
+
+def load_example(choice):
+    example = example_mapping.get(choice, {})
+    return example.get("audio", ""), example.get("image", "")
+
 @spaces.GPU
 def predict(prompt, upload_reference_audio, microphone_reference_audio, reference_audio_type):
     global result_dir
@@ -80,7 +92,6 @@ def main():
                 with gr.Tabs(elem_id="kdtalker_source_image"):
                     with gr.TabItem("Upload image"):
                         source_image = gr.Image(label="Source image", sources="upload", type="filepath", scale=256)
-                
                 with gr.Tabs(elem_id="kdtalker_driven_audio"):
                     with gr.TabItem("Upload"):
                         upload_driven_audio = gr.Audio(label="Upload audio", sources="upload", type="filepath")
@@ -116,6 +127,18 @@ def main():
                                 ],
                                 outputs=[gen_video]
                             )
+                    with gr.TabItem("Example"):
+                        example_choice = gr.Dropdown(choices=example_choices, label="Choose an example")
+                        example_audio = gr.Audio(label="Example Audio", type="filepath")
+                        example_image = gr.Image(label="Example Image", type="filepath")
+                        
+                        example_choice.change(
+                            fn=load_example, 
+                            inputs=[example_choice], 
+                            outputs=[upload_driven_audio, source_image]
+                        )
+                        example_choice.change(set_upload, outputs=driven_audio_type)
+
     return interface