Update app.py

app.py

@@ -1,13 +1,13 @@
-import whisper  # Ensure 'openai-whisper' is installed
+import whisper
 import gradio as gr
 import subprocess
+from autocorrect import Speller
 
 # Load the Whisper model
-try:
-    model = whisper.load_model("large")  # Official Whisper model
-except Exception as e:
-    print(f"Error loading Whisper model: {e}")
-    raise e
+model = whisper.load_model("large")
+
+# Initialize autocorrect for Urdu
+spell = Speller(lang='ur')  # Set the language for Urdu
 
 def transcribe_video(video_path):
     try:
@@ -20,7 +20,12 @@ def transcribe_video(video_path):
 
         # Transcribe the audio in Urdu
         result = model.transcribe(audio_path, task="transcribe", language="ur")
-        return result["text"]
+        transcribed_text = result["text"]
+
+        # Correct the transcribed text using autocorrect
+        corrected_text = spell(transcribed_text)
+
+        return corrected_text
 
     except FileNotFoundError:
         return "Error: ffmpeg is not installed or not found in the environment."
@@ -31,9 +36,9 @@ def transcribe_video(video_path):
 interface = gr.Interface(
     fn=transcribe_video,
     inputs=gr.Video(label="Upload your Urdu-speaking video"),
-    outputs=gr.Textbox(label="Transcribed Text"),
-    title="Urdu Video Transcription App",
-    description="Upload a video file in Urdu, and this app will transcribe the speech into text using Whisper.",
+    outputs=gr.Textbox(label="Corrected Transcribed Text"),
+    title="Urdu Video Transcription with Correction",
+    description="Upload a video file in Urdu, and this app will transcribe the speech and correct the text using Whisper and autocorrect.",
 )
 
 # Launch the app