Update app.py

app.py

@@ -69,35 +69,17 @@ def verify_wots(signature, message, public_key):
 @app.post("/relay-txn")
 async def relay_txn(payload: dict):
     # 1. Verification Math
-    proof = base58.b58decode(payload['winternitz_hash'])
-    anchor = base58.b58decode(payload['anchor']) # Stored in your DB
-    lap = payload['lap']
-
-    # Hash the proof 'lap' times to see if it hits the anchor
-    check = proof
-    for _ in range(lap):
-        check = hashlib.sha256(check).digest()
-
-    if check != anchor:
-        raise HTTPException(status_code=403, detail="Quantum Auth Failed")
+    message = base58.b58decode(payload["message"])
+    signature = payload["signature"]        # list[str]
+    public_key = payload["anchorB58"]      # list[str]
 
-    # 2. Transaction Execution
-    # Deserialize the transaction sent from the UI
-    raw_tx = base58.b58decode(payload['txn'])
-    txn = Transaction.from_bytes(raw_tx)
-    txn.partial_sign([SERVER_KEY])
+    is_valid = verify_wots(signature, message, public_key)
 
-    partial_signed_bytes = bytes(txn)
-    encoded_txn = base58.b58encode(partial_signed_bytes).decode("utf-8")
+    if not is_valid:
+        return {"success": False, "error": "Invalid WOTS signature"}
 
+    return {"success": True}
     
-    return {
-        "success": True, 
-        "encoded_txn": encoded_txn,
-        "status": "Finalized on Solana"
-    }
-
-