Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 from markitdown import MarkItDown
 import google.generativeai as genai
@@ -25,38 +24,48 @@ import tempfile
 import os
 import io
 
-def save_uploaded_file(file):
-    """Saves a file uploaded with gardio to a temporary location."""
+import tempfile
+import os
+import io
+from gardio import upload_file
 
-    if file is None:
-        return None
+def save_uploaded_file(file_data):
+    """Saves a file uploaded with gardio to a temporary location.
 
-    try:
-        temp_dir = tempfile.gettempdir()
-        temp_filename = tempfile.mkstemp(dir=temp_dir, suffix=os.path.splitext(file.filename)[1])[1]
+    Args:
+        file_data: The file data object returned by gardio.get_file().
 
-        # 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."
+    Returns:
+        The path to the saved file, or an error message as a string.
+    """
 
+    if file_data is None:
+        return "No file uploaded."
+    
+    try:
+        # Extract the filename; crucial to avoid errors
+        filename = file_data.filename if hasattr(file_data, 'filename') else "unknown_file"
+        temp_dir = tempfile.gettempdir()
+        temp_filename = tempfile.mkstemp(dir=temp_dir, suffix=os.path.splitext(filename)[1])[1]
 
         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)}"
+            try:
+                if hasattr(file_data, 'read'):
+                    for chunk in file_data.chunks():
+                        f.write(chunk)
+                elif hasattr(file_data, 'getvalue'):
+                    f.write(file_data.getvalue())
+                else:
+                    return "Error: Unsupported file-like object."
+
+            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)}"
 
+    except Exception as e:
+        return f"An error occurred during file saving: {str(e)}"
+        
 async def summarize_text(text):
     """Summarize the input text using Gemini AI"""
     try: