Update app.py

app.py

@@ -1,11 +1,19 @@
 import gradio as gr
 from main import extract_entities_from_file
+from starlette.requests import ClientDisconnect
 
 def process(file):
-    results = extract_entities_from_file(file)
-    if not results:
-        return "No entities found."
-    return "\n".join([f"{text} -> {label}" for text, label in results])
+    try:
+        results = extract_entities_from_file(file)
+        if not results:
+            return "No entities found."
+        return "\n".join([f"{text} -> {label}" for text, label in results])
+    except ClientDisconnect:
+        print("Client disconnected during processing.")
+        return "Client disconnected before processing could complete."
+    except Exception as e:
+        print(f"Unhandled error: {e}")
+        return "An error occurred during processing."
 
 iface = gr.Interface(
     fn=process,