Voice Mode

"""Format converters for transcription output."""

import csv
import io
from typing import Dict, Any, List

from .types import TranscriptionResult, OutputFormat


def format_timestamp_srt(seconds: float) -> str:
    """Format timestamp for SRT (HH:MM:SS,mmm)"""
    hours = int(seconds // 3600)
    minutes = int((seconds % 3600) // 60)
    secs = seconds % 60
    return f"{hours:02d}:{minutes:02d}:{secs:06.3f}".replace(".", ",")


def format_timestamp_vtt(seconds: float) -> str:
    """Format timestamp for WebVTT (HH:MM:SS.mmm)"""
    hours = int(seconds // 3600)
    minutes = int((seconds % 3600) // 60)
    secs = seconds % 60
    return f"{hours:02d}:{minutes:02d}:{secs:06.3f}"


def convert_to_srt(transcription: Dict[str, Any]) -> str:
    """
    Convert transcription to SRT subtitle format.
    """
    srt_lines = []
    
    for i, segment in enumerate(transcription.get("segments", []), 1):
        start = format_timestamp_srt(segment.get("start", 0))
        end = format_timestamp_srt(segment.get("end", 0))
        text = segment.get("text", "").strip()
        
        # Add speaker if available
        if "speaker" in segment:
            text = f"[{segment['speaker']}] {text}"
        
        srt_lines.append(str(i))
        srt_lines.append(f"{start} --> {end}")
        srt_lines.append(text)
        srt_lines.append("")
    
    return "\n".join(srt_lines)


def convert_to_vtt(transcription: Dict[str, Any]) -> str:
    """
    Convert transcription to WebVTT format.
    """
    vtt_lines = ["WEBVTT", ""]
    
    for segment in transcription.get("segments", []):
        start = format_timestamp_vtt(segment.get("start", 0))
        end = format_timestamp_vtt(segment.get("end", 0))
        text = segment.get("text", "").strip()
        
        # Add speaker if available
        if "speaker" in segment:
            text = f"<v {segment['speaker']}>{text}"
        
        vtt_lines.append(f"{start} --> {end}")
        vtt_lines.append(text)
        vtt_lines.append("")
    
    return "\n".join(vtt_lines)


def convert_to_csv(transcription: Dict[str, Any]) -> str:
    """
    Convert transcription to CSV format with word-level data.
    """
    output = io.StringIO()
    
    # Determine columns based on available data
    has_words = "words" in transcription and transcription["words"]
    has_speakers = any("speaker" in w for w in transcription.get("words", []))
    has_probability = any("probability" in w for w in transcription.get("words", []))
    
    # Write header
    if has_words:
        headers = ["word", "start", "end"]
        if has_speakers:
            headers.append("speaker")
        if has_probability:
            headers.append("probability")
    else:
        headers = ["text", "start", "end"]
        if has_speakers:
            headers.append("speaker")
    
    writer = csv.DictWriter(output, fieldnames=headers)
    writer.writeheader()
    
    # Write data
    if has_words:
        for word in transcription.get("words", []):
            row = {
                "word": word.get("word", ""),
                "start": word.get("start", 0),
                "end": word.get("end", 0)
            }
            if has_speakers:
                row["speaker"] = word.get("speaker", "")
            if has_probability:
                row["probability"] = word.get("probability", "")
            writer.writerow(row)
    else:
        for segment in transcription.get("segments", []):
            row = {
                "text": segment.get("text", "").strip(),
                "start": segment.get("start", 0),
                "end": segment.get("end", 0)
            }
            if has_speakers:
                row["speaker"] = segment.get("speaker", "")
            writer.writerow(row)
    
    return output.getvalue()


def convert_to_format(transcription: TranscriptionResult, format: OutputFormat) -> str:
    """
    Convert transcription to specified format.
    
    Args:
        transcription: The transcription result
        format: Target output format
        
    Returns:
        Formatted string representation
    """
    if format == OutputFormat.SRT:
        return convert_to_srt(transcription)
    elif format == OutputFormat.VTT:
        return convert_to_vtt(transcription)
    elif format == OutputFormat.CSV:
        return convert_to_csv(transcription)
    else:
        # Default to JSON (handled elsewhere)
        import json
        return json.dumps(transcription, indent=2)