import time
import random
from datetime import datetime
import base64
from enum import Enum
import os
import librosa
import numpy as np
import soundfile as sf

SAMPLING_RATE = int(os.getenv("DEFAULT_SAMPLING_RATE", "48000"))

output_audio_dir = os.getenv("OUTPUT_DIR", f"{os.path.dirname(os.path.realpath(__file__))}/outputs")
os.makedirs(output_audio_dir, exist_ok=True)

 
LANG_TEMPOS = {
    "malayalam": 1.2,
    "telugu": 1.3,
    "hindi": 0.9,
    "punjabi": 0.9
}

SUPPORTED_OUTPUT_LANGS = os.getenv("LANGUAGES", "hindi,bengali,tamil").split(",")

def isOutputLangSupported(lang: str): 
    if(lang not in SUPPORTED_OUTPUT_LANGS):
        raise ValueError(f"Unsupported output language: {lang}. Supported languages are: {SUPPORTED_OUTPUT_LANGS}")
    

def truncate_string(s, max_length=100):
    return s[:max_length] + ('...' if len(s) > max_length else '')


WARMUP_PARAGRAPHS = {
    "hindi": (
        "भारत एक विशाल देश है जहाँ विभिन्न भाषाएं, धर्म और संस्कृतियाँ एक साथ अस्तित्व में हैं। "
        "यहाँ की विविधता में एकता इसकी सबसे बड़ी विशेषता है। हर राज्य की अपनी एक परंपरा होती है और "
        "खानपान भी अलग होता है। भारतीय समाज में परिवार को अत्यंत महत्त्व दिया जाता है और लोग एक-दूसरे "
        "की मदद के लिए हमेशा तैयार रहते हैं। पर्व-त्योहार जैसे दीपावली, होली, ईद और क्रिसमस यहाँ बड़े "
        "हर्षोल्लास के साथ मनाए जाते हैं। यहाँ की कला, संगीत, और साहित्य ने दुनिया भर में अपनी पहचान बनाई है।"
    ),
    "kannada": (
        "ಭಾರತವು ವಿಭಿನ್ನ ಭಾಷೆಗಳು, ಧರ್ಮಗಳು ಮತ್ತು ಸಂಸ್ಕೃತಿಗಳ ನಾಡಾಗಿದೆ. ಇದು ಒಂದೇ ದೇಶದಲ್ಲಿ ಅನೇಕ ಭಿನ್ನತೆಗಳನ್ನು "
        "ಒಳಗೊಂಡಿದೆ. ಪ್ರತಿಯೊಂದು ರಾಜ್ಯಕ್ಕೂ ತನ್ನದೇ ಆದ ಪರಂಪರೆ, ಸಂಸ್ಕೃತಿ ಮತ್ತು ಆಹಾರದ ಶೈಲಿ ಇರುತ್ತದೆ. "
        "ಇಲ್ಲಿ ಕುಟುಂಬ ಮತ್ತು ಸಮುದಾಯಕ್ಕೆ ಹೆಚ್ಚಿನ ಮಹತ್ವ ನೀಡಲಾಗುತ್ತದೆ. ಹಬ್ಬಗಳು ಮತ್ತು ಆಚರಣೆಗಳು ಜನರ ನಡುವಿನ "
        "ಸಾಮರಸ್ಯವನ್ನು ಬಲಪಡಿಸುತ್ತವೆ. ಭಾರತದಲ್ಲಿ ಸಂಗೀತ, ನೃತ್ಯ ಮತ್ತು ಸಾಹಿತ್ಯ ಕ್ಷೇತ್ರಗಳಲ್ಲಿ ಸಮೃದ್ಧ ಪರಂಪರೆ ಇದೆ."
    ),
    "malayalam": (
        "ഇന്ത്യ ഒരു വൈവിധ്യങ്ങളാൽ സമ്പന്നമായ ദേശമാണ്. വിവിധ ഭാഷകളും മതങ്ങളും സംസ്‌കാരങ്ങളുമുള്ള ഈ രാജ്യത്ത് "
        "ഏറ്റവും വലിയ പ്രത്യേകത ഏകതയിലാണ്. ഓരോ സംസ്ഥാനത്തിനും തനതായ ചരിത്രവും സംസ്കാരവുമുണ്ട്. "
        "കുടുംബ ബന്ധങ്ങൾക്കും ഒത്തുചേരലിനും വലിയ വിലയാണ് ഇന്ത്യയിലെ ജനങ്ങൾ നൽകുന്നത്. "
        "ഓണവും വിഷുവും പോലുള്ള ഉത്സവങ്ങൾ ഒരേ സമയം ആത്മീയതയും സന്തോഷവും പങ്കുവെക്കാൻ സഹായിക്കുന്നു. "
        "ഇവിടെ സംഗീതം, നാടകം, ചലച്ചിത്രം തുടങ്ങിയ കലാരൂപങ്ങൾ ലോകപ്രശസ്തമാണ്."
    ),
    "punjabi": (
        "ਭਾਰਤ ਇੱਕ ਐਸਾ ਦੇਸ਼ ਹੈ ਜਿੱਥੇ ਬਹੁਤ ਸਾਰੀਆਂ ਭਾਸ਼ਾਵਾਂ, ਧਰਮਾਂ ਅਤੇ ਰਿਵਾਇਤਾਂ ਦਾ ਮਿਲਾਪ ਹੈ। "
        "ਇਸ ਦੀ ਸਭ ਤੋਂ ਵੱਡੀ ਖਾਸੀਅਤ ਇਹ ਹੈ ਕਿ ਇੱਥੇ ਵੱਖ-ਵੱਖਤਾ ਵਿੱਚ ਵੀ ਏਕਤਾ ਹੈ। ਹਰ ਰਾਜ ਦੀ ਆਪਣੀ ਪਛਾਣ ਹੁੰਦੀ ਹੈ "
        "ਅਤੇ ਇੱਥੇ ਦੇ ਲੋਕ ਆਪਣੇ ਪਰਿਵਾਰ ਅਤੇ ਸੰਸਕਾਰਾਂ ਨਾਲ ਜੁੜੇ ਰਹਿੰਦੇ ਹਨ। ਪੰਜਾਬੀ ਲੋਕਾਂ ਦੀ ਬੋਲੀ, ਭੰਗੜਾ, ਗਿੱਧਾ ਅਤੇ "
        "ਖਾਣ-ਪੀਣ ਦੀ ਰੀਤ ਸਾਰੀ ਦੁਨੀਆ ਵਿੱਚ ਮਸ਼ਹੂਰ ਹੈ। ਤਿਉਹਾਰ, ਖੇਡਾਂ ਅਤੇ ਲੋਕ ਸੱਭਿਆਚਾਰ ਇੱਥੇ ਦੀ ਰੋਹਕਤਾ ਵਧਾਉਂਦੇ ਹਨ।"
    ),
    "tamil": (
        "இந்தியா என்பது பல்வேறு கலாசாரங்களும் மொழிகளும் கொண்ட நாடாகும். ஒவ்வொரு மாநிலத்திற்கும் தனித்துவமான பாரம்பரியம் "
        "மற்றும் சுவையான உணவுகள் உள்ளன. தமிழ்நாடு இவற்றில் முக்கியமான ஒரு பகுதி. தமிழ் மொழியும் கலாசாரமும் உலகெங்கும் "
        "புகழ்பெற்றவை. குடும்பங்கள் ஒன்றிணைந்து வாழும் பாசத்திற்கும் எப்போதும் முக்கியத்துவம் அளிக்கப்படுகிறது. "
        "தீபாவளி, பொங்கல் போன்ற பண்டிகைகள் மகிழ்ச்சியுடனும் ஒருமைப்பாட்டுடனும் கொண்டாடப்படுகின்றன. "
        "இங்கே உள்ள பாரம்பரிய இசை, நடனம், இலக்கியம் ஆகியவை பாராட்டத்தக்கவை."
    ),
    "telugu": (
        "భారతదేశం అనేక భాషలు, సంస్కృతులు మరియు సంప్రదాయాలతో కూడిన దేశం. ప్రతి రాష్ట్రానికీ తనదైన ప్రత్యేకత ఉంది. "
        "తెలుగు రాష్ట్రాలు సంస్కృతిగా గొప్పదిగా పరిగణించబడతాయి. ఇక్కడ కుటుంబం, గౌరవం, సహాయం వంటి విలువలకు ఎక్కువ ప్రాధాన్యం "
        "ఇస్తారు. సంక్రాంతి, ఉగాది వంటి పండుగలు కుటుంబ సభ్యులందరినీ కలిపి ఉంచుతాయి. నాట్యం, సంగీతం మరియు సాహిత్యం వంటి కళలు "
        "తెలుగువారిలో భాగం. భోజన పరంపరలూ అద్భుతమైనవి. వీటన్నింటి సమ్మేళనమే భారతదేశాన్ని ప్రత్యేకంగా నిలబెడుతుంది."
    )
}