Update app.py

app.py

@@ -21,9 +21,15 @@ from werkzeug.wsgi import wrap_file
 import tameem3_2
 import pypdfium2 as pdfium
 import pixelconversion
+import urllib.request
+import base64
+import io
+from urllib.parse import unquote
+
 
 app = Flask(__name__)
 
+# out_file = open("static/data/myfile.json", "w")
 
 #retrieves project parts 
 @app.route('/getmethod/<jsdata>',methods=['GET','POST'])
@@ -73,12 +79,28 @@ def hexRGB(color):
   return color
 #_________________________________________________________________________________________________________________________
 #_________________________________________________________________________________________________________________________
+@app.route('/savebase64Img/',methods=["POST"])
+def stringToRGB():
+    name = request.get_json()
+    vv=name.get('allvalues')
+    if vv[1]==220:
+        imgdata = base64.b64decode(vv[6])
+        img=Image.open(io.BytesIO(imgdata))
+        opencv_img= cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
+    else:
+        opencv_img = plan2img('dropbox_plans/1.0/'+str(vv[0]) )
+    arr=measureproject(vv,opencv_img)
+    # # print(opencv_img)
+    # cv2.imwrite('img.png', opencv_img)
+    # return opencv_img 
+    return jsonify(arr)
+
 
 #MAIN FUNCTION -- calls python code to measure the chosen plan from the interface
-@app.route('/projecttomeasure/<jsdata>',methods=["GET","POST"])
-def measureproject(jsdata):
+# @app.route('/projecttomeasure/<jsdata>',methods=["GET","POST"])
+def measureproject(result,img):
     colorarr=[]
-    result = json.loads(jsdata) 
+    # result = json.loads(jsdata) 
     pdfpath='/'
     #################################  -1.0- #################################
     for word in result[5]:
@@ -86,13 +108,18 @@ def measureproject(jsdata):
     arr=[]
 
     if result[5][2].startswith('1.0'): #section value - 1.0 substructure- pile caps
-        img = plan2img('dropbox_plans/1.0/'+str(result[0]) )
+        # img0 = plan2img('dropbox_plans/1.0/'+str(result[0]) )
+       
+        # if (result[1]!=0):
+        # img=stringToRGB(result[6], img0.shape)
+        # print('p',result[6])
+        
 
         for item in result[2]:
-            item1 ='#'+item 
-            c=hexRGB(item1)
+            # item1 ='#'+item 
+            c=hexRGB(item)
             colorarr.append(c)
-        imgPerimeter1,image_new1,SimilarAreaDictionary , colorsUsed,spreadsheet_url, spreadsheetId,list1, pdflink, areas_Perimeters, namepathArr =pilecaps_adr.drawAllContours(img,int(result[1]),colorarr, result[3], result[4],'area', result[0],pdfpath)
+        imgPerimeter1,image_new1,SimilarAreaDictionary , colorsUsed,spreadsheet_url, spreadsheetId,list1, pdflink, areas_Perimeters, namepathArr =pilecaps_adr.drawAllContours(img,result[1],colorarr, result[3], result[4],'area', result[0],pdfpath)
         _, buffer = cv2.imencode('.png', image_new1)
         arr=[base64.b64encode(buffer).decode('utf-8'),SimilarAreaDictionary.to_dict(),spreadsheet_url , spreadsheetId,colorsUsed,list1.to_dict(), pdflink, areas_Perimeters, namepathArr]
    #################################  -3.2- #################################
@@ -115,21 +142,30 @@ def measureproject(jsdata):
 
         #let first result to be the img 
         #return results in arr=[base64.b64encode(buffer).decode('utf-8'),pdflink,spreadsheetId,spreadsheet_url] like the previous sections in the above lines 
-    return  jsonify(arr)
+    return  arr
 #_________________________________________________________________________________________________________________________
 #_________________________________________________________________________________________________________________________
 
+@app.route("/canvaspdftoimgBackground/<jsdata>",methods=["GET", "POST"])
+def pdftoimgCanvas(jsdata):
+    img=plan2img('dropbox_plans/1.0/'+jsdata)
+    # print('imgdone','dropbox_plans/1.0/'+jsdata )
+    _, buffer = cv2.imencode('.png', img)
+    arr=[base64.b64encode(buffer).decode('utf-8')  , img.shape[0],img.shape[1]]
+    return jsonify(arr)
+
+
 #pdf to img
 def plan2img(path):
     
     doc = fitz.open(path)
-    for page in doc:
-        pix = page.get_pixmap(dpi=200)  # render page to an image
-        pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
-        img=np.array(pl)
-        print(img.shape)
-        img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-    return img
+    page=doc[0]
+    pix = page.get_pixmap(dpi=200)  # render page to an image
+    pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
+    img=np.array(pl)
+    print(img.shape)
+    img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+    return img 
 #_________________________________________________________________________________________________________________________
 #_________________________________________________________________________________________________________________________