echo "[START text_gen_multimodal_two_pdf_inline]"
# Use a temporary file to hold the base64 encoded pdf data

PDF_PATH_1=${MEDIA_DIR}/test_1.pdf
PDF_PATH_2=${MEDIA_DIR}/test_2.pdf

TEMP_1_B64=$(mktemp)
trap 'rm -f "$TEMP_1_B64"' EXIT
base64 $B64FLAGS $PDF_PATH_1 > "$TEMP_1_B64"

TEMP_2_B64=$(mktemp)
trap 'rm -f "$TEMP_2_B64"' EXIT
base64 $B64FLAGS $PDF_PATH_2 > "$TEMP_2_B64"

# Use a temporary file to hold the JSON payload
TEMP_JSON=$(mktemp)
trap 'rm -f "$TEMP_JSON"' EXIT

cat > "$TEMP_JSON" << EOF
{
  "contents": [{
    "role": "user",
    "parts":[
      {"text": "Extract the pet names, type and ages from these documents."},
      {
        "inline_data": {
          "mime_type":"application/pdf",
          "data": "$(cat "$TEMP_1_B64")"
        },
        "inline_data": {
          "mime_type":"application/pdf",
          "data": "$(cat "$TEMP_2_B64")"
        }
      }
    ]
  }],
  "system_instruction": {
    "parts": [
      {"text": "Extract the pet names and ages from these documents and return them in the following JSON format:

                Pet = {\"name\": str, \"type\": str, \"age\": int}
                Return: list[Pet]"
      }
    ]
  },
  "generation_config": {
    "temperature": 0.2,
    "top_p": 0.95,
    "top_k": 40,
    "max_output_tokens": 1000,
    "response_mime_type": "application/json"
  }
}
EOF

curl "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key=$GOOGLE_API_KEY" \
    -H 'Content-Type: application/json' \
    -X POST \
    -d "@$TEMP_JSON" 2> /dev/null