Update main.py

main.py

@@ -1,35 +1,54 @@
 from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
 import requests
+import os
 
 app = FastAPI()
 
-CLIENT_KEY = "7502238777792790545"
-CLIENT_SECRET = "4db49a3c142fd84f41c099e9cdbb0949de00cff0"
-REDIRECT_URI = "https://mr-help-binrushd-tiktok-authenticator.hf.space/auth/callback"
+# 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 auth_callback(request: Request):
+async def tiktok_auth_callback(request: Request):
     code = request.query_params.get("code")
+    state = request.query_params.get("state")
 
-    token_url = "https://open.tiktokapis.com/v2/oauth/token"
+    if not code:
+        return JSONResponse(
+            status_code=400,
+            content={"message": "Missing 'code' parameter from TikTok OAuth."}
+        )
+
+    token_url = "https://business-api.tiktok.com/open_api/v1.3/oauth2/access_token/"
     payload = {
-        "client_key": "7502238777792790545",
-        "client_secret": "YOUR_CLIENT_SECRET",
-        "code": code,
-        "grant_type": "authorization_code",
-        "redirect_uri": "https://mr-help-binrushd-tiktok-authenticator.hf.space/auth/callback"
-    }
-    headers = {
-        "Content-Type": "application/x-www-form-urlencoded"
+        "app_id": CLIENT_KEY,
+        "secret": CLIENT_SECRET,
+        "auth_code": code
     }
 
-    response = requests.post(token_url, data=payload, headers=headers)
+    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"].get("access_token")
+    refresh_token = data["data"].get("refresh_token", None)  # ✅ safe fallback
+    advertiser_ids = data["data"].get("advertiser_ids", [])
+
+    print("✅ Access Token:", access_token)
+    print("🔁 Refresh Token:", refresh_token)
+    print("📢 Advertiser IDs:", advertiser_ids)
 
-    try:
-        return response.json()
-    except Exception:
-        return {
-            "error": "Failed to parse TikTok token response",
-            "status_code": response.status_code,
-            "text": response.text
+    return JSONResponse(
+        content={
+            "message": "Token exchange successful",
+            "access_token": access_token,
+            "refresh_token": refresh_token,
+            "advertiser_ids": advertiser_ids
         }
+    )