Marthee commited on
Commit
30ed4a5
·
verified ·
1 Parent(s): ef7e520

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +35 -12
app.py CHANGED
@@ -295,17 +295,40 @@ def stringToRGB():
295
 
296
 
297
  if vv[5][2].startswith('2.8') or vv[5][2].startswith('2.6') or vv[5][2].startswith('2.1'):
298
- pdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF= str(vv[0]))
299
- dbxTeam= tsadropboxretrieval.ADR_Access_DropboxTeam('user')
300
- md, res =dbxTeam.files_download(path=pdfpath)
301
- dataDoc = res.content
302
- doc = fitz.open("pdf",dataDoc)
303
- page=doc[0]
304
- if page.rotation!=0:
305
- page.set_rotation(0)
306
- pix = page.get_pixmap(dpi=300) # render page to an image
307
- pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
308
- arr=measureproject(vv,dataDoc,pl)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
309
  return jsonify(arr)
310
 
311
  @app.route('/2.1Trial/',methods=["POST"])
@@ -414,7 +437,7 @@ def measureproject(result,dataDoc=0,img=0,dxffile=0):
414
  # annotatedimg,pdf_document,spreadsheet_url, list1 , df_doors=doors_fasterrcnn.main_run(img,dataDoc,'separated_classes_all.pth',result[0],pdfpath, result[4]) #single_double.pth
415
  annotatedimg, pdf_document , list1, repeated_labels , not_found =Doors_Schedule.mainRun(dataDocDoorsSchedule, dataDoc, SearchArray)
416
  dbPath='/TSA JOBS/ADR Test'+pdfpath+'Measured Plan/'
417
- pdflink= tsadropboxretrieval.uploadanyFile(doc=pdf_document,path=dbPath,pdfname=result[0]) #doc=doc,pdfname=path,pdfpath=pdfpath+'Measured Plan/
418
  _, buffer = cv2.imencode('.png', annotatedimg)
419
  repeatedLabelsReturn=''
420
  NotFoundReturn=''
 
295
 
296
 
297
  if vv[5][2].startswith('2.8') or vv[5][2].startswith('2.6') or vv[5][2].startswith('2.1'):
298
+ print(f"el mawgood fe vv[0]: {vv[0]}")
299
+ arr_s = []
300
+ dataDocs = []
301
+ pls = []
302
+ for v in vv[0]:
303
+ print(f"el mawgood fe vv[0]: {vv[0]}")
304
+ arr_s = []
305
+ dataDocs = []
306
+ pls = []
307
+ for v in vv[0]:
308
+ pdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF= str(v))
309
+ dbxTeam= tsadropboxretrieval.ADR_Access_DropboxTeam('user')
310
+ md, res =dbxTeam.files_download(path=pdfpath)
311
+ dataDoc = res.content
312
+ dataDocs.append(dataDoc)
313
+ doc = fitz.open("pdf",dataDoc)
314
+ page=doc[0]
315
+ if page.rotation!=0:
316
+ page.set_rotation(0)
317
+ pix = page.get_pixmap(dpi=300) # render page to an image
318
+ pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
319
+ pls.append(pl)
320
+ arr=measureproject(vv,dataDocs,pls)
321
+ #pdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF= str(vv[0]))
322
+ #dbxTeam= tsadropboxretrieval.ADR_Access_DropboxTeam('user')
323
+ #md, res =dbxTeam.files_download(path=pdfpath)
324
+ #dataDoc = res.content
325
+ #doc = fitz.open("pdf",dataDoc)
326
+ #page=doc[0]
327
+ #if page.rotation!=0:
328
+ # page.set_rotation(0)
329
+ #pix = page.get_pixmap(dpi=300) # render page to an image
330
+ #pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
331
+ #arr=measureproject(vv,dataDoc,pl)
332
  return jsonify(arr)
333
 
334
  @app.route('/2.1Trial/',methods=["POST"])
 
437
  # annotatedimg,pdf_document,spreadsheet_url, list1 , df_doors=doors_fasterrcnn.main_run(img,dataDoc,'separated_classes_all.pth',result[0],pdfpath, result[4]) #single_double.pth
438
  annotatedimg, pdf_document , list1, repeated_labels , not_found =Doors_Schedule.mainRun(dataDocDoorsSchedule, dataDoc, SearchArray)
439
  dbPath='/TSA JOBS/ADR Test'+pdfpath+'Measured Plan/'
440
+ pdflink= tsadropboxretrieval.uploadanyFile(doc=pdf_document,path=dbPath,pdfname="combined_output.pdf") #doc=doc,pdfname=path,pdfpath=pdfpath+'Measured Plan/
441
  _, buffer = cv2.imencode('.png', annotatedimg)
442
  repeatedLabelsReturn=''
443
  NotFoundReturn=''