Update app.py

app.py

@@ -1,36 +1,21 @@
 import gradio as gr
-import whisper
 import yt_dlp
 import torch
 import numpy as np
-import os
 from faster_whisper import WhisperModel
 
-# Load model Whisper
+# Load model sekali aja
 model = WhisperModel("small", device="cpu", compute_type="float32")
 
+# Fungsi untuk transkripsi dari file
 def transcribe_audio(file):
     segments, _ = model.transcribe(file)
     transcript = "\n".join(segment.text for segment in segments)
-    return f"**Transcription:**\n{transcript}"
-
-
-# Load model Whisper (bisa pilih: tiny, base, small, medium, large)
-model = WhisperModel("small", device="cpu", compute_type="float32")
-
-# Fungsi untuk transkripsi dari file
-def transcribe_audio(file):
-    audio = whisper.load_audio(file)
 
-    # Konversi tensor ke NumPy jika diperlukan
-    if isinstance(audio, torch.Tensor):
-        audio = audio.cpu().numpy().astype(np.float32)
+    # Simpan ke file TXT
+    with open("transcription.txt", "w", encoding="utf-8") as f:
+        f.write(transcript)
 
-    # Transkripsi menggunakan faster-whisper
-    segments, _ = model.transcribe(audio)
-    
-    transcript = "\n".join(segment.text for segment in segments)
-    
     return f"**Transcription:**\n{transcript}"
 
 # Fungsi untuk ambil audio dari YouTube
@@ -57,13 +42,14 @@ with gr.Blocks() as app:
         audio_input = gr.File(label="Upload Audio File")
         file_transcribe_button = gr.Button("Transcribe")
         file_output = gr.Textbox(label="Transcription")
-        
+        download_file = gr.File(label="Download Transcription", value="transcription.txt")
+
     with gr.Tab("YouTube Video"):
         youtube_url = gr.Textbox(label="YouTube URL")
         yt_transcribe_button = gr.Button("Transcribe")
         yt_output = gr.Textbox(label="Transcription")
 
-    file_transcribe_button.click(transcribe_audio, inputs=audio_input, outputs=file_output)
+    file_transcribe_button.click(transcribe_audio, inputs=audio_input, outputs=[file_output, download_file])
     yt_transcribe_button.click(transcribe_youtube, inputs=youtube_url, outputs=yt_output)
 
 # Run Gradio app