File size: 4,281 Bytes
70b382b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import requests
import json # Para trabajar con la respuesta JSON
import uuid
def ejecutar_post(creation_id, jwt_token):
url = 'https://api.vidu.studio/vidu/v1/tasks'
headers = {
'Connection': 'keep-alive',
'sec-ch-ua': '"Chromium";v="128", "Not;A=Brand";v="24", "Google Chrome";v="128"',
'Accept-Language': 'en',
'sec-ch-ua-mobile': '?0',
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
'content-type': 'application/json',
'X-Request-Id': str(uuid.uuid4()),
'sec-ch-ua-platform': '"Windows"',
'Accept': '*/*',
'Origin': 'https://www.vidu.studio',
'Sec-Fetch-Site': 'same-site',
'Sec-Fetch-Mode': 'cors',
'Sec-Fetch-Dest': 'empty',
'Referer': 'https://www.vidu.studio/',
'Cookie': f'JWT={jwt_token}', # Incluir solo el JWT
'Cookie': f'sajssdk_2015_cross_new_user=1; _ga=GA1.1.77204265.1725572952; amp_af43d4=61603120ab0445398cd1cb92fd88aef0...1i727srs7.1i727ujms.6.1.7; JWT={jwt_token}; Shunt=; sensorsdata2015jssdkcross=dfm-enc-%7B%22Va28a6y8_aV%22%3A%22sSEGIEItsRRRsHGE%22%2C%22gae28_aV%22%3A%22EGEySsGigsIHGA-AsSSHnIgEEySVVs-snAAEEHE-EsGnAAA-EGEySsGigsSisH%22%2C%22OemO2%22%3A%7B%22%24ki8r28_8eiggay_2mbeyr_8cOr%22%3A%22%E7%9B%B4%E6%8E%A5%E6%B5%81%E9%87%8F%22%2C%22%24ki8r28_2rieyz_lrcMmeV%22%3A%22%E6%9C%AA%E5%8F%96%E5%88%B0%E5%80%BC_%E7%9B%B4%E6%8E%A5%E6%89%93%E5%BC%80%22%2C%22%24ki8r28_ergreere%22%3A%22%22%7D%2C%22aVr68a8ar2%22%3A%22rc3liZ7ku67OV5kgPsGCiskkDskl3qmawFlJPhNcpZTowqwEpF08wX3AQXKswsPJwZwAW9NcBF3swX0JwFKJBF1cpFPMwX08wFlJPhNcpZTowq7zwqKaBv3liZ7ku67OV5kgu9G6iZHgiZNapa3cQX1Hwh1hpX3IQhycQFlJ36A%3D%22%2C%22za28mec_kmfa6_aV%22%3A%7B%226ior%22%3A%22%24aVr68a8c_kmfa6_aV%22%2C%22Cikbr%22%3A%22sSEGIEItsRRRsHGE%22%7D%7D; _ga_ZJBV7VYP55=GS1.1.1725576214.2.1.1725578996.0.0.0',
'Accept-Encoding': 'gzip, deflate'
}
# Datos del cuerpo de la solicitud POST
data = {
"input": {
"creation_id": f"{creation_id}"
},
"type": "upscale",
"settings": {
"model": "stable",
"duration": 4
}
}
# Realizar la solicitud POST
response = requests.post(url, headers=headers, json=data)
# Convertir la respuesta a JSON
response_json = json.loads(response.text)
# Extraer el 'id' de la respuesta
task_id = response_json.get('id', 'ID no encontrado')
# Mostrar la respuesta
print(f"Status code: {response.status_code}")
#print(f"Response: {response.text}")
return task_id
# Funci贸n para guardar el task_id en el archivo en el formato deseado
def guardar_task_id(task_id):
task_id_formateado = f"task_id: {task_id}" # Formato deseado
file_path = '/tmp/task_id_upscale.txt' # Ruta del archivo actualizada
try:
with open(file_path, 'w') as file:
file.write(task_id_formateado)
print(f"task_id guardado")
except IOError as e:
print(f"Error al escribir el archivo: {e}")
def read_file(file_path):
try:
with open(file_path, "r") as file:
return file.readline().strip()
except FileNotFoundError:
print(f"Archivo no encontrado: {file_path}")
return None
except IOError as e:
print(f"Error al leer el archivo: {e}")
return None
def main():
creation_id_path = "/tmp/creation_id.txt"
jwt_token_path = "/tmp/jwt_token.txt"
# Leer creation_id y jwt_token desde los archivos
creation_id = read_file(creation_id_path)
jwt_token = read_file(jwt_token_path)
if not creation_id or not jwt_token:
print("No se pudo leer el creation_id o jwt_token. Aseg煤rate de que los archivos existan y contengan la informaci贸n correcta.")
return
# Llamar a la funci贸n para realizar el POST y obtener el task_id
task_id = ejecutar_post(creation_id, jwt_token)
if task_id != 'ID no encontrado':
# Guardar el task_id en el archivo de texto
guardar_task_id(task_id)
else:
print("No se pudo obtener el task_id.")
if __name__ == "__main__":
main()
|