Update app.py

app.py

@@ -13,14 +13,23 @@ def process_file(pdf_file, extractor):
     Returns:
         The extracted file and status message.
     """
-    if extractor == "ALNISF":
-        return process_alnisf(pdf_file)
-    elif extractor == "Federal Electric":
-        return process_federal_electric(pdf_file)
-    elif extractor == "BHEL":
-        return process_bhel(pdf_file)
-    else:
-        return None, "Invalid extractor selected."
+    try:
+        if extractor == "ALNISF":
+            output_file, status = process_alnisf(pdf_file)
+        elif extractor == "Federal Electric":
+            output_file, status = process_federal_electric(pdf_file)
+        elif extractor == "BHEL":
+            output_file, status = process_bhel(pdf_file)
+        else:
+            return None, "Invalid extractor selected."
+
+        # Validate the output
+        if not output_file:
+            return None, "Processing failed. Please ensure the PDF file is valid."
+        
+        return output_file, status
+    except Exception as e:
+        return None, f"An error occurred: {str(e)}"
 
 # Gradio interface setup
 def create_main_interface():