Update app.py

app.py

@@ -26,53 +26,47 @@ def ensure_headers():
     try:
         sheets_service = build('sheets', 'v4', credentials=CREDENTIALS)
         
-        # Get current values
-        result = sheets_service.spreadsheets().values().get(
+        # Directly update first row with headers
+        headers = [
+            ['姓名 Name', '學號 Student ID', '購買商品 Product', 
+             '金額 Cost', '發票檔案 Receipt File', '時間 Timestamp', 'IP位址 IP Address']
+        ]
+        
+        # Update A1:G1 with headers
+        sheets_service.spreadsheets().values().update(
             spreadsheetId=SPREADSHEET_ID,
-            range='Sheet1!A1:G1'
+            range='Sheet1!A1:G1',
+            valueInputOption='RAW',
+            body={'values': headers}
         ).execute()
         
-        # If no values or empty first row, add headers
-        if 'values' not in result or not result['values']:
-            headers = [
-                ['姓名 Name', '學號 Student ID', '購買商品 Product', 
-                 '金額 Cost', '發票檔案 Receipt File', '時間 Timestamp', 'IP位址 IP Address']
-            ]
-            
-            sheets_service.spreadsheets().values().update(
-                spreadsheetId=SPREADSHEET_ID,
-                range='Sheet1!A1:G1',
-                valueInputOption='RAW',
-                body={'values': headers}
-            ).execute()
-            
-            # Format headers (make bold and center)
-            requests = [{
-                'repeatCell': {
-                    'range': {
-                        'sheetId': 0,
-                        'startRowIndex': 0,
-                        'endRowIndex': 1
-                    },
-                    'cell': {
-                        'userEnteredFormat': {
-                            'horizontalAlignment': 'CENTER',
-                            'textFormat': {
-                                'bold': True
-                            }
+        # Format headers to be bold and centered
+        requests = [{
+            'repeatCell': {
+                'range': {
+                    'sheetId': 0,
+                    'startRowIndex': 0,
+                    'endRowIndex': 1
+                },
+                'cell': {
+                    'userEnteredFormat': {
+                        'horizontalAlignment': 'CENTER',
+                        'textFormat': {
+                            'bold': True
                         }
-                    },
-                    'fields': 'userEnteredFormat(horizontalAlignment,textFormat)'
-                }
-            }]
+                    }
+                },
+                'fields': 'userEnteredFormat(horizontalAlignment,textFormat)'
+            }
+        }]
+        
+        sheets_service.spreadsheets().batchUpdate(
+            spreadsheetId=SPREADSHEET_ID,
+            body={'requests': requests}
+        ).execute()
             
-            sheets_service.spreadsheets().batchUpdate(
-                spreadsheetId=SPREADSHEET_ID,
-                body={'requests': requests}
-            ).execute()
     except Exception as e:
         print(f"Error setting headers: {str(e)}")
-
 @app.before_first_request
 def initialize():
     ensure_headers()