•UzMaxBoy 🐲

from Crypto.Cipher import AES
from Crypto.Protocol.KDF import PBKDF2
from Crypto.Hash import SHA256
import requests, os, base64, json, time

class Vheer:
    def __init__(self):
        self.s = requests.Session()
        self.s.headers.update({'User-Agent': 'Mozilla/5.0'})
        self.key = PBKDF2(b"vH33r_2025_AES_GCM_S3cur3_K3y_9X7mP4qR8nT2wE5yU1oI6aS3dF7gH0jK9lZ", b"vheer-salt-2024", dkLen=32, count=10000, hmac_hash_module=SHA256)
    
    def enc(self, d):
        n = os.urandom(12)
        c = AES.new(self.key, AES.MODE_GCM, nonce=n)
        ct, t = c.encrypt_and_digest(json.dumps(d, separators=(',', ':')).encode())
        return base64.b64encode(n + ct + t).decode()
    
    def dec(self, b):
        x = base64.b64decode(b)
        return json.loads(AES.new(self.key, AES.MODE_GCM, nonce=x[:12]).decrypt_and_verify(x[12:-16], x[-16:]).decode())
    
    def upload(self, p):
        f = os.path.basename(p)
        e = self.enc({"taskType":5, "type":5, "filename":f})
        b = '----WebKitFormBoundary' + os.urandom(16).hex()
        m = {'jpg':'image/jpeg','jpeg':'image/jpeg','png':'image/png','webp':'image/webp'}.get(f.split('.')[-1].lower(), 'image/jpeg')
        with open(p,'rb') as x: d = x.read()
        body = (f'--{b}\r\nContent-Disposition: form-data; name="file"; filename="{f}"\r\nContent-Type: {m}\r\n\r\n').encode() + d + (f'\r\n--{b}\r\nContent-Disposition: form-data; name="params"\r\n\r\n{e}\r\n--{b}--\r\n').encode()
        r = self.s.post("https://vheer.com/app/api/vheer/upload", data=body, headers={'content-type':f'multipart/form-data; boundary={b}'})
        return self.dec(r.json()['data_enc']) if r.json().get('data_enc') else None
    
    def status(self, c):
        r = self.s.post("https://vheer.com/app/api/vheer/status", json={"params":self.enc({"taskType":5,"type":5,"code":c})})
        return self.dec(r.json()['data_enc']) if r.json().get('data_enc') else None
    
    def process(self, p, wait=120):
        print(json.dumps({"s":"start"}))
        if not os.path.exists(p): return print(json.dumps({"s":"error","e":"no file"}))
        self.s.get("https://vheer.com/app/image-to-image")
        self.s.get("https://vheer.com/app/api/auth/csrf")
        u = self.upload(p)
        if not u: return print(json.dumps({"s":"error","e":"upload failed"}))
        c = u.get('code') or u.get('imageId')
        if not c: return print(json.dumps({"s":"error","e":"no code"}))
        print(json.dumps({"s":"uploaded","c":c}))
        t = time.time()
        while time.time() - t < wait:
            r = self.status(c)
            if r:
                print(json.dumps({"s":r.get('status'),"c":c,"d":r}))
                if r.get('status') == 'success': return r
                if r.get('status') == 'failed': return print(json.dumps({"s":"failed"}))
            time.sleep(5)
        return print(json.dumps({"s":"timeout"}))

if __name__ == "__main__":
    r = Vheer().process("1.jpg")
    if r and r.get('downloadUrls'):
        url = r['downloadUrls'][0]
        resp = requests.get(url)
        if resp.status_code == 200:
            os.makedirs("out", exist_ok=True)
            with open(f"out/result.mp4", 'wb') as f: f.write(resp.content)
            print(json.dumps({"s":"done","f":"out/result.mp4"}))