chore: show inference results in notebook

Author: Daniel Lorch

notebooks/misc/e2e-example-with-multimodal-page-classification-few-shot-prompting.ipynb

@@ -108,6 +108,7 @@
     "import boto3\n",
     "import logging\n",
     "import datetime\n",
+    "import pprint\n",
     "\n",
     "# Import base libraries\n",
     "from idp_common.models import Document, Status, Section, Page\n",
@@ -393,7 +394,16 @@
     "\n",
     "document_dict = document.to_dict()\n",
     "sections_json = json.dumps(document_dict[\"sections\"][:n], indent=2)\n",
-    "print(f\"{sections_json}...\")"
+    "print(f\"{sections_json}...\")\n",
+    "\n",
+    "for section in document.sections[:n]:\n",
+    "    print(f\"\\nSection {section.section_id} extraction result:\")\n",
+    "    extraction_result_uri = section.extraction_result_uri\n",
+    "    bucket, key = parse_s3_uri(extraction_result_uri)\n",
+    "    response = s3_client.get_object(Bucket=bucket, Key=key)\n",
+    "    content = response['Body'].read().decode('utf-8')\n",
+    "    result = json.loads(content)\n",
+    "    pprint.pp(result[\"inference_result\"])"
    ]
   },
   {