Update app.py

app.py

@@ -4,26 +4,27 @@ 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"}
+csv_files = {"A1": "OF3KA1.csv", "A2": "OF3KA2.csv", "B1": "OF3KB1.csv", "B2": "OF3KB2.csv", "C1": "OF3KC1.csv", "5K": "OF5K.csv"}
 
-# Function to load the DataFrame based on level selection
 def load_data(level):
-    # Use the csv_files dictionary to get the correct file name for the given level
-    csv_file_path = f"./{csv_files[level]}"  # This correctly uses the dictionary
+    csv_file_path = f"/content/{csv_files[level]}"
     data = pd.read_csv(csv_file_path)
     return data
 
-def generate_speech(level, x, y):
-    data = load_data(level)  # Load data for the selected level
+def generate_speech(level, x, y, audio_option):
+    data = load_data(level)
     
-    # Ensure x and y are integers and within range
     x, y = int(x), int(y)
     filtered_df = data[(data['SID'] >= x) & (data['SID'] <= y)]
 
-    combined_audio = AudioSegment.silent(duration=1000)  # Start with silence for padding
+    combined_audio = AudioSegment.silent(duration=1000)
 
     for _, row in filtered_df.iterrows():
-        sentence = f"Number {row['SID']}. {row['WORD']}! {row['WORD']} is {row['POS']}."
+        if audio_option == "Audio with number":
+            sentence = f"Number {row['SID']}. {row['SID']} is {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)
@@ -36,17 +37,17 @@ def generate_speech(level, x, y):
     mp3_io.seek(0)
     return mp3_io.getvalue()
 
-# Interface with updated level selection including all specified options
 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']),  # Updated choices
+        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.Number(label="End Number (y)"),
+        gr.Radio(label="Audio Option", choices=["Audio with number", "Audio without number"])
     ],
     outputs=gr.Audio(label="Generated Speech"),
     title="Oxford Learner Vocabulary by CEFR levels: Learn with Sound",
-    description="Choose a level and define the starting and ending numbers for that level. The system will create a single audio file formatted as 'Number 1. Word is a noun.' After submission, you have the option to download the audio file. Additionally, you can download the numbered lists for each level from the 'My Apps' section at https://mrkim21.github.io (web address)"
+    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."
 )
 
 iface.launch(share=True, debug=True)