Update app.py

app.py

@@ -18,7 +18,7 @@ import time
 
 # 🔥 Initialize Roboflow
 API_KEY = st.secrets["roboflow_api_key"]
-rf = roboflow.Roboflow(api_key=API_KEY )
+rf = roboflow.Roboflow(api_key=API_KEY  )
 project = rf.workspace(st.secrets["roboflow_workspace"]).project(st.secrets["roboflow_project"])
 model = project.version(st.secrets["roboflow_version"]).model
 model.confidence = 80
@@ -30,7 +30,7 @@ with st.expander("⚙️ Advanced Settings", expanded=True):
 
 # 📁 Setup Google Drive and Sheets (seção aprimorada)
 scope = ["https://www.googleapis.com/auth/drive", "https://www.googleapis.com/auth/spreadsheets"]
-credentials_dict = json.loads(st.secrets["gcp_service_account"] )
+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)
@@ -63,29 +63,48 @@ def safe_predict(image_path):
 def resize_image(image):
     return image.resize((640, 640))
 
+# --- CÓDIGO CORRIGIDO ---
 def upload_to_drive(image_bytes, filename, folder_id):
     media = MediaIoBaseUpload(image_bytes, mimetype='image/png')
     drive_service.files().create(
         body={"name": filename, "parents": [folder_id]},
         media_body=media,
-        fields='id'
+        fields='id',
+        supportsAllDrives=True  # <-- CORREÇÃO APLICADA
     ).execute()
 
+# --- CÓDIGO CORRIGIDO ---
 def find_or_create_folder(folder_name, parent=None):
     query = f"name='{folder_name}' and mimeType='application/vnd.google-apps.folder' and trashed=false"
     if parent:
         query += f" and '{parent}' in parents"
-    results = drive_service.files().list(q=query, spaces='drive', fields='files(id, name)').execute()
+    
+    # Adiciona os parâmetros para buscar em todos os drives/pastas compartilhadas
+    results = drive_service.files().list(
+        q=query, 
+        spaces='drive', 
+        fields='files(id, name)',
+        supportsAllDrives=True,  # <-- CORREÇÃO APLICADA
+        includeItemsFromAllDrives=True  # <-- CORREÇÃO APLICADA
+    ).execute()
+    
     folders = results.get('files', [])
     if folders:
         return folders[0]['id']
+    
     file_metadata = {
         'name': folder_name,
         'mimeType': 'application/vnd.google-apps.folder'
     }
     if parent:
         file_metadata['parents'] = [parent]
-    file = drive_service.files().create(body=file_metadata, fields='id').execute()
+        
+    # Adiciona o parâmetro para criar a pasta em um ambiente compartilhado
+    file = drive_service.files().create(
+        body=file_metadata, 
+        fields='id',
+        supportsAllDrives=True  # <-- CORREÇÃO APLICADA
+    ).execute()
     return file.get('id')
 
 def get_image_bytes(image):