Update app.py

app.py

@@ -62,7 +62,10 @@ def fetch_fields(object_name):
         return gr.update(choices=[]), f"Error: {error}"
     return gr.update(choices=fields), "Fields fetched successfully"
 
-def process_pdf(pdf_path):
+def process_pdf(pdf_paths):
+    if not pdf_paths or not isinstance(pdf_paths, list) or not pdf_paths[0]:
+        return "Error: No valid PDF file provided"
+    pdf_path = pdf_paths[0]  # Take the first file from the list
     text_data, error = extract_text_from_pdf(pdf_path)
     if error:
         return f"Error: {error}"
@@ -71,7 +74,10 @@ def process_pdf(pdf_path):
         return f"Error: {error}"
     return f"Text: {text_data}\nKey-Value Pairs: {kv_pairs}"
 
-def display_mappings(pdf_path, object_name):
+def display_mappings(pdf_paths, object_name):
+    if not pdf_paths or not isinstance(pdf_paths, list) or not pdf_paths[0]:
+        return "Error: No valid PDF file provided"
+    pdf_path = pdf_paths[0]  # Take the first file from the list
     sf, error = get_salesforce_client()
     if error:
         return f"Error: {error}"
@@ -89,7 +95,10 @@ def display_mappings(pdf_path, object_name):
         output += f"{key} -> {field} (Confidence: {confidence_scores[key]})\n"
     return output
 
-def migrate_to_salesforce(pdf_path, object_name):
+def migrate_to_salesforce(pdf_paths, object_name):
+    if not pdf_paths or not isinstance(pdf_paths, list) or not pdf_paths[0]:
+        return "Error: No valid PDF file provided"
+    pdf_path = pdf_paths[0]  # Take the first file from the list
     sf, error = get_salesforce_client()
     if error:
         log_failure(pdf_path, object_name, error)
@@ -129,7 +138,7 @@ def display_failures():
         return "No failures logged"
 
 def retry_migration(pdf_path, object_name):
-    return migrate_to_salesforce(pdf_path, object_name)
+    return migrate_to_salesforce([pdf_path], object_name)  # Wrap single path in list for consistency
 
 # Gradio UI
 with gr.Blocks() as app: