Update app.py

app.py

@@ -20,7 +20,10 @@ from db import dropbox_upload_file
 from pathlib import Path
 from PreprocessingFoundation import rmv_text,rmv_dashedLines
 
-
+from __future__ import print_function
+from googleapiclient.discovery import build 
+from google.oauth2 import service_account
+import pygsheets
 #############################################################################################
 
 '''push output to dropbox'''
@@ -679,7 +682,7 @@ def FillDictionary(green2,SimilarAreaDictionary,img,number,num1,num2,flag,finalC
 
 
 
-def drawAllContours(green2,img,number,finalColorArray,num1,num2,flag):
+def drawAllContours(plan,green2,img,number,finalColorArray,num1,num2,flag):
   # colors=[color,color1,color2,color3,color4,color5,color6,color7]
   # img=np.array(img)
   areasinImage=[]
@@ -895,11 +898,185 @@ def drawAllContours(green2,img,number,finalColorArray,num1,num2,flag):
   return  imgPerimeter1,image_new1,SimilarAreaDictionary , s
 # drawAllContours(img,0,[],1.11,25579,'area') #,[190,47,250] ,[47,251,255] ,[80,240,15],[253,163,40]
 # imgPerimeter1,image_new1,Dictionary=drawAllContours(img,220,[47,251,255],[251,163,47],1.105,27233.5,'area')
+###############################################
+    ##Google Sheets Legend
+def getTitle(path):
+  planName= path.split("/")[-1].split('.')
+  LegendName='Legend of ' + str(planName[0]) + ' plan'
+  return LegendName
+    
+def getTitle(path):
+  planName= path.split("/")[-1].split('.')
+  LegendName='Legend of ' + str(planName[0]) + ' plan'
+  return LegendName
+    
+def getdropdownValues(gc,spreadsheet_service,spreadsheetid):
+  dropdownValues=[]
+  ws=gc.open_by_key('1lIIpv6jR0aSWQRtc-TPYn2Lxh17U6H_XrZE4EjTK69Y')  ## spreadsheet containing mc-t names 
+  worksheet = ws.worksheet(0)
+  response = spreadsheet_service.spreadsheets().get(
+        spreadsheetId=spreadsheetid, fields='*',
+        ranges='A2:A60',includeGridData=True).execute()
+  r=list(response['sheets'][0]['data'][0]['rowData'][0]['values'][0])
+  print(r)
+  if 'dataValidation' in r:
+    print('yes')
+    colvals= response['sheets'][0]['data'][0]['rowData'][0]['values'][0]['dataValidation']
+    colvalsList=list(colvals.items())
+    print(colvalsList[0][1])
+    lengthVals=len(colvalsList[0][1]['values'])
+    for i in range(lengthVals):
+      dictVal=(colvalsList[0][1]['values'][i].values())
+      # val=[*dictVal]
+     
+      dropdownValues.append(*dictVal)
+    print(dropdownValues)
+  worksheet.update_col(index=1, values=dropdownValues)
+  return dropdownValues
+
+def retrieveMCCol(gc):
+  ws=gc.open_by_key('1lIIpv6jR0aSWQRtc-TPYn2Lxh17U6H_XrZE4EjTK69Y')
+  worksheet = ws.worksheet(0)
+  mcT_Names=worksheet.get_col(1)
+  newMcTNames=[]
+  for i in mcT_Names:
+    if i != '':
+      newMcTNames.append(i)
+  return newMcTNames
+
+def legendGoogleSheets(SimilarAreaDictionary,colorsUsed):
+  # authorize uing json file 
+  # SimilarAreaDictionary.drop(['Rounded', 'Width','Height','R','G','B'], axis=1, inplace=True)
+  SCOPES = [
+  'https://www.googleapis.com/auth/spreadsheets',
+  'https://www.googleapis.com/auth/drive'
+  ]
+  credentials = service_account.Credentials.from_service_account_file('/content/credentials.json', scopes=SCOPES)
+  spreadsheet_service = build('sheets', 'v4', credentials=credentials)
+  drive_service = build('drive', 'v3', credentials=credentials)
+  gc = pygsheets.authorize(custom_credentials=credentials, client_secret='/content/credentials.json')
+
+
+########
+  legendTitle=getTitle(path)
+  titles=gc.spreadsheet_titles()
+  # print(titles)
+  if legendTitle in  titles:
+    print('found sheet ')
+  else:
+  ####### create new sheet
+    print('creating new sheeet')
+
+    
+    LegendName=getTitle(path)
+    spreadsheet_details = {
+    'properties': {
+        'title': LegendName
+        }
+    }
+    sheet = spreadsheet_service.spreadsheets().create(body=spreadsheet_details,
+                                    fields='spreadsheetId').execute()
+                                        
+    spreadsheetId = sheet.get('spreadsheetId')
+    # print('Spreadsheet ID: {0}'.format(spreadsheetId))
+    permission1 = {
+    'type': 'anyone',
+    'role': 'writer',
+    # 'emailAddress': 'marthe.adr@gmail.com'
+    }
+    drive_service.permissions().create(fileId=spreadsheetId, body=permission1).execute()
+  ###################3
+  #open sheet 
+  spreadsheetId='1dtDi_6-g3jkn6ePVlzM6PM3FE8wIHzyL2Rt4ksH59SE'
+  ws=gc.open_by_key(spreadsheetId)
+  worksheet = ws.worksheet(0)
+
+  #get lengths of df
+  columnsLen=len(SimilarAreaDictionary.columns.values.tolist()) #kam column -- last col = columnsLen+1 3shan base0
+  lastUsedCol=columnsLen+1
+ 
+  rowsLen=len(SimilarAreaDictionary.values.tolist()) #kam row -- last row = rowsLen +1
+  lastUsedRow=rowsLen+1
+  #append to googlesheet
+  worksheet.update_row(1,SimilarAreaDictionary.columns.values.tolist() ,col_offset=1)
+  worksheet.append_table(SimilarAreaDictionary.values.tolist(), dimension='ROWS' )
+  #names
+  worksheet.update_col(index=1,values=['MC-Template Names'],row_offset=0)
+
+  #apply 'A1' notation
+  firstcell=worksheet.cell((2,1)) #row,col
+  firstcellNotation=str(firstcell.address.label)
+
+  lastcell=worksheet.cell((rowsLen+1,1)) #row,col
+  lastcellNotation=str(lastcell.address.label)
+
+  lastcolumn=worksheet.cell((1,lastUsedCol)) #row,col
+  lastcolumnNotation=str(lastcolumn.address.label)
+
+  #dropdowns - 
+  mcT_Names=retrieveMCCol(gc)
+  worksheet.set_data_validation(firstcellNotation,lastcellNotation, condition_type='ONE_OF_LIST', condition_values=mcT_Names, showCustomUi=True)
+  #format first row as bold
+  model_cell =worksheet.cell('A1')
+  model_cell.set_text_format('bold', True)
+  pygsheets.DataRange('A1',lastcolumnNotation, worksheet=worksheet).apply_format(model_cell)
+  worksheet.adjust_column_width(start=1,end=lastUsedCol)
+
+
+  sheetId = '0'  # Please set sheet ID.
+  for i in range(len(colorsUsed)):
+
+    print(colorsUsed[i])
+    r,g,b=colorsUsed[i]
+    body = {
+      "requests": [
+        { 
+          "updateCells": {
+            "range": {
+              "sheetId": sheetId,
+              "startRowIndex": i+1,
+              # "endRowIndex":4 ,
+              "startColumnIndex":1,
+
+              # "endColumnIndex": 0
+            },
+
+            "rows": [
+              {
+                "values": [ 
+                  {
+                    "userEnteredFormat": {
+                      "backgroundColor": {
+                    
+                              "red": r/255,
+                              "green": g/255,
+                              "blue": b/255,
+                              # "alpha": 0.8
+                                   
+                      }
+                      
+                    }
+                  }
+                ]
+              }
+            ],
+            "fields": "userEnteredFormat.backgroundColor",
+        
+          }
+        }
+      ]
+    }
+    res =  spreadsheet_service.spreadsheets().batchUpdate(spreadsheetId=spreadsheetId, body=body).execute()
+  spreadsheet_url = "https://docs.google.com/spreadsheets/d/%s" % spreadsheetId
+  print(spreadsheet_url)
+
+  return gc,spreadsheet_service,spreadsheetId
+gc,spreadsheet_service,spreadsheetId=legendGoogleSheets(SimilarAreaDictionary,colorsUsed)
 
 
-def MainFunc(green2,img,dp,finalColorArray,number,num1,num2,flag):
+def MainFunc(plan,green2,img,dp,finalColorArray,number,num1,num2,flag):
 
-  imgPerimeter1,image_new1,Df2,s=drawAllContours(green2,img,number,finalColorArray,num1,num2,flag)
+  imgPerimeter1,image_new1,Df2,s=drawAllContours(plan,green2,img,number,finalColorArray,num1,num2,flag)
   # x = path.split("/")
   # x=x.pop()
   Df2=Df2.T
@@ -920,7 +1097,8 @@ def MainFunc(green2,img,dp,finalColorArray,number,num1,num2,flag):
   # display(Df2)
   return Df2, imgPerimeter1,image_new1 ,s 
 
-def PickColorContours(dp,img,radioButton,radioButton1, color,color1,color2,color3,color4,color5,color6,color7,color8,num1=0,num2=0):
+def 
+Contours(plan,dp,img,radioButton,radioButton1, color,color1,color2,color3,color4,color5,color6,color7,color8,num1=0,num2=0):
   print(type(img))
   green2=allpreSteps(img)
   
@@ -936,30 +1114,30 @@ def PickColorContours(dp,img,radioButton,radioButton1, color,color1,color2,color
     if radioButton=="Measure Full Image" :
      
       if radioButton1=='Area':
-        imgPerimeter1,image_new1,df ,s =drawAllContours(green2,img,0,finalColorArray,num1,num2,'area')
+        imgPerimeter1,image_new1,df ,s =drawAllContours(plan,green2,img,0,finalColorArray,num1,num2,'area')
         return image_new1
       else:
-        imgPerimeter1,image_new1,df ,s =drawAllContours(green2,img,0,finalColorArray,num1,num2,'perimeter')
+        imgPerimeter1,image_new1,df ,s =drawAllContours(plan,green2,img,0,finalColorArray,num1,num2,'perimeter')
         return imgPerimeter1
     else:   
       if radioButton1=='Area':
-        imgPerimeter1,image_new1,df ,s =drawAllContours(green2,img,220,finalColorArray,num1,num2,'area') #rgbValue
+        imgPerimeter1,image_new1,df ,s =drawAllContours(plan,green2,img,220,finalColorArray,num1,num2,'area') #rgbValue
         return image_new1
       else:
-        imgPerimeter1,image_new1,df ,s =drawAllContours(green2,img,220,finalColorArray,num1,num2,'perimeter') #rgbValue
+        imgPerimeter1,image_new1,df ,s =drawAllContours(plan,green2,img,220,finalColorArray,num1,num2,'perimeter') #rgbValue
         return imgPerimeter1
   else:
     if radioButton=="Measure Full Image":
       if radioButton1=='Area':
-        Dictionary, imgPerimeter1,image_new1 ,s =MainFunc(green2,img,dp,finalColorArray,0,num1,num2,'area')
+        Dictionary, imgPerimeter1,image_new1 ,s =MainFunc(plan,green2,img,dp,finalColorArray,0,num1,num2,'area')
       else:
-        Dictionary, imgPerimeter1,image_new1 ,s =MainFunc(green2,img,dp,finalColorArray,0,num1,num2,'perimeter')
+        Dictionary, imgPerimeter1,image_new1 ,s =MainFunc(plan,green2,img,dp,finalColorArray,0,num1,num2,'perimeter')
 
     else:
       if radioButton1=='Area':
-        Dictionary, imgPerimeter1,image_new1,s =MainFunc(green2,img,dp,finalColorArray,220,num1,num2,'area')
+        Dictionary, imgPerimeter1,image_new1,s =MainFunc(plan,green2,img,dp,finalColorArray,220,num1,num2,'area')
       else:
-        Dictionary, imgPerimeter1,image_new1,s =MainFunc(green2,img,dp,finalColorArray,220,num1,num2,'perimeter')
+        Dictionary, imgPerimeter1,image_new1,s =MainFunc(plan,green2,img,dp,finalColorArray,220,num1,num2,'perimeter')
 
     Dictionary=Dictionary.T
     s=s.to_html()
@@ -989,7 +1167,8 @@ def getMeasurement(plan,SaveOP,check1, dp,in2,in3,in4,in5,in6,in7,in8,in9,in10,i
 
 
     if num1!=0 and num2!=0:
-       imgPerimeter1,Dictionary , s= PickColorContours(dp,img,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,in12,num1,num2)
+       imgPerimeter1,Dictionary , s= 
+        Contours(plan1,dp,img,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,in12,num1,num2)
        imgPerimeter1=  cv2.cvtColor(imgPerimeter1, cv2.COLOR_RGB2BGR)
        if SaveOP:
             pushToDropbox(plan,imgPerimeter1,Dictionary)
@@ -997,13 +1176,15 @@ def getMeasurement(plan,SaveOP,check1, dp,in2,in3,in4,in5,in6,in7,in8,in9,in10,i
             exportToExcel(plan,Dictionary)
        return imgPerimeter1,Dictionary , s
     else:
-      area=PickColorContours(dp,img,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,in12)
+      area=
+        Contours(plan1,dp,img,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,in12)
       area = cv2.cvtColor(area, cv2.COLOR_BGR2RGB)
       # if SaveOP:
       #   pushToDropbox(plan, area)
       return area
         # else:
-          # areaPerimeterImg,df=PickColorContours(dp,img,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,num1,num2)
+          # areaPerimeterImg,df=
+        Contours(dp,img,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,num1,num2)
     # if SaveOP:
     #     pushToDropbox(plan,area,perim,df)
 
@@ -1106,9 +1287,11 @@ with gr.Blocks(css="#search {background: orangered}") as mainBlock:
   show_button.click(fn=getMeasurement, inputs=[ drop, check ,check1, dp,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,in12],outputs=img1)
   buttonSaveDf.click(fn=saveExcel,inputs=[out2,project])
     # clr_btn.click(fn=clear,outputs=[project,radio_button,check,drop])
-  # btn1.click(fn=PickColorContours, inputs=[dp,in1,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11], outputs=out1)
+  # btn1.click(fn=
+    Contours, inputs=[dp,in1,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11], outputs=out1)
   #secoond part 
-  # btn.click(fn=PickColorContours, inputs=[dp,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,num1,num2], outputs=outputs1)
+  # btn.click(fn=
+Contours, inputs=[dp,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,num1,num2], outputs=outputs1)
   btn.click(fn=getMeasurement, inputs=[ drop, check,check1 ,dp,in2,in3,in4,in5,in6,in7,in8,in9,in10,in11,in12,num1,num2],outputs=[out1,out2,out3])
   project.change(fn=update_dropdown, inputs=[project], outputs=drop)