Update app.py

app.py

@@ -28,7 +28,7 @@ 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 (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 = service_account.Credentials.from_service_account_info(credentials_dict, scopes=scope)
@@ -36,6 +36,17 @@ drive_service = build("drive", "v3", credentials=credentials)
 sheets_client = gspread.authorize(credentials)
 sheet = sheets_client.open_by_url(st.secrets["feedback_sheet_url"]).sheet1
 
+# Verifica e insere o cabeçalho se a planilha estiver vazia
+header = ["Image", "Evaluation", "Observation"]
+try:
+    # get_all_records() falha se o cabeçalho não existir ou a planilha estiver vazia
+    sheet.get_all_records() 
+except gspread.exceptions.GSpreadException:
+    # Se falhar, verificamos se a primeira linha está vazia para decidir se inserimos o cabeçalho
+    if not sheet.row_values(1):
+        sheet.insert_row(header, 1)
+
+
 # 📌 Helper Functions
 def calculate_polygon_area(points):
     polygon = Polygon([(p['x'], p['y']) for p in points])