Update app.py

app.py

@@ -11,17 +11,21 @@ def load_data(level):
     data = pd.read_csv(csv_file_path)
     return data
 
-def generate_speech(level, x, y, audio_option):
+def generate_speech(level, x, y, audio_option, pos_filter):
     data = load_data(level)
     
     x, y = int(x), int(y)
-    filtered_df = data[(data['SID'] >= x) & (data['SID'] <= y)]
+    if pos_filter != "Any":
+        filtered_df = data[(data['SID'] >= x) & (data['SID'] <= y) & (data['POS'].str.lower() == pos_filter.lower())]
+    else:
+        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":
-            sentence = f"Number {row['SID']}. {row['SID']} is {row['POS']}."
+            # 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']}!"
         
@@ -37,17 +41,19 @@ def generate_speech(level, x, y, audio_option):
     mp3_io.seek(0)
     return mp3_io.getvalue()
 
+
 iface = gr.Interface(
     fn=generate_speech,
     inputs=[
-        gr.Dropdown(label="Select Level (3K: A1, A2, B1, B2, C1; 5K: additional B2 and C1)", choices=['A1', 'A2', 'B1', 'B2', 'C1', '5K']),
-        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 Level", choices=['A1', 'A2', 'B1', 'B2', 'C1', '5K']),
+        gr.Number(label="Range: Start Number (x)"),
+        gr.Number(label="Range: End Number (y)"),
+        gr.Radio(label="Audio Option", choices=["Audio with number", "Audio without number"]),
+        gr.Dropdown(label="Select Part of Speech", choices=["Any", "Noun", "Verb", "Adjective", "Adverb"])  # Corrected line
     ],
     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, and select the audio option. The system will create a single audio file. After submission, you have the option to download the audio file."
+    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)