Update app.py

app.py

@@ -101,6 +101,12 @@ def enhance_prompt(prompt, type="t2v"):
         print(f"Error during prompt enhancement: {e}")
         return prompt
 
+def update_prompt_t2v(prompt, enhance_toggle):
+    return update_prompt(prompt, enhance_toggle, "t2v")
+
+def update_prompt_i2v(prompt, enhance_toggle):
+    return update_prompt(prompt, enhance_toggle, "i2v")
+    
 def update_prompt(prompt, enhance_toggle, type="t2v"):
     if enhance_toggle:
         return enhance_prompt(prompt, type)
@@ -611,15 +617,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                 )
 
     # Event handlers
+# Event handlers
     txt2vid_preset.change(
-        fn=preset_changed, inputs=[txt2vid_preset], outputs=txt2vid_advanced[3:]
+        fn=preset_changed, 
+        inputs=[txt2vid_preset], 
+        outputs=txt2vid_advanced[3:]
     )
 
     txt2vid_enhance_toggle.change(
-        fn=update_prompt,
+        fn=update_prompt_t2v,
         inputs=[txt2vid_prompt, txt2vid_enhance_toggle],
-        outputs=txt2vid_prompt,
-        kwargs={"type": "t2v"}
+        outputs=txt2vid_prompt
     )
 
     txt2vid_generate.click(
@@ -638,14 +646,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     )
 
     img2vid_preset.change(
-        fn=preset_changed, inputs=[img2vid_preset], outputs=img2vid_advanced[3:]
+        fn=preset_changed, 
+        inputs=[img2vid_preset], 
+        outputs=img2vid_advanced[3:]
     )
 
     img2vid_enhance_toggle.change(
-        fn=update_prompt,
+        fn=update_prompt_i2v,
         inputs=[img2vid_prompt, img2vid_enhance_toggle],
-        outputs=img2vid_prompt,
-        kwargs={"type": "i2v"}
+        outputs=img2vid_prompt
     )
 
     img2vid_generate.click(