Spaces:
Running
Running
| import gradio as gr | |
| import voice_recognition as vr | |
| # Example lists for source and target languages | |
| source_languages = ["English", "French", "Arabic", "MA-Arabic", "Tagalog", "Cebuano"] | |
| target_languages = ["English", "French", "Arabic", "MA-Arabic", "Tagalog", "Cebuano"] | |
| # Model and task options | |
| models = ["Whisper", "Google"] | |
| tasks = ["Transcript", "Transcript and Translate"] | |
| # Dictionary mapping language labels to ISO codes | |
| language_mapping = { | |
| "English": "en", | |
| "French": "fr", | |
| "Arabic": "ar", | |
| "MA-Arabic": "ar", | |
| "Tagalog": "tl", | |
| "Cebuano": "fil-PH" | |
| } | |
| # Method to get the ISO code from a language label | |
| def get_iso_code(language_label): | |
| return language_mapping.get(language_label, "en") | |
| # Function to handle the input and return the selections | |
| def process_inputs(source_lang, target_lang, model, task, audio_mic, audio_upload): | |
| print("button submit clicked !!") | |
| audio = None | |
| if audio_mic: | |
| audio = audio_mic | |
| elif audio_upload: | |
| audio = audio_upload | |
| print("audio") | |
| print(audio) | |
| print("----") | |
| response = vr.process_audio_recognition(model=model, audio_path=audio, source_lang=get_iso_code(source_lang), | |
| target_lang=get_iso_code(target_lang), translate= task == "Transcript and Translate") | |
| return response | |
| # return { | |
| # "Source Language": source_lang, | |
| # "Target Language": target_lang, | |
| # "Model": model, | |
| # "Task": task, | |
| # "Audio Received": "Yes" if audio else "No" | |
| # } | |
| # Gradio interface setup | |
| with gr.Blocks(title = "Voice Transcription") as demo: | |
| with gr.Row(): | |
| # Dropdown for source and target languages (display name, return ISO code) | |
| source_lang_dropdown = gr.Dropdown(label="Source Language", choices=source_languages, value="English") | |
| target_lang_dropdown = gr.Dropdown(label="Target Language", choices=target_languages, value="French") | |
| with gr.Row(): | |
| # Dropdown for models and tasks | |
| model_dropdown = gr.Dropdown(label="Model", choices=models, value=models[0], info="The model that will be used") | |
| task_dropdown = gr.Dropdown(label="Task", choices=tasks, value=tasks[0], info="Transcript and/or Translate") | |
| # Audio input (mic or upload) | |
| with gr.Tab("Record"): | |
| audio_input_mic = gr.Audio(sources=["microphone"], type="filepath") | |
| with gr.Tab("Upload"): | |
| audio_input_upload = gr.Audio(sources=["upload"], type="filepath") | |
| #audio_input = gr.Audio(sources=["microphone"], type="filepath") | |
| #audio_input2 = gr.Audio(sources=["upload"], type="filepath") | |
| # Button to submit | |
| submit_button = gr.Button("Submit") | |
| # Output label to show the selected inputs | |
| output = gr.Textbox(label="Result") | |
| # Link button click to function | |
| submit_button.click(fn=process_inputs, | |
| inputs=[source_lang_dropdown, target_lang_dropdown, model_dropdown, task_dropdown, audio_input_mic, audio_input_upload], | |
| outputs=output) | |
| if __name__ == '__main__': | |
| # check the model rate | |
| demo.launch(favicon_path = "./favicon.ico") | |