File size: 4,353 Bytes
f9a6181 097a36f f9a6181 8f85a53 f9a6181 3d3b58a 097a36f f9a6181 3d3b58a ffb422b 3d3b58a f9a6181 3d3b58a f9a6181 5b088d2 3d3b58a 5b088d2 3d3b58a f9a6181 ffb422b 3d3b58a f9a6181 3d3b58a ffb422b 3d3b58a f9a6181 ffb422b f9a6181 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import gradio as gr
from gtts import gTTS
from pydub import AudioSegment
import pandas as pd
import io
csv_files = {"A1": "OF3KA1.csv", "A2": "OF3KA2.csv", "B1": "OF3KB1.csv", "B2": "OF3KB2.csv", "C1": "OF3KC1.csv", "5K": "OF5K.csv"}
def load_data(level):
csv_file_path = f"./{csv_files[level]}"
data = pd.read_csv(csv_file_path)
return data
def generate_speech(t1, level, t2, x, y, t3, audio_option, t4, all_pos, noun, verb, adjective, adverb, preposition):
data = load_data(level)
x, y = int(x), int(y)
if all_pos: # Include all words if "ALL" is selected
filtered_df = data[(data['SID'] >= x) & (data['SID'] <= y)]
else:
# Initialize an empty DataFrame to accumulate results
filtered_df = pd.DataFrame()
# Check each selected POS and append matches to the filtered_df
if noun:
filtered_df = filtered_df.append(data[(data['SID'] >= x) & (data['SID'] <= y) & (data['POS'].str.lower() == 'noun')])
if verb:
filtered_df = filtered_df.append(data[(data['SID'] >= x) & (data['SID'] <= y) & (data['POS'].str.lower() == 'verb')])
if adjective:
filtered_df = filtered_df.append(data[(data['SID'] >= x) & (data['SID'] <= y) & (data['POS'].str.lower() == 'adjective')])
if adverb:
filtered_df = filtered_df.append(data[(data['SID'] >= x) & (data['SID'] <= y) & (data['POS'].str.lower() == 'adverb')])
if preposition:
filtered_df = filtered_df.append(data[(data['SID'] >= x) & (data['SID'] <= y) & (data['POS'].str.lower() == 'preposition')])
# Ensure the filtered_df is unique in case of overlapping conditions
filtered_df = filtered_df.drop_duplicates()
# Proceed with generating the speech
combined_audio = AudioSegment.silent(duration=1000) # Initial silence
if filtered_df.empty:
sentence = "No matching words found."
tts = gTTS(text=sentence, lang='en')
mp3_fp = io.BytesIO()
tts.write_to_fp(mp3_fp)
mp3_fp.seek(0)
else:
for _, row in filtered_df.iterrows():
# Adjust the sentence based on the audio_option
if audio_option == "Audio without number":
sentence = f"{row['WORD']} is {row['POS']}"
elif audio_option == "Audio with number":
sentence = f"{row['SID']}. {row['WORD']}. {row['WORD']} is {row['POS']}"
else: # "Word only"
sentence = f"{row['WORD']}"
tts = gTTS(text=sentence, lang='en')
mp3_fp = io.BytesIO()
tts.write_to_fp(mp3_fp)
mp3_fp.seek(0)
sentence_audio = AudioSegment.from_file(mp3_fp, format="mp3")
# Add a pause after each word
combined_audio += sentence_audio + AudioSegment.silent(duration=1000)
mp3_io = io.BytesIO()
combined_audio.export(mp3_io, format='mp3')
mp3_io.seek(0)
return mp3_io.getvalue()
# Adjust inputs for POS with checkboxes
iface = gr.Interface(
fn=generate_speech,
inputs=[
gr.Markdown("#### [1] Select Level"), # Adding a label for POS options
gr.Dropdown(label="Select Level", choices=['A1', 'A2', 'B1', 'B2', 'C1', '5K']),
gr.Markdown("#### [2] Select Range"),
gr.Number(label= "Start Number (x)"),
gr.Number(label = "End Number (y)"),
gr.Markdown("#### [3] Audio options (text)"),
gr.Radio(label="Audio Option", choices=["Audio with number", "Audio without number", "Word only"]),
gr.Markdown("#### Select Part of Speech"), # Adding a label for POS options
gr.Checkbox(label="Any", value=False),
gr.Checkbox(label="Noun", value=False),
gr.Checkbox(label="Verb", value=False),
gr.Checkbox(label="Adjective", value=False),
gr.Checkbox(label="Adverb", value=False),
gr.Checkbox(label="Preposition", value=False)
],
outputs=gr.Audio(label="Generated Speech"),
title="Oxford Learner Vocabulary by CEFR levels: Learn with Sound",
description="Choose a level, define the starting and ending numbers, select the audio option, and filter by Part of Speech if desired. The system will create a single audio file. After submission, you have the option to download the audio file."
)
iface.launch(share=True, debug=True)
|