import torch
from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
import jiwer

# try:
#     device = "mps"
#     torch_dtype = torch.float16 if torch.backends.mps.is_available() else torch.float32
# except e as Exception:
#     print(f"Error: {e}")

device = "cuda:0" if torch.cuda.is_available() else "cpu"
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32

model_id = "openai/whisper-large-v3"

model = AutoModelForSpeechSeq2Seq.from_pretrained(
    model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True
)
model.to(device)

processor = AutoProcessor.from_pretrained(model_id)

pipe = pipeline(
    "automatic-speech-recognition",
    model=model,
    tokenizer=processor.tokenizer,
    feature_extractor=processor.feature_extractor,
    chunk_length_s=30,
    batch_size=32,
    torch_dtype=torch_dtype,
    device=device,
)
# dataset = load_dataset("distil-whisper/librispeech_long", "clean", split="validation")

# Define reference transcription for WER calculation
reference_transcription = "قال المصنف رحمه الله تعالى ونفعنا الله بعلومه في الدارين امين فصل والربا في الذهب والفضه والمطعومات الربا شكل من اشكال البيوع فيه تبادل مثل الثمن والمثمن فهنا فيه مال بازاء مال ولذلك اختلط الامر على المشركين في مكه فقالوا انما البيع مثل الربا يعني هذا فيه مبادله وهذا فيه مبادله وفي المبادله زياده بين الثمن و المثمن وهنا زياده بين الثمن والمثمن فما الفرق نحن نبيع ونشتري والفرق هو ان الله سبحانه وتعالى امر بحرمة الربا من اجل الحفاظ على نظام الاقتصاد النقدي من داخله فلو اباح الله الربا لكان هناك زياده في وسيط التبادل دون زيادة الانتاج ويحدث بذلك التضخم من داخل النظام فاذا حدث التضخم من ذات النظام ف الذي يضار بذلك محدود الدخل الذي يضار بذلك الفقير فحرم الله الربا من اجل الا يقع ظلم على الفقير واوجب الزكاه حتى يتعادل دخل الفقير لاداء مقتضيات الحياه فتحريم الربا مرتبط في النظر الالهي مع ايجاب الزكاه حرم الله الربا واوجب حرم واوجب الزكاه من اجل حماية انسانية الفقير محدود الدخل فلا يجد الاسعار فوق الطاقه ويستطيع ان يحصل باستهلاكه لما اخذه من الزكاه ما يريد من مقتضيات الحياه فاذا كان الامر كذلك وهذا نظام رباني الهي كان حرا لا يستطيع احد ان يتحكم فيه لا في اجتماع ولا في سياسه ولا في اقتصاد يكمل المنظومه الصلاه فاذا ذهبها الناس وقفوا متجاورين الغني بجواره الفقير والفقير بجواره المسكين والمسكين بجواره الغني لا فرق بينهما والصف الاول مناخ من سبق سبق الى الصف الاول الغني او سبق الى الصف الثاني او سبق الى الصف الاول الفقير لا يستطيع احد ان يجذبه من اوليته حتى يتاخر بسبب غناه او جاهه او سلطانه لان من جاء فوجد الصلاه فليلحق بها حيث ما انتهى به الصف وكذلك في طلب العلم حيث ما انتهى به المجلس من اين تتاتى هذه النفسيه الابيه من دون كبر انما بعزه ورقي وانسانيه بوجوب الزكاه فالزكاة ليست منه ولا عطيه ولا هي يد عليا و بحرمة الربا حتي يحافظ على النظام الاقتصادي من داخله من التضخم ذهبت الايام وجاءت الايام وتسلط علينا ابالسه الجن والانس واخرجوا الذهب والفضه من التعامل واستبدلوا بها حيله سخيفه وهي حيله الورق النقدي جعلوا له اولا غطاء ثم بعد ذلك تناسوا الغطاء فاصبح ورقا لا قيمه له في انتاجه مع قيمته دينار الذهب 4 جرام وربع لو سيحناه يطلع 4 جرام وربع ذهب لو صكيناه يطلع 4 جرام وربع ذهب لو بعناه يطلع 4 جرام وربع الذهب قيمته فيه الورقه ال 200 جنيه اللي في جيبك متكلفه 16 قرش حتة الورقه دي وشوية الاحبار اللي عليها 16 قرش طب الفرق بين 16 قرش و200 جنيه فين سلطة الدوله سلطة الاصدار وسلطة الدوله دي بتتحسب ازاي في نفسها بها يعني جوه مصر كده يبقى بها يعني ال 200 جنيه 200 جنيه طب وفي خارجها يبقى بقوة انتاجها فالدولار يبقى ب 50 جنيه اللي هو كان ب 39 قرش ونص يعني كان الجنيه ب 5 دولار وبعدين لما انهار بقى 2 ونص في الاول في الاربعينات كان الجنيه المصري ب 5 دولار 5 دولار في 50 جنيه يبقى بكام يبقى ب 250 مره يبقى الجنيه ابو 40 سنة 40 بيساوي 250 جنيه من الجنيهات اللي معانا دي تخيل اضرب دخلك في 250 فلو سالنا الشيخ فلان ولا علان بتاخد كام يقوللك عباره عن 6000 جنيه في الشهر طب 6000 جنيه في الشهر اضربها في 250 يبقى 150 قصادها صفر يبقى مليون ونص ها تخيل نفسك بقى ان انت معاك مليون ونص كل شهر هتعرف تسكن هتعرف تشتري لك عربيه من ام 2 مليون دي اقل حاجه حتى العربيه ام 14 مليون ممكن تجيبها بس بالتقسيط الله هو ايه اللي حصل تضخم والتضخم ده معناته ايه معناته ان هذا الوسيط للتبادل لم يعد سببا لاغناء الفقير الكلام ده كله يا اخي لقينا الشافعي فهمه الكلام اللي احنا قلناه ده والامام الغزالي ركن الشافعيه فاهمينه بيكتبوه عادي كده فيقول رضي الله تعالى عنه في الام وعلة الربا تعبديه فالذهب والفضه جزء عله يعني جزء العله يعني اذا خرج عن الذهبيه والفضيه فلا ربا ويقول ابن الحاجب وهل حرم الربا لجوهرهما او لجوهريتهما الذهب والفضه لجوهرهما عشان هم ذهب وفضه ولا لانهم وسيط للتبادل قال بل لجوهرهما لجوهرهما يعني لانهم ذهب وفضه مش لجوهريتهما يعني كونهم وسيط للتبادل بين الناس لو كان وسيط  التبادل كان الورق المعفن اللي معانا ده يبقى فيه الربا وتبقى البنوك حرام والشركات حرام وكله حرام في حرام لكنه لما كان ورقا بنكن نوت فانه خرج عن نطاق الربا الشرعيه فلما نيجي نقول لاخواننا النابته يا جماعه ده البنك ده ما بيعملش حاجه لا بيسلف ولا بيستلف ولا بيعمل ولا كذا الى اخره ده ورق ده مافيش دهب وفض فما يرضاش يفهم عشان مش فاهم الكلمتين اللي انا قلته لكم دلوقتي واخاف انه بعد ما يسمع الكلام ده برض ما يفهمش خلاص انا لله وانا اليه راجعون اجمع الائمه الاربعه ان عله الربا تعبديه وان الذهب والفضه مقصود لذاتها وانها جوهرهما هو العله ولذلك تجد كان زمان لما اجي اقول لك دي ب 10 دينار الدينار ذهب اربعه جرام وربع قلت لي انا هشتري منك 10 حبات كل واحده ب 10 دينار يبقى 100 دينار يلا نعد قلته تعد ايه قال لي عد الدنانير 1 ا 3 قلت له لا ده ممكن الدينار من دول تكون جبت صنفره وصنفرته ونزلت منه قد ربع جرام ولا حاجه مش هو اربعه وربع تلقى الجرام اللي انت الدينار اللي انت عطيتي ده اربعه بس قال طب وبعدين نعمل ايه في الورطه دي قلتله نوزن نجيب ميزان اربعه وربع في 10 يبقى ب 42 جرام وربع طيب 42 جرام ونص 42 جرام ونص في 10 يبقى ب 425 هات بقى الكيس اللي معاك ووزنه يطلع 4 25 طب لو طلعوا 420 يبقى انت في 5 جرام ناقصه يعني دينار طب ولو طلع 410 يبقى في 15 ناقصين فكانوا بيوزن الذهب والفضه فابو حنيفه يجي يقوللك ايه في كل مكيل وموزون طب البنكنوت اللي في جبنه ده بيتو ما هو ما بيتو زينش فاختلط على حتى كثير من العلماء هذا الفكر ان وجوب الزكاه وحرمه الربا من عند الله وانهما معا يهبون انسانيه الانسان للفقير من غير منه ومن غير تكبر من الغني ولا صاحب السلطان على الفقير الذهب والفضه من الموزون فالشافعي اختلف فقهاء الاسلام على عشره اقوال في علة الربا اختيار الامام الشافعي بعد ابحاثه وادلته ان الربا يجري في امرين في الذهب والفضه من الاثمان وفي المطعوم من الاشياء"

