Update app.py

app.py

@@ -26,40 +26,34 @@ import os
 import io
 
 def save_uploaded_file(file):
-    """Save uploaded file to temporary location and return path.  
-    Handles only file uploads, not URLs."""
+    """Saves a file uploaded with gardio to a temporary location."""
 
     if file is None:
         return None
 
     try:
         temp_dir = tempfile.gettempdir()
-        temp_filename = tempfile.mkstemp(dir=temp_dir, suffix=os.path.splitext(file.name)[1])[1]
-
-        # Crucial: Check for valid file-like object
-        if hasattr(file, 'read'):  
-          with open(temp_filename, 'wb') as f:
-            try:
-              while True:
-                chunk = file.read(8192)
-                if not chunk:
-                  break
-                f.write(chunk)
-            except Exception as e:
-                return f"Error during file write: {str(e)}"
-
-        elif hasattr(file, 'getvalue'):  # In-memory file-like
-          with open(temp_filename, 'wb') as f:
-            try:
-              f.write(file.getvalue())
-            except Exception as e:
-              return f"Error during file write: {str(e)}"
-        
-        else:
-           return "Error: Unsupported file type."
+        temp_filename = tempfile.mkstemp(dir=temp_dir, suffix=os.path.splitext(file.filename)[1])[1]
+
+        # Check if the file object is valid before attempting to read it
+        if not hasattr(file, 'filename') or not file.filename:
+            return "Error: Invalid file object. Missing filename."
+
+
+        with open(temp_filename, 'wb') as f:
+          try:
+            # Crucial: Check for a 'read' attribute to handle various file-like objects
+            if hasattr(file, 'read'):
+                for chunk in file.chunks():
+                    f.write(chunk)
+            elif hasattr(file, 'getvalue'):
+                f.write(file.getvalue())
+            else:
+                return "Error: Unsupported file-like object. Missing 'read' or 'getvalue'."
+          except Exception as e:
+              return f"Error writing to file: {str(e)}"
 
         return temp_filename
-    
     except Exception as e:
         return f"Error saving file: {str(e)}"