chore: Refactor TikTok login and callback routes in FastAPI

app.py

@@ -593,34 +593,50 @@ async def tiktok_login():
 
 @app.get("/tiktok_callback")
 async def tiktok_callback(request: Request):
-    code = request.query_params.get('code')
-    state = request.query_params.get('state')
-    csrf_state = request.cookies.get('csrf_state')
-
-    if state != csrf_state:
-        raise HTTPException(status_code=400, detail="Invalid state parameter")
-
-    # Exchange code for access token
-    token_response = requests.post(
-        'https://www.tiktok.com/v2/auth/token/',
-        data={
-            'client_key': CLIENT_KEY,
-            'client_secret': CLIENT_SECRET,
-            'code': code,
-            'grant_type': 'authorization_code',
-            'redirect_uri': REDIRECT_URI
-        }
-    ).json()
+    try:
+        code = request.query_params.get('code')
+        state = request.query_params.get('state')
+        csrf_state = request.cookies.get('csrf_state')
+
+        # Debugging information
+        print(f"Received code: {code}")
+        print(f"Received state: {state}")
+        print(f"CSRF state from cookie: {csrf_state}")
+
+        if state != csrf_state:
+            raise HTTPException(status_code=400, detail="Invalid state parameter")
+
+        # Exchange code for access token
+        token_response = requests.post(
+            'https://www.tiktok.com/v2/auth/token/',
+            data={
+                'client_key': CLIENT_KEY,
+                'client_secret': CLIENT_SECRET,
+                'code': code,
+                'grant_type': 'authorization_code',
+                'redirect_uri': REDIRECT_URI
+            }
+        ).json()
 
-    if "error" in token_response:
-        raise HTTPException(status_code=400, detail=token_response["error_description"])
+        # Log the token response for debugging
+        print(f"Token response: {token_response}")
 
-    access_token = token_response.get('data').get('access_token')
-    open_id = token_response.get('data').get('open_id')
+        if "error" in token_response:
+            raise HTTPException(status_code=400, detail=token_response["error_description"])
 
-    # Save access_token and open_id for future use (Consider storing securely in your database)
+        access_token = token_response.get('data').get('access_token')
+        open_id = token_response.get('data').get('open_id')
 
-    return {"message": "Authorization successful", "access_token": access_token, "open_id": open_id}
+        # Log access token and open_id
+        print(f"Access token: {access_token}")
+        print(f"Open ID: {open_id}")
+
+        return {"message": "Authorization successful", "access_token": access_token, "open_id": open_id}
+    
+    except Exception as e:
+        # Log the exception
+        print(f"Exception occurred: {str(e)}")
+        raise HTTPException(status_code=500, detail="Internal Server Error")
 
 @app.get("/generate_video")
 def generate_video():