Update app.py

app.py

@@ -10,9 +10,9 @@ from shapely.geometry import Polygon
 from PIL import Image
 from io import BytesIO
 from concurrent.futures import ThreadPoolExecutor
-from google.oauth2 import service_account
-from googleapiclient.discovery import build
-from googleapiclient.http import MediaIoBaseUpload
+from google.auth.transport.requests import Request
+from google.oauth2.credentials import Credentials
+from google_auth_oauthlib.flow import InstalledAppFlow
 import gspread
 import time
 
@@ -28,14 +28,38 @@ dpi_value = 300
 with st.expander("⚙️ Advanced Settings", expanded=True):
     model.confidence = st.slider("Model Confidence (%)", 20, 100, 80)
 
-# 📁 Setup Google Drive and Sheets
+# 📁 Setup Google Drive and Sheets (Híbrido: Local e Hugging Face)
 scope = ["https://www.googleapis.com/auth/drive", "https://www.googleapis.com/auth/spreadsheets"]
-credentials_dict = json.loads(st.secrets["gcp_service_account"] )
-credentials = service_account.Credentials.from_service_account_info(credentials_dict, scopes=scope)
-drive_service = build("drive", "v3", credentials=credentials)
-sheets_client = gspread.authorize(credentials)
+creds = None
+
+# Tenta carregar as credenciais a partir do segredo do Hugging Face
+# st.secrets.get( ) retorna None se o segredo não existir, evitando erros.
+token_json_secret = st.secrets.get("GOOGLE_TOKEN_JSON")
+
+if token_json_secret:
+    # Se estiver no Hugging Face, carrega as credenciais a partir do segredo
+    creds = Credentials.from_authorized_user_info(json.loads(token_json_secret), scope)
+else:
+    # Se estiver rodando localmente, usa o fluxo de arquivos token.json/credentials.json
+    if os.path.exists('token.json'):
+        creds = Credentials.from_authorized_user_file('token.json', scope)
+    
+    if not creds or not creds.valid:
+        if creds and creds.expired and creds.refresh_token:
+            creds.refresh(Request())
+        else:
+            flow = InstalledAppFlow.from_client_secrets_file('credentials.json', scope)
+            creds = flow.run_local_server(port=0)
+        # Salva o token para futuras execuções locais
+        with open('token.json', 'w') as token:
+            token.write(creds.to_json())
+
+# Constrói os serviços com as credenciais obtidas
+drive_service = build("drive", "v3", credentials=creds)
+sheets_client = gspread.authorize(creds)
 sheet = sheets_client.open_by_url(st.secrets["feedback_sheet_url"]).sheet1
 
+
 # 📌 Helper Functions
 def calculate_polygon_area(points):
     polygon = Polygon([(p['x'], p['y']) for p in points])