Update app.py

app.py

@@ -54,14 +54,8 @@ def get_video_svd(image_in):
     print(result)
     return result[0]["video"]
 
-def infer(image_in, camera_shot, conditional_pose, prompt, style, chosen_model):
-    if camera_shot == "custom":
-        if conditional_pose != None:
-            conditional_pose = conditional_pose
-        else :
-            raise gr.Error("No custom conditional shot found !")
-    
-    elif camera_shot == "close-up":
+def load_sample_shot(camera_shot):
+    if camera_shot == "close-up":
         conditional_pose = "camera_shots/close_up_shot.jpeg"
     elif camera_shot == "medium close-up":
         conditional_pose = "camera_shots/medium_close_up.jpeg"
@@ -73,6 +67,19 @@ def infer(image_in, camera_shot, conditional_pose, prompt, style, chosen_model):
         conditional_pose = "camera_shots/medium_full_shot.png"
     elif camera_shot == "full shot":
         conditional_pose = "camera_shots/full_shot.jpeg"
+    return conditional_pose
+
+def use_custom_cond():
+    return "custom"
+
+def infer(image_in, camera_shot, conditional_pose, prompt, style, chosen_model):
+    
+    if camera_shot == "custom":
+        if conditional_pose != None:
+            conditional_pose = conditional_pose
+        else :
+            raise gr.Error("No custom conditional shot found !")
+    
     
     iid_img = get_instantID(image_in, conditional_pose, prompt, style)
     
@@ -127,6 +134,19 @@ with gr.Blocks(css=css) as demo:
             submit_btn = gr.Button("Submit")
             video_out = gr.Video()
 
+    
+    camera_shot.change(
+        fn = load_sample_shot,
+        inputs = camera_shot,
+        outputs = condition_shot,
+        queue=False
+    )
+    condition_shot.upload(
+        fn = use_custom_cond,
+        inputs = None,
+        outputs = camera_shot,
+        queue=False,
+    )
     submit_btn.click(
         fn = infer,
         inputs = [