Update pixelconversion.py

pixelconversion.py

@@ -18,21 +18,64 @@ import numpy as np
 import cv2
 import db
 import tsadropboxretrieval
+import fitz
+from io import BytesIO
+
+################################################
+### Check if page is visually rotated or not 
+
+def is_content_rotated(mask):
+      contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+      # Check the orientation of the purple shape 
+      for contour in contours:
+          rect = cv2.minAreaRect(contour)
+          angle = rect[-1]
+          print(angle)
+          if angle != 0.0 and angle != 90.0 and angle != 180.0:
+              print("Page content appears visually rotated.")
+              return True
+      print("Page content does not appear visually rotated.")
+      return False
+
+def drawisrotated(data):
+  docPath = fitz.open("pdf",data) #dropbox path
+  pageDocPath=docPath[0]
+  if pageDocPath.rotation !=0:
+   pageDocPath.draw_rect([250,250,pageDocPath.mediabox.width-500,pageDocPath.mediabox.height-500],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9)
+  else:
+   pageDocPath.draw_rect([0+250,0+250,pageDocPath.mediabox.width-500,pageDocPath.mediabox.height-500],  color =(75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9)
+  
+  pix = pageDocPath.get_pixmap()
+  image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
+  img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
+  mask=DetectColor(img,color=(75,0,130))
+  bool_T_F=is_content_rotated(mask)
+  print(bool_T_F)
+  if bool_T_F==True:
+    if pageDocPath.mediabox.height >pageDocPath.mediabox.width:
+      rectText=fitz.Rect(300, 200, (pageDocPath.mediabox.width-80),( pageDocPath.mediabox.width+80) )
+    else:
+      rectText=fitz.Rect(500, 200, (pageDocPath.mediabox.height-80),( pageDocPath.mediabox.height+80) )
+
+    text = """Scale Document"""
+    annot1=pageDocPath.add_freetext_annot(rectText, text, fontsize=45, fontname='helv', border_color=(1,1,1), text_color=(1,1,1), rotate= pageDocPath.rotation, align=1)
+    annot1.update()
+    return docPath
+  else:
+    doc=openDrawPDF(data)
+    return doc
+
 
 """### Open PDF and draw a rectangle on it using Fitz"""
 def openDrawPDF(data):
   
   doc=fitz.open("pdf", data)
   page = doc[0]
-
-
+  
   if page.rect.height > page.rect.width:
     rectText=fitz.Rect(300, 200, (page.rect.width-80),( page.rect.width+80) )
-    rotate=0
   else:
     rectText=fitz.Rect(500, 200, (page.rect.height-80),( page.rect.height+80) )
-    rotate=90
-
   if page.rotation !=0:
     page.draw_rect([0+10,0+10,page.rect.height-10,page.rect.width-10],  color = (75/255,0,130/255), width = 1,fill=(75/255,0,130/255),fill_opacity=0.9 )
   else: