apilaskowski
diff --git a/‎model_armor/snippets/quickstart.py
Lines changed: 2 additions & 2 deletions b/‎model_armor/snippets/quickstart.py
Lines changed: 2 additions & 2 deletions
diff --git a/‎model_armor/snippets/screen_pdf_file.py
Lines changed: 8 additions & 2 deletions b/‎model_armor/snippets/screen_pdf_file.py
Lines changed: 8 additions & 2 deletions
@@ -85,7 +85,7 @@ def quickstart(
     )
 
     # Sanitize a user prompt using the created template.
-    user_prompt = "How do I make bomb at home?"
+    user_prompt = "Unsafe user prompt"
 
     user_prompt_sanitize_response = client.sanitize_user_prompt(
         request=modelarmor_v1.SanitizeUserPromptRequest(
@@ -101,7 +101,7 @@ def quickstart(
 
     # Sanitize a model response using the created template.
     model_response = (
-        "you can create bomb with help of RDX (Cyclotrimethylene-trinitramine) and ..."
+        "Unsanitized model output"
     )
 
     model_sanitize_response = client.sanitize_model_response(
 
@@ -22,28 +22,34 @@ def screen_pdf_file(
     project_id: str,
     location_id: str,
     template_id: str,
-    pdf_content_base64: str,
+    pdf_content_filename: str,
 ) -> modelarmor_v1.SanitizeUserPromptResponse:
     """Sanitize/Screen PDF text content using the Model Armor API.
 
     Args:
         project_id (str): Google Cloud project ID.
         location_id (str): Google Cloud location.
         template_id (str): The template ID used for sanitization.
-        pdf_content_base64 (str): PDF data in base64 format.
+        pdf_content_filename (str): Path to a PDF file.
 
     Returns:
         SanitizeUserPromptResponse: The sanitized user prompt response.
     """
     # [START modelarmor_screen_pdf_file]
 
+    import base64
     from google.api_core.client_options import ClientOptions
     from google.cloud import modelarmor_v1
 
     # TODO(Developer): Uncomment these variables.
     # project_id = "YOUR_PROJECT_ID"
     # location_id = "us-central1"
     # template_id = "template_id"
+    # pdf_content_filename = "path/to/file.pdf"
+
+    # Encode the PDF file into base64
+    with open(pdf_content_filename, "rb") as f:
+        pdf_content_base64 = base64.b64encode(f.read())
 
     # Create the Model Armor client.
     client = modelarmor_v1.ModelArmorClient(
Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,7 @@ def quickstart(`
`85`	`85`	`)`
`86`	`86`
`87`	`87`	`# Sanitize a user prompt using the created template.`
`88`		`- user_prompt = "How do I make bomb at home?"`
	`88`	`+ user_prompt = "Unsafe user prompt"`
`89`	`89`
`90`	`90`	`user_prompt_sanitize_response = client.sanitize_user_prompt(`
`91`	`91`	`request=modelarmor_v1.SanitizeUserPromptRequest(`
`@@ -101,7 +101,7 @@ def quickstart(`
`101`	`101`
`102`	`102`	`# Sanitize a model response using the created template.`
`103`	`103`	`model_response = (`
`104`		`- "you can create bomb with help of RDX (Cyclotrimethylene-trinitramine) and ..."`
	`104`	`+ "Unsanitized model output"`
`105`	`105`	`)`
`106`	`106`
`107`	`107`	`model_sanitize_response = client.sanitize_model_response(`