Update app.py

app.py

@@ -257,8 +257,8 @@ def translate_audio(audio_file_path, prompt, model):
 
 # subtitles maker
 
+# helper function convert json transcription to srt
 def create_srt_from_text(transcription_text):
-    """Converts Groq text transcription to SRT format (assuming no timestamps)."""
     srt_lines = []
     # Assuming no timestamps, we'll assign a default duration of 1 second to each line
     duration = timedelta(seconds=1)
@@ -267,6 +267,7 @@ def create_srt_from_text(transcription_text):
     for i, text_part in enumerate(text_parts):
         text_part = text_part.strip()
         if text_part: # Only add lines with text
+            # Format timestamps correctly
             start_timestamp = f"{start_time.seconds}:{start_time.microseconds // 1000:03}"
             end_timestamp = f"{(start_time + duration).seconds}:{(start_time + duration).microseconds // 1000:03}"
             srt_lines.append(f"{i+1}\n{start_timestamp} --> {end_timestamp}\n{text_part.strip()}\n\n")
@@ -274,14 +275,14 @@ def create_srt_from_text(transcription_text):
     return "".join(srt_lines)
 
 
+# getting transcription + using helper function + adding subs to video if input is video
 def generate_subtitles(audio_file_path, prompt, language, auto_detect_language, model):
-    """Converts Whisper JSON transcription to SRT format."""
     # Check and process the file first
     processed_path, error_message = check_file(audio_file_path)
 
     # If there's an error during file check
     if error_message:
-        return error_message, None, None, None  # Return None for the download button
+        return error_message, None, None
 
     with open(processed_path, "rb") as file:
         transcription_json = client.audio.transcriptions.create(
@@ -304,7 +305,7 @@ def generate_subtitles(audio_file_path, prompt, language, auto_detect_language,
         temp_srt_path = temp_srt_file.name
         temp_srt_file.write(srt_content)
 
-    # Generate subtitles and add to video if MP4
+    # Generate subtitles and add to video if input is video
     if audio_file_path.lower().endswith((".mp4", ".webm")):
         try:
             # Use ffmpeg to add subtitles to the video
@@ -334,9 +335,9 @@ def generate_subtitles(audio_file_path, prompt, language, auto_detect_language,
                 ],
                 check=True,
             )
-            return None, output_file_path, None, temp_srt_path  # Return the temp_srt_path
+            return None, output_file_path, None
         except subprocess.CalledProcessError as e:
-            return None, None, f"Error during subtitle addition: {e}", temp_srt_path  # Return the temp_srt_path
+            return None, None, gr.Error(f"Error during subtitle addition: {e}")
 
     return temp_srt_path, None, None 
     
@@ -345,7 +346,8 @@ with gr.Blocks() as demo:
     gr.Markdown(
         """
     # Groq API UI
-    Inference by Groq. Hugging Face Space by [Nick088](https://linktr.ee/Nick088)
+    Inference by Groq
+    Hugging Face Space by [Nick088](https://linktr.ee/Nick088)
     """
     )
     with gr.Tabs():