import requests
import json
import time
import socket

URL = "https://api.aditu.eus/"
API_ID = "api_id"
API_KEY = "api key"
VIDEO = 'video path'
GROUP_NAME = "Speaker group name"


def transcribe_file_api(api_id, api_key, language, file, speaker_group=None, channel=2):
	print("Transcribing: %s" % file)
	PARAMS = {"api_id":api_id, "api_key": api_key, "language": language, "channel": channel}
	if speaker_group:
		PARAMS["speaker_group"] = speaker_group
	files = {'file': open(file,'rb')}
	r = requests.post(url = URL + "transcribe", data = PARAMS, files=files)
	if r.status_code != 200:
		print(r.content)
		print("There was an error: %d" % r.status_code)
		return None
	return r.json()

def get_file_api(api_id, api_key, task_id):
	PARAMS = {"api_id":api_id, "api_key": api_key, "task_id": task_id}
	r = requests.post(url = URL + "get_file", data = PARAMS)
	if r.status_code != 200:
		print(r.content)
		print("There was an error: %d" % r.status_code)
		return None
	return r.json()

def export_api(api_id, api_key, file_id, export_format, language='none', speaker_info='none'):
	print(f"Exporting {file_id}-{language} to {export_format} with speaker {speaker_info}")
	PARAMS = {"api_id":api_id, "api_key": api_key, "file_id": file_id, "language": language, "format": export_format, "speaker_info": speaker_info}
	r = requests.post(url = URL + "export", data = PARAMS)
	if r.status_code != 200:
		print(r.content)
		print("There was an error: %d" % r.status_code)
		return None
	return r.content

def get_user_time_info_api(api_id, api_key):
	PARAMS = {"api_id":api_id, "api_key": api_key}
	r = requests.post(url = URL + "time", data = PARAMS)
	if r.status_code != 200:
		print(r.content)
		print("There was an error: %d" % r.status_code)
		return None
	return r.json()

def get_user_space_info_api(api_id, api_key):
	PARAMS = {"api_id":api_id, "api_key": api_key}
	r = requests.post(url = URL + "space", data = PARAMS)
	if r.status_code != 200:
		print(r.content)
		print("There was an error: %d" % r.status_code)
		return None
	return r.json()

def get_speaker_group_id_api(api_id, api_key, group_name):
	PARAMS = {"api_id":api_id, "api_key": api_key, "group_name":group_name}
	r = requests.post(url = URL + "get_speaker_group_id", data = PARAMS)
	if r.status_code != 200:
		print(r.content)
		print("There was an error: %d" % r.status_code)
		return None
	return r.json()


def transcribe_file(api_id, api_key, language, file, json, transcription, docx, docx2, subtitles, vtt_subtitles, stl_subtitles,group_name=None, channel=2):
	response = get_speaker_group_id_api(api_id, api_key, group_name)
	if response:
		speaker_group_id = response["speaker_group_id"]
	else:
		speaker_group_id = None

	response = transcribe_file_api(api_id, api_key, language, file, speaker_group_id, channel)
	if not response:
		return False
	print("File succesfully uploaded to server")

	retries = 0
	while True:
		if retries == 5:
			return False
		time.sleep(2)
		result = get_file_api(api_id, api_key, response["task_id"])
		print (result)
		if result:
			if result["finished"] and not result["error"]:
				export_content=export_api(api_id, api_key, result["file_id"], "WORDS", 'none', 'none')
				open(json, 'wb').write(export_content)
				export_content=export_api(api_id, api_key, result["file_id"], "TXT", 'none', 'none')
				open(transcription, 'wb').write(export_content)
				export_content=export_api(api_id, api_key, result["file_id"], "DOCX", 'none', 'names')
				open(docx, 'wb').write(export_content)
				export_content=export_api(api_id, api_key, result["file_id"], "MIN", 'none', 'none')
				open(docx2, 'wb').write(export_content)
				export_content=export_api(api_id, api_key, result["file_id"], "SRT", 'none', 'hyphens')
				open(subtitles, 'wb').write(export_content)
				export_content=export_api(api_id, api_key, result["file_id"], "VTT", 'none', 'colors')
				open(vtt_subtitles, 'wb').write(export_content)
				export_content=export_api(api_id, api_key, result["file_id"], "STL", 'none', 'none')
				open(stl_subtitles, 'wb').write(export_content)
				return True
			elif result["finished"] and result["error"]:
				return False
			else:
				print("[ " + str(result["percentage"]) + "% ] " + result["message"])
		else:
			retries += 1

result = transcribe_file(API_ID, API_KEY, "eu", VIDEO, "proba.json", "proba.txt", "proba.docx", "proba2.docx", "proba.srt", "proba.vtt", "proba.stl", GROUP_NAME)
if result:
	print("File succesfully transcribed")
else:
	print("There was an error")

#print(get_user_time_info_api(API_ID, API_KEY))