Update app.py

app.py

@@ -10,7 +10,7 @@ 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 google.oauth2.credentials import Credentials
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaIoBaseUpload
 import gspread
@@ -28,10 +28,16 @@ dpi_value = 300
 with st.expander("⚙️ Advanced Settings", expanded=True):
     model.confidence = st.slider("Model Confidence (%)", 20, 100, 80)
 
-# 📁 Setup Google Drive e Sheets
+# 📁 Setup Google Drive e Sheets com OAuth 2.0
 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)
+credentials = Credentials(
+    token=None,
+    refresh_token=st.secrets["GOOGLE_DRIVE_REFRESH_TOKEN"],
+    token_uri="https://oauth2.googleapis.com/token",
+    client_id=st.secrets["GOOGLE_DRIVE_CLIENT_ID"],
+    client_secret=st.secrets["GOOGLE_DRIVE_CLIENT_SECRET"],
+    scopes=scope
+)
 drive_service = build("drive", "v3", credentials=credentials)
 sheets_client = gspread.authorize(credentials)
 sheet = sheets_client.open_by_url(st.secrets["feedback_sheet_url"]).sheet1
@@ -222,7 +228,7 @@ if results:
         sheet.append_row(row)
 
         if avaliacao in ["Acceptable", "Bad", "No segmentation"]:
-            sufixo = "aceitavel" if avaliacao == "Acceptable" else "ruim" if avaliacao == "Bad" else "sem_segmentacao"
+            sufixo = "aceitavel" if avaliacao == "Acceptable" else "ruim" if avaliacao == "Bad" else "No segmentation"
             parent_folder = find_or_create_folder("Feedback Segmentacoes")
             subfolder = find_or_create_folder(imagem_escolhida.replace(".png", ""), parent_folder)
 
@@ -245,4 +251,4 @@ if results:
                             upload_to_drive(buffer, f"{nome}_{sufixo}.png", subfolder)
                     break
 
-        st.success("✅ Feedback saved successfully!")
+        st.success("✅ Feedback saved successfully!")