Update pixelconversion.py

pixelconversion.py

@@ -21,9 +21,7 @@ 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]
-  print(page.rotation)
   pix = page.get_pixmap()  # render page to an image
   pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
   img=np.array(pl)
@@ -32,26 +30,29 @@ def openDrawPDF(path):
   ratio =  pix.width / img.shape[1]
 
   imgGry = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-  imgBW=cv2.threshold(imgGry, 250, 255, cv2.THRESH_BINARY_INV)[1]
+  imgBW=cv2.threshold(imgGry, 254, 255, cv2.THRESH_BINARY_INV)[1]
+
   contours, hierarchy = cv2.findContours(imgBW, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
 
   newcontours = contours[2:]
   areas = [cv2.contourArea(c) for c in newcontours]
   max_index2 = np.argmax(areas)
-  contour=newcontours[max_index2]
-  
-  area=cv2.contourArea(contour)
-  if area >500:
-    perimeter = cv2.arcLength(contour, True)
-    approx = cv2.approxPolyDP(contour, 0.01* perimeter, True)
+  # print(max_index)
+  cnt=newcontours[max_index2]
+  contour=cnt
+
+  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
+      p1 = fitz.Point(x1*ratio,y1*ratio)
+      p1=p1*page.derotation_matrix
+      shape.append([p1[0],p1[1]])
+      # cv2.circle (img, (x1, y1), 5, 255, 5)
+  print(shape)
+  # rotate=0
   if page.rect.height > page.rect.width:
     rectText=fitz.Rect(300, 200, (page.rect.width-80),( page.rect.width+80) )
     rotate=0
@@ -63,11 +64,14 @@ def openDrawPDF(path):
     page.draw_rect([shape[3][0]-20,shape[3][1]-20,shape[1][0]+20,shape[1][1]+20],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9 )
   else:
     page.draw_rect([shape[0][0]-20,shape[0][1]-20,shape[2][0]+20,shape[2][1]+20],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9 )
+ 
   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=page.add_freetext_annot(rectText, text, fontsize=45, fontname='helv', border_color=(1,1,1), text_color=(1,1,1), rotate= page.rotation, align=1)
   annot1.update()
-  # doc.save('k.pdf')
-  return doc
+
+  doc.save('2kk.pdf')#, encryption=encrypt, permissions=perm)
+
+  return doc#,encrypt,perm
 """### Extract color"""
 
 def DetectColor(img,color=0):