Update app.py

app.py

@@ -26,20 +26,33 @@ image_extensions = Image.registered_extensions()
 video_extensions = ("avi", "mp4", "mov", "mkv", "flv", "wmv", "mjpeg", "wav", "gif", "webm", "m4v", "3gp")
 
 
-def identify_and_save_blob(blob):
+def identify_and_save_blob(blob_path):
     """Identifies if the blob is an image or video and saves it accordingly."""
     try:
-        Image.open(io.BytesIO(blob)).verify()  # Check if it's a valid image
-        extension = ".png"
-        media_type = "image"
-    except:
-        extension = ".mp4"
-        media_type = "video"
-
-    filename = f"temp_{uuid.uuid4()}_media{extension}"
-    with open(filename, "wb") as f:
-        f.write(blob)
-    return filename, media_type
+        with open(blob_path, 'rb') as file:
+            blob_content = file.read()
+            
+            # Try to identify if it's an image
+            try:
+                Image.open(io.BytesIO(blob_content)).verify()  # Check if it's a valid image
+                extension = ".png"  # Default to PNG for saving
+                media_type = "image"
+            except (IOError, SyntaxError):
+                # If it's not a valid image, assume it's a video
+                extension = ".mp4"  # Default to MP4 for saving
+                media_type = "video"
+            
+            # Create a unique filename
+            filename = f"temp_{uuid.uuid4()}_media{extension}"
+            with open(filename, "wb") as f:
+                f.write(blob_content)
+                
+            return filename, media_type
+            
+    except FileNotFoundError:
+        raise ValueError(f"The file {blob_path} was not found.")
+    except Exception as e:
+        raise ValueError(f"An error occurred while processing the file: {e}")
 
 
 @spaces.GPU