# Transcribe the audio file in chunks
sample = "sidiali_reba_zahab_feda.wav"

# kwargs as listed on the model, can be changed
generate_kwargs = {
    "max_new_tokens": 444,
    "num_beams": 1,
    "language": "arabic",
    "condition_on_prev_tokens": False,
    "compression_ratio_threshold": 1.35,  # zlib compression ratio threshold (in token space)
    "temperature": (0.0, 0.2, 0.4, 0.6, 0.8, 1.0),
    "logprob_threshold": -1.0,
    "no_speech_threshold": 0.6,
    "return_timestamps": True,
}

transcription = pipe(sample, generate_kwargs=generate_kwargs, return_timestamps=True)

if transcription:
    print("Final Transcription:\n", transcription)
    
    # Calculate Word Error Rate (WER)
    wer = jiwer.wer(reference_transcription, transcription)
    print(f"\nWord Error Rate (WER): {wer}")
    
    # Save the transcription and WER to a file
    with open("evaluation_results.txt", "w", encoding="utf-8") as f:
        f.write(f"Transcription:\n{transcription}\n\n")
        f.write(f"Reference:\n{reference_transcription}\n\n")
        f.write(f"Word Error Rate (WER): {wer}\n")
    
    print("\nEvaluation results saved to evaluation_results.txt")
else:
    print("Transcription failed.")