Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -41,7 +41,7 @@ from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
 from PIL import Image, ImageDraw
 import torchvision.transforms.functional as F
 import matplotlib.pyplot as plt
-
+import google_sheet_Legend
 """# updated for (fullpath, pdf_name)
 
 
@@ -166,7 +166,7 @@ def add_annotations_to_pdf(image, pdf_name, annotation_s, annotation_d):
     # pdf_document.save(output_pdf_path)
     # pdf_document.close()
 
-def main_run(pdf_fullpath, weights_path, pdf_name):
+def main_run(pdf_fullpath, weights_path, pdf_name,pdfpath): ####pdf_fullpath here is the data and not the path 
     img_pillow = convert2pillow(pdf_fullpath)
     new_image = img_pillow.resize((2384, 1684))
     
@@ -214,8 +214,10 @@ def main_run(pdf_fullpath, weights_path, pdf_name):
     pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
     img=np.array(pl)
     annotatedimg = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-    
-    return annotatedimg,pdf_document
+
+    df_doors = df_doors.fillna(' ')
+    gc,spreadsheet_service,spreadsheetId, spreadsheet_url , namepathArr=google_sheet_Legend.legendGoogleSheets(df_doors , pdf_name,pdfpath)
+    return annotatedimg, pdf_document , spreadsheet_url
 
 # model_path =  '/content/drive/MyDrive/combined.pth'
 # #pdf_name = data