Update pixelconversion.py

pixelconversion.py

@@ -17,10 +17,12 @@ from PIL import Image
 import numpy as np
 import cv2
 import db
+import tsadropboxretrieval
 
 """### Open PDF and draw a rectangle on it using Fitz"""
-def openDrawPDF(path):
-  doc = fitz.open(path)
+def openDrawPDF(data):
+  
+  doc=fitz.open("pdf", data)
   page = doc[0]
   pix = page.get_pixmap()  # render page to an image
   pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
@@ -61,9 +63,9 @@ def openDrawPDF(path):
     rotate=90
 
   if page.rotation !=0:
-    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 )
+    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.95 )
   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 )
+    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.95 )
  
   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= page.rotation, align=1)
@@ -104,11 +106,13 @@ def DetectColor(img,color=0):
 
 """### For backend - calc area and perim"""
 
-def getAreaPerimeter(pdfpath, plan):
-  dbx=db.dropbox_connect()
-  md, res =dbx.files_download(path= pdfpath+plan)
+def getAreaPerimeter(path,name):
+  dbxTeam=tsadropboxretrieval.ADR_Access_DropboxTeam('user')
+  md, res =dbxTeam.files_download(path= path+name)
   data = res.content
   doc=fitz.open("pdf", data)
+  area=0
+  perimeter=0
   for page in doc:
       pix = page.get_pixmap(dpi=300)  # render page to an image
       pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
@@ -116,7 +120,7 @@ def getAreaPerimeter(pdfpath, plan):
       print(img.shape)
       img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
   mask=DetectColor(img,color=(73,0,130)) #detect colored rect drawn on the pdf
-  cv2.imwrite('maskk.png',mask)
+  # cv2.imwrite('maskk.png',mask)
   contours, hierarchy = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
   for contour in contours:
       area = cv2.contourArea(contour)