Update app.py

app.py

@@ -341,7 +341,19 @@ def generate_subtitles(audio_file_path, prompt, language, auto_detect_language,
 
 
 
-with gr.TabItem("LLMs"):
+
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+    # Groq API UI
+    Inference by Groq
+    Hugging Face Space by [Nick088](https://linktr.ee/Nick088)
+    """
+    )
+    with gr.Tabs():
+        with gr.TabItem("select option here:"):
+            with gr.Tabs():
+                with gr.TabItem("LLMs"):
                     with gr.Column():
                         model = gr.Dropdown(
                             choices=[
@@ -394,3 +406,95 @@ with gr.TabItem("LLMs"):
                             ],
                         )
                         model.change(update_max_tokens, inputs=[model], outputs=max_tokens)
+
+                with gr.TabItem("Speech To Text"):
+                    gr.Markdown("Speech to Text coming soon!")
+                with gr.TabItem("Transcription"):
+                    gr.Markdown("Transcript audio from files to text!")
+                    with gr.Column():
+                        audio_input = gr.File(
+                            type="filepath", label="Upload File containing Audio", file_types=[f".{ext}" for ext in ALLOWED_FILE_EXTENSIONS]
+                        )
+                        model_choice_transcribe = gr.Dropdown(
+                            choices=["whisper-large-v3"],  # Only include 'whisper-large-v3'
+                            value="whisper-large-v3",
+                            label="Model",
+                        )
+                        transcribe_prompt = gr.Textbox(
+                            label="Prompt (Optional)",
+                            info="Specify any context or spelling corrections.",
+                        )
+                        language = gr.Dropdown(
+                             choices=[(lang, code) for lang, code in LANGUAGE_CODES.items()],
+                             value="en",
+                             label="Language",
+                        )
+                        auto_detect_language = gr.Checkbox(label="Auto Detect Language")
+                        transcribe_button = gr.Button("Transcribe")
+                    transcription_output = gr.Textbox(label="Transcription")
+                    transcribe_button.click(
+                        transcribe_audio,
+                        inputs=[audio_input, transcribe_prompt, language, auto_detect_language, model_choice_transcribe],
+                        outputs=transcription_output,
+                    )
+                with gr.TabItem("Translation"):
+                    gr.Markdown("Transcript audio from files and translate them to English text!")
+                    with gr.Column():
+                        audio_input_translate = gr.File(
+                            type="filepath", label="Upload File containing Audio", file_types=[f".{ext}" for ext in ALLOWED_FILE_EXTENSIONS]
+                        )
+                        model_choice_translate = gr.Dropdown(
+                            choices=["whisper-large-v3"],  # Only include 'whisper-large-v3'
+                            value="whisper-large-v3",
+                            label="Model",
+                        )
+                        translate_prompt = gr.Textbox(
+                            label="Prompt (Optional)",
+                            info="Specify any context or spelling corrections.",
+                        )
+                        translate_button = gr.Button("Translate")
+                    translation_output = gr.Textbox(label="Translation")
+                    translate_button.click(
+                        translate_audio,
+                        inputs=[audio_input_translate, translate_prompt, model_choice_translate],
+                        outputs=translation_output,
+                    )
+                with gr.TabItem("Subtitle Maker"):
+                    with gr.Column():
+                        audio_input_subtitles = gr.File(
+                            label="Upload Audio/Video",
+                            file_types=[f".{ext}" for ext in ALLOWED_FILE_EXTENSIONS],
+                        )
+                        model_choice_subtitles = gr.Dropdown(
+                            choices=["whisper-large-v3"],  # Only include 'whisper-large-v3'
+                            value="whisper-large-v3",
+                            label="Model",
+                        )
+                        transcribe_prompt_subtitles = gr.Textbox(
+                            label="Prompt (Optional)",
+                            info="Specify any context or spelling corrections.",
+                        )
+                        language_subtitles = gr.Dropdown(
+                            choices=[(lang, code) for lang, code in LANGUAGE_CODES.items()],
+                            value="en",
+                            label="Language",
+                        )
+                        auto_detect_language_subtitles = gr.Checkbox(
+                            label="Auto Detect Language"
+                        )
+                        transcribe_button_subtitles = gr.Button("Generate Subtitles")
+                        srt_output = gr.File(label="SRT Output File")
+                        video_output = gr.File(label="Output Video with Subtitles")
+                        transcribe_button_subtitles.click(
+                            generate_subtitles,
+                            inputs=[
+                                audio_input_subtitles,
+                                transcribe_prompt_subtitles,
+                                language_subtitles,
+                                auto_detect_language_subtitles,
+                                model_choice_subtitles,
+                            ],
+                            outputs=[srt_output, video_output, gr.Textbox(label="Error")]
+                        )
+                
+demo.launch()