decrypt

app.py

@@ -333,12 +333,25 @@ def decrypt_output(user_id):
     with encrypted_output_path.open("rb") as encrypted_output_file:
         encrypted_output = encrypted_output_file.read()
 
+    logger.debug(f"Encrypted output size: {len(encrypted_output)} bytes")
+    logger.debug(f"Encrypted output (first 100 bytes): {encrypted_output[:100].hex()}")
+
+    if not encrypted_output:
+        raise gr.Error("The encrypted output is empty. Please try running the FHE execution again.")
+
     # Retrieve the client API
     client = get_client(user_id)
 
     # Deserialize, decrypt and post-process the encrypted output
-    decrypted_output = client.deserialize_decrypt_post_process(encrypted_output)
-
+    try:
+        decrypted_output = client.deserialize_decrypt_post_process(encrypted_output)
+    except RuntimeError as e:
+        logger.error(f"Error during deserialization: {str(e)}")
+        raise gr.Error("Failed to deserialize the encrypted output. The data might be corrupted or in an unexpected format.")
+    except Exception as e:
+        logger.error(f"Unexpected error during decryption: {str(e)}")
+        raise gr.Error(f"An unexpected error occurred during decryption: {str(e)}")
+    
     return "Seizure detected" if decrypted_output else "No seizure detected"
 
 def resize_img(img, width=256, height=256):