import os
import time

import requests
from pyrogram import Client, enums, filters
from pyrogram.errors import MessageTooLong
from pyrogram.types import Message
from utils.misc import modules_help, prefix

# Replace with your Gladia API key
gladia_key = 'your key '
gladia_url = 'https://api.gladia.io/v2/transcription/'


# Function to make fetch requests to the Gladia API
def make_fetch_request(url, headers, method='GET', data=None):
    if method == 'POST':
        response = requests.post(url, headers=headers, json=data, timeout=10)
    else:
        response = requests.get(url, headers=headers, timeout=10)
    return response.json()


@Client.on_message(filters.command('transcribeyt', prefix) & filters.me)
async def transcribe_audio(_, message: Message):
    """
    Transcribe an audio URL using the Gladia API.

    Usage: .transcribe <audio_url>
    """
    audio_url = message.text.split(maxsplit=1)[1] if len(message.text.split()) > 1 else None

    # Check if a valid URL was provided
    if not audio_url:
        await message.reply('Please provide a valid audio URL.')
        return

    headers = {'x-gladia-key': gladia_key, 'Content-Type': 'application/json'}

    request_data = {'audio_url': audio_url}

    # Send initial request to Gladia API
    status_message = await message.reply('- Sending initial request to Gladia API...')
    initial_response = make_fetch_request(gladia_url, headers, 'POST', request_data)

    # Check if the response contains the result_url
    if 'result_url' not in initial_response:
        await status_message.edit(f'Error in transcription request: {initial_response}')
        return

    result_url = initial_response['result_url']
    await status_message.edit('Initial request sent. Polling for transcription results...')

    # Polling for transcription result
    while True:
        poll_response = make_fetch_request(result_url, headers)

        if poll_response.get('status') == 'done':
            transcription = poll_response.get('result', {}).get('transcription', {}).get('full_transcript')
            if transcription:
                # Format the transcription result with HTML
                result_html = f"""
                <u><b>Transcription Result</b></u>
                <br>
                <pre>{transcription}</pre>
                """
                try:
                    # Attempt to send transcription as a message
                    await message.reply_text(result_html, parse_mode=enums.ParseMode.HTML)
                except MessageTooLong:
                    # Save the large response to a file
                    with open('transcription.txt', 'w') as f:
                        f.write(result_html)

                    # Read general details to include in the caption
                    general_details = "Here's the transcription result."

                    # Send the file with a caption
                    await message.reply_document(
                        'transcription.txt',
                        caption=f'<u><b>General Details</b></u>:\n{general_details}',
                    )

                    # Clean up by removing the file
                    os.remove('transcription.html')
            else:
                await status_message.edit('Transcription completed, but no transcript was found.')
            break
        else:
            await status_message.edit(f'Transcription status: {poll_response.get("status")}')
            time.sleep(30)  # Wait for a few seconds before polling again


modules_help['transcribeyt'] = {'transcribeyt [yt video url]': 'Reply with a YT video link to get transcribed '}