Update app.py

app.py

@@ -395,6 +395,30 @@ template_sheet_id = '1i5mrmlTs5sPWtx2mBtc_f-zm1D3x21r1T_77guki8_8'
 @app.post("/edit-spreadsheet")
 async def edit_spreadsheet(target_sheet_id: str = Form(...)):
     try:
+        # Retrieve the target spreadsheet's metadata
+        target_spreadsheet = sheet_service.spreadsheets().get(spreadsheetId=target_sheet_id).execute()
+
+        # Collect all sheet IDs in the target spreadsheet
+        target_sheet_ids = [sheet['properties']['sheetId'] for sheet in target_spreadsheet.get('sheets', [])]
+
+        # Prepare batch update request to delete existing sheets
+        requests = []
+        for sheet_id in target_sheet_ids:
+            requests.append({
+                "deleteSheet": {
+                    "sheetId": sheet_id
+                }
+            })
+
+        if requests:
+            # Execute batch update to delete sheets
+            batch_update_request = {"requests": requests}
+            sheet_service.spreadsheets().batchUpdate(
+                spreadsheetId=target_sheet_id,
+                body=batch_update_request
+            ).execute()
+            logging.info(f"Deleted existing sheets in target spreadsheet {target_sheet_id}.")
+
         # Retrieve the template spreadsheet's metadata
         template_spreadsheet = sheet_service.spreadsheets().get(spreadsheetId=template_sheet_id).execute()
 
@@ -410,7 +434,28 @@ async def edit_spreadsheet(target_sheet_id: str = Form(...)):
             )
             copy_response = copy_request.execute()
 
-            logging.info(f"Sheet {sheet['properties']['title']} copied to {target_sheet_id} with new sheet ID {copy_response['sheetId']}.")
+            # Get the copied sheet ID
+            copied_sheet_id = copy_response['sheetId']
+
+            # Rename the copied sheet to remove "Copy of"
+            copied_sheet_title = sheet['properties']['title'].replace("Copy of ", "")
+            rename_request = {
+                "updateSheetProperties": {
+                    "properties": {
+                        "sheetId": copied_sheet_id,
+                        "title": copied_sheet_title
+                    },
+                    "fields": "title"
+                }
+            }
+
+            # Execute rename request
+            sheet_service.spreadsheets().batchUpdate(
+                spreadsheetId=target_sheet_id,
+                body={"requests": [rename_request]}
+            ).execute()
+
+            logging.info(f"Sheet {sheet['properties']['title']} copied and renamed to {copied_sheet_title} in {target_sheet_id}.")
 
         logging.info(f"All sheets from template {template_sheet_id} copied to {target_sheet_id}.")
         return {"message": "Spreadsheet copied successfully"}