Update main.py

main.py

@@ -78,15 +78,14 @@ async def create_user(user: User):
 async def get_followers(channel_owner: str, limit: typing.Annotated[int, fastapi.Path(title = "The number of followers to retrieve")]):
     return {"channel_owner": channel_owner, "limit": limit}
 
-encryption_key = os.environ.get("ENCRYPTION_KEY")
-encryption_algorithm = os.environ.get("ENCRYPTION_ALGORITHM")
-if encryption_key is None:
-    raise ValueError("No ENCRYPTION_KEY set for the application")
-if encryption_algorithm is None:
-    raise ValueError("No ENCRYPTION_ALGORITHM set for the application")
-
-
 def encode_data(user: User):
+    encryption_key = os.environ.get("ENCRYPTION_KEY")
+    encryption_algorithm = os.environ.get("ENCRYPTION_ALGORITHM")
+    if encryption_key is None:
+        raise ValueError("No ENCRYPTION_KEY set for the application")
+    if encryption_algorithm is None:
+        raise ValueError("No ENCRYPTION_ALGORITHM set for the application")
+        
     print(f"{encryption_key} and {encryption_algorithm}")
     print(jwt.get_algorithm_by_name(encryption_algorithm))
     payload = {
@@ -99,6 +98,13 @@ def encode_data(user: User):
     return payload
 
 def decode_data(token: str):
+    encryption_key = os.environ.get("ENCRYPTION_KEY")
+    encryption_algorithm = os.environ.get("ENCRYPTION_ALGORITHM")
+    if encryption_key is None:
+        raise ValueError("No ENCRYPTION_KEY set for the application")
+    if encryption_algorithm is None:
+        raise ValueError("No ENCRYPTION_ALGORITHM set for the application")
+        
     try:
         payload = jwt.decode(token, encryption_key, algorithms = [encryption_algorithm])
         print(payload)