Update core/input_handler.py

core/input_handler.py

@@ -123,30 +123,25 @@ def validate_video(path):
 def show_preview_from_upload(files):
     if not files or len(files) == 0:
         return gr.update(visible=False), gr.update(visible=False)
-    
     file = files[0]
     file_path = file.name.lower()
-
-    # Handle Image Preview via PIL
     if file_path.endswith((".png", ".jpg", ".jpeg", ".webp")):
         pil_image = Image.open(file).convert("RGB")
         return gr.update(value=pil_image, visible=True), gr.update(visible=False)
-
-    # Handle Video Preview via .name
     elif file_path.endswith((".mp4", ".mov", ".avi")):
         return gr.update(visible=False), gr.update(value=file.name, visible=True)
-
     return gr.update(visible=False), gr.update(visible=False)
 
-def show_preview_from_url(url):
-    if not url:
+def show_preview_from_url(url_input):
+    if not url_input:
         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)
+    url_lower = url_input.strip().lower()
+    if url_lower.endswith((".png", ".jpg", ".jpeg", ".webp")):
+        return gr.update(value=url_input, visible=True), gr.update(visible=False)
+    elif url_lower.endswith((".mp4", ".mov", ".avi")):
+        return gr.update(visible=False), gr.update(value=url_input, visible=True)
     return gr.update(visible=False), gr.update(visible=False)
+
     
 # Input Resolution
 def resolve_input(mode, media_upload, url):