Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+from io import BytesIO
 from utils import (
     get_salesforce_client, get_salesforce_objects, get_object_fields,
     extract_text_from_pdf, extract_key_value_pairs, map_fields,
@@ -15,16 +16,34 @@ def upload_pdfs(files):
     if isinstance(files, list):
         for file in files:
             if file is not None:
-                file_path = f"uploads/{os.path.basename(file.name)}"
-                with open(file_path, "wb") as f:
-                    f.write(file.read())
-                uploaded_files.append(file_path)
+                file_name = os.path.basename(file.name)
+                file_path = f"uploads/{file_name}"
+                try:
+                    # Handle NamedString or file-like object
+                    if hasattr(file, 'read'):
+                        content = file.read()
+                    else:
+                        # Assume NamedString; convert string content to bytes
+                        content = str(file).encode('utf-8')
+                    with open(file_path, "wb") as f:
+                        f.write(content)
+                    uploaded_files.append(file_path)
+                except Exception as e:
+                    return f"Error processing {file_name}: {str(e)}"
     else:
         if files is not None:
-            file_path = f"uploads/{os.path.basename(files.name)}"
-            with open(file_path, "wb") as f:
-                f.write(files.read())
-            uploaded_files.append(file_path)
+            file_name = os.path.basename(files.name)
+            file_path = f"uploads/{file_name}"
+            try:
+                if hasattr(files, 'read'):
+                    content = files.read()
+                else:
+                    content = str(files).encode('utf-8')
+                with open(file_path, "wb") as f:
+                    f.write(content)
+                uploaded_files.append(file_path)
+            except Exception as e:
+                return f"Error processing {file_name}: {str(e)}"
     return f"Uploaded {len(uploaded_files)} PDF(s): {', '.join(uploaded_files)}"
 
 def fetch_objects():