Update main.py

main.py

@@ -5,53 +5,60 @@ app = FastAPI()
 
 # Constants
 ACCESS_TOKEN = "2585415_877e3999ec927cff5d896269c348106949c50f9b8ad54fdaeac07ad00687ce3e"
-FOLDER_ID = "90182650380"
-TEMPLATE_ID = "t-901804880155"   # No "t-" prefix
-TEAM_ID = "9018441024"         # This is the ClickUp "Team ID" shown in the URL
+TEAM_ID = "9018441024"
 
-@app.post("/")
-async def create_list_from_template(request: Request):
-    data = await request.json()
-    list_name = data.get("listName")
-    contractor_type = data.get("contractorType", "").lower()
+# Video Contractor Settings
+VIDEO_FOLDER_ID = "90182650380"
+VIDEO_TEMPLATE_ID = "t-901804880155"
 
-    if not list_name:
-        return {"error": "Missing 'listName' in request"}
-    if not contractor_type:
-        return {"error": "Missing 'contractorType' in request"}
-
-    if contractor_type == "video":
-        url = f"https://api.clickup.com/api/v2/folder/{FOLDER_ID}/list_template/{TEMPLATE_ID}"
-        headers = {
-            "Authorization": ACCESS_TOKEN,
-            "Content-Type": "application/json"
-        }
-        payload = {
-            "name": list_name,
-            "options": {
-                "return_immediately": True
-            }
+# Image Contractor Settings
+IMAGE_FOLDER_ID = "90182896356"
+IMAGE_TEMPLATE_ID = "t-901805178757"
+
+def create_list_from_template(folder_id, template_id, list_name):
+    url = f"https://api.clickup.com/api/v2/folder/{folder_id}/list_template/{template_id}"
+    headers = {
+        "Authorization": ACCESS_TOKEN,
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "name": list_name,
+        "options": {
+            "return_immediately": True
         }
+    }
 
-        response = requests.post(url, headers=headers, json=payload)
-
-        if response.status_code == 200:
-            result = response.json()
-            list_id = result.get("id")
-            # ✅ Use the correct ClickUp team ID and path format
-            list_url = f"https://app.clickup.com/{TEAM_ID}/v/li/{list_id}"
-            return {
-                "message": "✅ Video contractor list created successfully.",
-                "list_url": list_url
-            }
-        else:
-            return {
-                "message": "❌ Failed to create list.",
-                "status_code": response.status_code,
-                "error": response.text
-            }
+    response = requests.post(url, headers=headers, json=payload)
 
+    if response.status_code == 200:
+        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}
     else:
         return {
-            "message": f"Contractor type '{contractor_type}' is not handled. No action taken."
+            "message": "❌ Failed to create list.",
+            "status_code": response.status_code,
+            "error": response.text
         }
+
+@app.post("/video")
+async def create_video_contractor_list(request: Request):
+    data = await request.json()
+    list_name = data.get("listName")
+
+    if not list_name:
+        return {"error": "Missing 'listName' in request"}
+
+    return create_list_from_template(VIDEO_FOLDER_ID, VIDEO_TEMPLATE_ID, list_name)
+
+
+@app.post("/image")
+async def create_image_contractor_list(request: Request):
+    data = await request.json()
+    list_name = data.get("listName")
+
+    if not list_name:
+        return {"error": "Missing 'listName' in request"}
+
+    return create_list_from_template(IMAGE_FOLDER_ID, IMAGE_TEMPLATE_ID, list_name)