ryuzaki-api / template.py
randydev's picture
Update template.py
d10dfb0 verified
raw
history blame
2.27 kB
import requests
import secrets
import random
import re
import string
import json
import urllib
from base64 import b64decode as m
class Blackbox:
def __init__(self) -> None:
"""API Class for various purposes"""
pass
@staticmethod
def chat(args: str) -> dict:
url = m("aHR0cHM6Ly93d3cuYmxhY2tib3guYWkvYXBpL2NoYXQ=").decode("utf-8")
payload = {
"agentMode": {},
"codeModelMode": True,
"id": "XM7KpOE",
"isMicMode": False,
"maxTokens": None,
"messages": [
{"id": "XM7KpOE", "content": urllib.parse.unquote(args), "role": "user"}
],
"previewToken": None,
"trendingAgentMode": {},
"userId": "87cdaa48-cdad-4dda-bef5-6087d6fc72f6",
"userSystemPrompt": None,
}
headers = {
"Content-Type": m("YXBwbGljYXRpb24vanNvbg==").decode("utf-8"),
"Cookie": m("c2Vzc2lvbklkPWY3N2E5MWUxLWNiZTEtNDdkMC1iMTM4LWMyZTIzZWViNWRjZjtpbnRlcmNvbS1pZC1qbG1xeGljYj00Y2YwN2RkOC03NDJlLTRlM2YtODFkZS0zODY2OTgxNmQzMDA7aW50ZXJjb20tZGV2aWNlLWlkLWpsbXF4aWNiPTFlYWZhYWNiLWYxOGQtNDAyYS04MjU1LWI3NjNjZjM5MGRmNjtpbnRlcmNvbS1zZXNzaW9uLWpsbXF4aWNiPQ==").decode("utf-8"),
"Origin": m("aHR0cHM6Ly93d3cuYmxhY2tib3guYWk=").decode("utf-8"),
"User-Agent": m("TW96aWxsYS81LjAgKFdpbmRvd3MgTlQgMTAuMDsgV2luNjQ7IHg2NCkgQXBwbGVXZWJLaXQvNTM3LjM2IChLSFRNTCBsaWtlIEdlY2tvKUNocm9tZS8xMjEuMC4wLjAgU2FmYXJpLzUzNy4zNg==").decode("utf-8"),
}
try:
response = requests.post(url, json=payload, headers=headers)
response.raise_for_status()
clean_text = response.text.replace("$@$v=undefined-rv1$@$", "")
split_text = clean_text.split("\n\n", 2)
if len(split_text) >= 3:
content_after_second_newline = split_text[2]
else:
content_after_second_newline = ""
return {"answer": content_after_second_newline, "success": True}
except requests.exceptions.RequestException as e:
return {"results": str(e), "success": False}
def send_blackbox_chat(query):
response = Blackbox.chat(query)
return response.get("answer")