Update core/input_handler.py

core/input_handler.py

@@ -123,13 +123,20 @@ 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_path = files[0].name  # Use .name to get the temp file path
+    
+    file = files[0]
+    file_path = file.name.lower()
+
+    # Handle Image Preview via PIL
     if file_path.endswith((".png", ".jpg", ".jpeg", ".webp")):
-        return gr.update(value=file_path, visible=True), gr.update(visible=False)
+        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_path, visible=True)
-    return gr.update(visible=False), gr.update(visible=False)
+        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: