Update app.py

app.py

@@ -15,26 +15,33 @@ def generate_speech(level, x, y, audio_option, pos_filter):
     data = load_data(level)
     
     x, y = int(x), int(y)
-    if pos_filter != "Any":
+    if pos_filter != "ALL":  # If a specific POS is selected, filter by it
         filtered_df = data[(data['SID'] >= x) & (data['SID'] <= y) & (data['POS'].str.lower() == pos_filter.lower())]
-    else:
+    else:  # Include all words if "ALL" is selected
         filtered_df = data[(data['SID'] >= x) & (data['SID'] <= y)]
 
-    combined_audio = AudioSegment.silent(duration=1000)
-
-    for _, row in filtered_df.iterrows():
-        if audio_option == "Audio with number":
-            # Correcting the sentence structure to include both SID and the word's details properly
-            sentence = f"Number {row['SID']}. {row['WORD']}! {row['WORD']} is a {row['POS']}."
-        else:  # "Audio without number"
-            sentence = f"{row['WORD']}!"
-        
+    if len(filtered_df) == 0:  # Check if the filtered DataFrame is empty
+        sentence = f"There is no {pos_filter} in the selected range."
         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")
-        combined_audio += sentence_audio + AudioSegment.silent(duration=1500)
+        combined_audio = sentence_audio
+    else:
+        combined_audio = AudioSegment.silent(duration=1000)
+        for _, row in filtered_df.iterrows():
+            if audio_option == "Audio with number":
+                sentence = f"Number {row['SID']}. {row['WORD']} is a {row['POS']}."
+            else:  # "Audio without number"
+                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")
+            combined_audio += sentence_audio + AudioSegment.silent(duration=1500)
 
     mp3_io = io.BytesIO()
     combined_audio.export(mp3_io, format='mp3')
@@ -46,10 +53,10 @@ iface = gr.Interface(
     fn=generate_speech,
     inputs=[
         gr.Dropdown(label="Select Level", choices=['A1', 'A2', 'B1', 'B2', 'C1', '5K']),
-        gr.Number(label="Range: Start Number (x)"),
-        gr.Number(label="Range: End Number (y)"),
+        gr.Number(label="Start Number (x)"),
+        gr.Number(label="End Number (y)"),
         gr.Radio(label="Audio Option", choices=["Audio with number", "Audio without number"]),
-        gr.Dropdown(label="Select Part of Speech", choices=["ALL", "Noun", "Verb", "Adjective", "Adverb", "Preposition"])  # Corrected line
+        gr.Dropdown(label="Select Part of Speech", choices=["ALL", "Noun", "Verb", "Adjective", "Adverb", "Preposition"])  # Updated POS filter input
     ],
     outputs=gr.Audio(label="Generated Speech"),
     title="Oxford Learner Vocabulary by CEFR levels: Learn with Sound",