Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
| 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
|