Update main.py

main.py

@@ -6,6 +6,7 @@ app = FastAPI()
 # Constants
 ACCESS_TOKEN = "2585415_877e3999ec927cff5d896269c348106949c50f9b8ad54fdaeac07ad00687ce3e"
 TEAM_ID = "9018441024"
+GOOGLE_SHEET_WEBHOOK = "https://script.google.com/macros/s/AKfycbwOqg1PNYyHTh8OfiaHy1ROY0GI3s5Emf4SYA7eiNz-vPrKNwS3CBZsI_vhzTM8KDfm/exec"
 
 # Video Contractor Settings
 VIDEO_FOLDER_ID = "90182650380"
@@ -34,31 +35,52 @@ def create_list_from_template(folder_id, template_id, list_name):
         result = response.json()
         list_id = result.get("id")
         list_url = f"https://app.clickup.com/{TEAM_ID}/v/li/{list_id}"
-        return {"message": "✅ List created successfully.", "list_url": list_url}
+        return {"success": True, "list_url": list_url}
     else:
-        return {
-            "message": "❌ Failed to create list.",
-            "status_code": response.status_code,
-            "error": response.text
-        }
+        return {"success": False, "status_code": response.status_code, "error": response.text}
+
+def send_to_google_sheet(name, email, contractor_type):
+    payload = {
+        "name": name,
+        "email": email,
+        "type": contractor_type
+    }
+    try:
+        sheet_response = requests.post(GOOGLE_SHEET_WEBHOOK, json=payload)
+        return sheet_response.json()
+    except Exception as e:
+        return {"error": str(e)}
 
 @app.post("/video")
 async def create_video_contractor_list(request: Request):
     data = await request.json()
     list_name = data.get("listName")
+    email = data.get("email")
 
-    if not list_name:
-        return {"error": "Missing 'listName' in request"}
+    if not list_name or not email:
+        return {"error": "Missing 'listName' or 'email' in request"}
 
-    return create_list_from_template(VIDEO_FOLDER_ID, VIDEO_TEMPLATE_ID, list_name)
+    result = create_list_from_template(VIDEO_FOLDER_ID, VIDEO_TEMPLATE_ID, list_name)
 
+    if result.get("success"):
+        sheet_result = send_to_google_sheet(list_name, email, "video")
+        return {"list_url": result.get("list_url"), "sheet_result": sheet_result}
+    else:
+        return result
 
 @app.post("/image")
 async def create_image_contractor_list(request: Request):
     data = await request.json()
     list_name = data.get("listName")
+    email = data.get("email")
+
+    if not list_name or not email:
+        return {"error": "Missing 'listName' or 'email' in request"}
 
-    if not list_name:
-        return {"error": "Missing 'listName' in request"}
+    result = create_list_from_template(IMAGE_FOLDER_ID, IMAGE_TEMPLATE_ID, list_name)
 
-    return create_list_from_template(IMAGE_FOLDER_ID, IMAGE_TEMPLATE_ID, list_name)
+    if result.get("success"):
+        sheet_result = send_to_google_sheet(list_name, email, "image")
+        return {"list_url": result.get("list_url"), "sheet_result": sheet_result}
+    else:
+        return result