debug

Consumer.py

@@ -76,7 +76,7 @@ def callback(ch, method, properties, body):
 
     # simulate processing
     print(f"TYPE {type(recieved_msg)}, CONTENT {recieved_msg}")
-    model_call(recieved_msg)
+    #model_call(recieved_msg)
 
     # (put your logic here)
     print(f"CONSUMER {consumer_id}:::: Processing done")

app.py

@@ -39,7 +39,7 @@ print("Starting API Server...")
 
 class RequestModel(BaseModel):
     prompt: str
-    ht_token : str
+    ht_token_encrypted_dumped : str
     user_email : str
     user_name : str
     memory: list[str]
@@ -49,6 +49,52 @@ class RequestAnswer(BaseModel):
     user_id: str
     msg_id:str
 
+##################################################
+import os
+import hashlib
+from cryptography.hazmat.primitives.ciphers.aead import AESGCM
+
+
+# ===== Key (same idea as Node) =====
+def get_key():
+    raw_key = os.environ["TOKEN_KEY_ENCRYPTION"].encode()
+    return hashlib.sha256(raw_key).digest()  # 32 bytes
+
+
+# ===== Encrypt =====
+def encrypt_token_to_json(token: str) -> dict:
+    key = get_key()
+    aesgcm = AESGCM(key)
+
+    iv = os.urandom(12)  # same as crypto.randomBytes(12)
+
+    encrypted = aesgcm.encrypt(iv, token.encode(), None)
+
+    # في AESGCM في Python: التاج (tag) بيكون في آخر 16 بايت
+    ciphertext = encrypted[:-16]
+    tag = encrypted[-16:]
+
+    return {
+        "iv": iv.hex(),
+        "data": ciphertext.hex(),
+        "tag": tag.hex(),
+    }
+
+
+# ===== Decrypt =====
+def decrypt_token_from_json(enc: dict) -> str:
+    key = get_key()
+    aesgcm = AESGCM(key)
+
+    iv = bytes.fromhex(enc["iv"])
+    ciphertext = bytes.fromhex(enc["data"])
+    tag = bytes.fromhex(enc["tag"])
+
+    encrypted = ciphertext + tag
+
+    decrypted = aesgcm.decrypt(iv, encrypted, None)
+
+    return decrypted.decode()
 
 ##################################################
 # ROUTES

requirements.txt

@@ -12,4 +12,5 @@ fastapi
 uvicorn
 markdown_pdf
 pika
-redis
+redis
+cryptography