Update core/input_handler.py

core/input_handler.py

@@ -120,6 +120,29 @@ def validate_video(path):
         logger.error(f"Error validating video: {e}")
         return False, str(e)
 
+def show_preview_from_upload(files):
+    if not files:
+        return gr.update(visible=False), gr.update(visible=False)
+
+    file = files[0]
+    name = file.name.lower()
+    if name.endswith((".png", ".jpg", ".jpeg", ".webp")):
+        return gr.update(value=file, visible=True), gr.update(visible=False)
+    elif name.endswith((".mp4", ".mov", ".avi")):
+        return gr.update(visible=False), gr.update(value=file, visible=True)
+    return gr.update(visible=False), gr.update(visible=False)
+
+def show_preview_from_url(url):
+    if not url:
+        return gr.update(visible=False), gr.update(visible=False)
+
+    url = url.strip().lower()
+    if url.endswith((".png", ".jpg", ".jpeg", ".webp")):
+        return gr.update(value=url, visible=True), gr.update(visible=False)
+    elif url.endswith((".mp4", ".mov", ".avi")):
+        return gr.update(visible=False), gr.update(value=url, visible=True)
+    return gr.update(visible=False), gr.update(visible=False)
+    
 # Input Resolution
 def resolve_input(mode, media_upload, url):
     """