Update app.py

app.py

@@ -2,7 +2,7 @@ from __future__ import print_function
 from flask import Flask, render_template,request,flash , json, url_for,g , redirect , jsonify , send_file ,make_response
 import os
 import json
-import fitz 
+import fitz
 from PIL import Image
 import cv2
 import numpy as np
@@ -167,7 +167,7 @@ def getfromdropboxImg(jsdata):
 @app.route('/savebase64Img/',methods=["POST"])
 def stringToRGB():
     vv = eval(request.form.get('answers'))
-    
+    print(vv)
     if ( vv[5][2].startswith('3.2') or vv[5][2].startswith('3.3') ) :
         print('3.2 section')
         pdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF=vv[0])
@@ -192,7 +192,17 @@ def stringToRGB():
        
 
     if vv[5][2].startswith('2.8'):
-        arr=measureproject(vv)
+        pdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF= str(vv[0]))
+        dbxTeam= tsadropboxretrieval.ADR_Access_DropboxTeam('user')
+        md, res =dbxTeam.files_download(path=pdfpath)
+        dataDoc = res.content
+        doc = fitz.open("pdf",dataDoc)
+        page=doc[0]
+        if page.rotation!=0:
+            page.set_rotation(0)
+        pix = page.get_pixmap(dpi=300)  # render page to an image
+        pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
+        arr=measureproject(vv,dataDoc,pl)
     return jsonify(arr)
 
 @app.route('/2.1Trial/',methods=["POST"])
@@ -267,11 +277,11 @@ def measureproject(result,dataDoc=0,img=0,dxffile=0):
         listofmarkups=[]
         print('RATIOS=',result[4])
         # for i in range(len(result[0])):
-        dbpdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF=result[0])
-        dbxTeam= tsadropboxretrieval.ADR_Access_DropboxTeam('user')
-        md, res =dbxTeam.files_download(path=dbpdfpath)
-        dataDoc = res.content
-        annotatedimg,pdf_document,spreadsheet_url, list1 , df_doors=doors_fasterrcnn.main_run(dataDoc,'separated_classes.pth',result[0],pdfpath, result[4]) #single_double.pth
+        # dbpdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF=result[0])
+        # dbxTeam= tsadropboxretrieval.ADR_Access_DropboxTeam('user')
+        # md, res =dbxTeam.files_download(path=dbpdfpath)
+        # dataDoc = res.content
+        annotatedimg,pdf_document,spreadsheet_url, list1 , df_doors=doors_fasterrcnn.main_run(img,dataDoc,'separated_classes.pth',result[0],pdfpath, result[4]) #single_double.pth
         
         dbPath='/TSA JOBS/ADR Test'+pdfpath+'Measured Plan/' 
         # global pdflink
@@ -734,6 +744,8 @@ def cvtSummarytoXML(jsdata):
     # print(result[1])
     xmllink=google_sheet_to_xml.create_xml(documentname=result[1],dbPath=path)
     return jsonify(xmllink)
+ 
+
 #_________________________________________________________________________________________________________________________
 #_________________________________________________________________________________________________________________________