Marthee commited on
Commit
05c4500
·
verified ·
1 Parent(s): af25a54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -211,19 +211,23 @@ def measureproject(result,dataDoc=0,img=0,dxffile=0):
211
  ################################# -2.8- #################################
212
  elif result[5][2].startswith('2.8'): #section value - 2.8 floor finishes
213
  #vv[0] array of pdf names
 
 
214
  for i in range(len(result[0])):
215
  dbpdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF=result[0][i])
216
  dbxTeam= tsadropboxretrieval.ADR_Access_DropboxTeam('user')
217
  md, res =dbxTeam.files_download(path=dbpdfpath)
218
  dataDoc = res.content
219
  annotatedimg,pdf_document=doors_fasterrcnn.main_run(dataDoc,'combined.pth',result[0][i])
220
-
221
  dbPath='/TSA JOBS/ADR Test'+pdfpath+'Measured Plan/'
222
  pdflink= tsadropboxretrieval.uploadanyFile(doc=pdf_document,path=dbPath,pdfname='annotated-'+result[0][i]) #doc=doc,pdfname=path,pdfpath=pdfpath+'Measured Plan/'
223
  print(pdflink)
 
224
 
225
  _, buffer = cv2.imencode('.png', annotatedimg)
226
- arr=[ base64.b64encode(buffer).decode('utf-8'),pdflink]# , spreadsheetId, spreadsheet_url , list1.to_dict()]
 
227
 
228
  ################################# -2.2- #################################
229
  # elif result[5][2].startswith('2.2'): #section value - 2.2 rc slabs
 
211
  ################################# -2.8- #################################
212
  elif result[5][2].startswith('2.8'): #section value - 2.8 floor finishes
213
  #vv[0] array of pdf names
214
+ imgss=[]
215
+ links=[]
216
  for i in range(len(result[0])):
217
  dbpdfpath,pdflink=tsadropboxretrieval.getPathtoPDF_File(nameofPDF=result[0][i])
218
  dbxTeam= tsadropboxretrieval.ADR_Access_DropboxTeam('user')
219
  md, res =dbxTeam.files_download(path=dbpdfpath)
220
  dataDoc = res.content
221
  annotatedimg,pdf_document=doors_fasterrcnn.main_run(dataDoc,'combined.pth',result[0][i])
222
+
223
  dbPath='/TSA JOBS/ADR Test'+pdfpath+'Measured Plan/'
224
  pdflink= tsadropboxretrieval.uploadanyFile(doc=pdf_document,path=dbPath,pdfname='annotated-'+result[0][i]) #doc=doc,pdfname=path,pdfpath=pdfpath+'Measured Plan/'
225
  print(pdflink)
226
+ links.append(pdflink)
227
 
228
  _, buffer = cv2.imencode('.png', annotatedimg)
229
+ imgss.append(base64.b64encode(buffer).decode('utf-8'))
230
+ arr=[imgss ,links]# , spreadsheetId, spreadsheet_url , list1.to_dict()]
231
 
232
  ################################# -2.2- #################################
233
  # elif result[5][2].startswith('2.2'): #section value - 2.2 rc slabs