Update pixelconversion.py

pixelconversion.py

@@ -21,21 +21,45 @@ import db
 """### Open PDF and draw a rectangle on it using Fitz"""
 def openDrawPDF(path):
   doc = fitz.open(path)
+  out = fitz.open()  # output PDF
   page = doc[0]
-  pix = page.get_pixmap(dpi=200)  # render page to an image
+  pix = page.get_pixmap()  # render page to an image
+  pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
+  img=np.array(pl)
+
+  img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+  ratio =  pix.width / img.shape[1]  
+
+  img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+  imgBW=cv2.threshold(img, 250, 255, cv2.THRESH_BINARY_INV)[1]
+
+  contours, hierarchy = cv2.findContours(imgBW, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
+  img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
+
+  contour=contours[2]
+  perimeter = cv2.arcLength(contour, True)
+  approx = cv2.approxPolyDP(contour, 0.01* perimeter, True)
+  shape=[]
+  for point in approx:
+      x1, y1 = point[0]
+      
+      shape.append([(int(x1*ratio)),(int(y1*ratio))])
+      cv2.circle (img, (x1, y1), 5, 255, 5)
+
   rotate=0
-  if page.rect.height > page.rect.width: 
-    rect =  fitz.Rect(80,80, (page.rect.width-80),( page.rect.width+80))
+  if page.rect.height > page.rect.width:
     rectText=fitz.Rect(300, 200, (page.rect.width-80),( page.rect.width+80) )
     rotate=0
-
   else:
-    
-    rect =  fitz.Rect(80,80, (page.rect.height-80),( page.rect.height+80))
     rectText=fitz.Rect(500, 200, (page.rect.height-80),( page.rect.height+80) )
     rotate=90
 
-  annot = page.add_rect_annot(rect)
+
+  if page.rotation !=0:
+    rect =  fitz.Rect(shape[0][1],shape[0][0],shape[2][1],shape[2][0])
+  else:
+    rect= fitz.Rect(shape[0][0],shape[0][1],shape[2][0],shape[2][1])
+  annot = page.add_rect_annot( rect=rect )  # 'rectangle'
   annot.set_colors( fill=(75/255,0,130/255) ,stroke=(75/255,0,130/255) )
   annot.set_opacity(0.9)
   annot.set_border(border=None, width=0)
@@ -44,13 +68,6 @@ def openDrawPDF(path):
   text = """Scale Document"""
   annot1=page.add_freetext_annot(rectText, text, fontsize=45, fontname='helv', border_color=(1,1,1), text_color=(1,1,1), rotate=rotate, align=1)
   annot1.update()
-  perm = int(  # permissions bit flags
-    fitz.PDF_PERM_ACCESSIBILITY  #always use this
-    | fitz.PDF_PERM_PRINT # permits printing
-
-  )
-  encrypt = fitz.PDF_ENCRYPT_AES_256  # strongest algorithm
-  # Ndoc=doc.write(encryption=encrypt,permissions=perm)
 
   return doc
 """### Extract color"""