Update main.py

main.py

@@ -1,8 +1,14 @@
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
+import requests
+import os
 
 app = FastAPI()
 
+# Replace with your real TikTok credentials
+CLIENT_KEY = os.getenv("TIKTOK_CLIENT_KEY", "YOUR_APP_ID")
+CLIENT_SECRET = os.getenv("TIKTOK_CLIENT_SECRET", "YOUR_APP_SECRET")
+
 @app.get("/auth/callback")
 async def tiktok_auth_callback(request: Request):
     code = request.query_params.get("code")
@@ -14,13 +20,35 @@ async def tiktok_auth_callback(request: Request):
             content={"message": "Missing 'code' parameter from TikTok OAuth."}
         )
 
-    print("✅ Received TikTok OAuth code:", code)
-    print("📦 State parameter (if any):", state)
+    # Exchange code for tokens
+    token_url = "https://business-api.tiktok.com/open_api/v1.3/oauth2/access_token/"
+    payload = {
+        "app_id": CLIENT_KEY,
+        "secret": CLIENT_SECRET,
+        "auth_code": code
+    }
+
+    response = requests.post(token_url, json=payload)
+    data = response.json()
+
+    if data.get("code") != 0:
+        return JSONResponse(
+            status_code=400,
+            content={"message": "Token exchange failed", "details": data}
+        )
+
+    access_token = data["data"]["access_token"]
+    refresh_token = data["data"]["refresh_token"]
+    advertiser_ids = data["data"]["advertiser_ids"]
+
+    print("✅ Access Token:", access_token)
+    print("🔁 Refresh Token:", refresh_token)
+    print("📢 Advertiser IDs:", advertiser_ids)
 
     return JSONResponse(
         content={
-            "message": "OAuth code received successfully.",
-            "code_received": code,
-            "note": "Token exchange will be implemented after app approval."
+            "message": "Token exchange successful",
+            "access_token": access_token,
+            "advertiser_ids": advertiser_ids
         }
-    )
+    )