SNR THEORY

import requests
import re
import base64
import uuid

URL = "https://veoaifree.com/wp-admin/admin-ajax.php"
PAGE_URL = "https://veoaifree.com/veo-video-generator/"

session = requests.Session()


# ---------- GET NONCE ----------
def get_nonce():
    r = session.get(PAGE_URL, timeout=15)

    patterns = [
        r'"nonce"\s*:\s*"([a-zA-Z0-9]+)"',
        r'nonce["\']\s*[:=]\s*["\']([a-zA-Z0-9]+)["\']',
        r'name="nonce"\s*value="([a-zA-Z0-9]+)"'
    ]

    for p in patterns:
        match = re.search(p, r.text)
        if match:
            return match.group(1)

    return None


# ---------- GENERATE IMAGE ----------
def generate_image(prompt, nonce):
    payload = {
        "action": "veo_video_generator",
        "nonce": nonce,
        "promptText": prompt,
        "totalImages": "1",
        "ratio": "IMAGE_ASPECT_RATIO_PORTRAIT",
        "actionType": "whisk_final_image"
    }

    headers = {
        "User-Agent": "Mozilla/5.0",
        "x-requested-with": "XMLHttpRequest",
        "origin": PAGE_URL,
        "referer": PAGE_URL
    }

    r = session.post(URL, data=payload, headers=headers, timeout=30)

    try:
        return r.json()
    except:
        return {}


# ---------- BASE64 EXTRACT ----------
def extract_base64(result):
    data_uri = result.get("data_uri", "")
    if "base64," in data_uri:
        return data_uri.split("base64,")[1]
    return data_uri if data_uri else None


# ---------- UPLOAD ----------
def upload_base64(data):
    file_data = base64.b64decode(data)
    filename = f"{uuid.uuid4().hex}.png"

    files = {
        "file": (filename, file_data, "image/png")
    }

    res = session.post("https://tmpfiles.org/api/v1/upload", files=files)
    js = res.json()

    if js.get("status") == "success":
        url = js["data"]["url"]
        return url.replace("tmpfiles.org/", "tmpfiles.org/dl/")

    return None


# ---------- MAIN ----------
prompt = input("Enter prompt: ")

nonce = get_nonce()

if not nonce:
    print("❌ Nonce not found")
    exit()

result = generate_image(prompt, nonce)

base64_img = extract_base64(result)

if not base64_img:
    print("❌ Image generation failed")
    exit()

link = upload_base64(base64_img)

if link:
    print("🔥 IMAGE URL:")
    print(link)
else:
    print("❌ Upload failed")