Update app.py

app.py

@@ -22,46 +22,24 @@ emotion_dict = {
     'neu': 'Neutral'
 }
 
-def predict_emotion(audio):
-    out_prob, score, index, text_lab = learner.classify_file(audio.name)
-    return emotion_dict[text_lab[0]]
 
-def predict_emotion_from_file(file_path):
-    audio = gr.Audio(file_path=file_path, source="upload", type="filepath")
-    return predict_emotion(audio)
 
-# Define the Gradio interface with the first tab for uploading an audio file
-inputs = gr.inputs.Audio(label="Input Audio", type="file")
+# Assuming emotion_dict and learner are defined elsewhere in your code
+# and learner.classify_file is a method that classifies the audio file
+
+def predict_emotion(audio, rec_file):
+    rec_path = os.path.join("rec", rec_file.name)
+    # Assuming you want to use the audio file from the 'rec' directory for prediction
+    out_prob, score, index, text_lab = learner.classify_file(rec_path)
+    return emotion_dict[text_lab[0]]
+
+# Loading gradio interface
+inputs = [
+    gr.inputs.Audio(label="Input Audio", type="file"),
+    gr.inputs.File(label="Choose file from rec directory", type="file", default="rec/")
+]
 outputs = "text"
 title = "ML Speech Emotion Detection"
 description = "Speechbrain powered wav2vec 2.0 pretrained model on IEMOCAP dataset using Gradio."
 
-# Define the second tab for selecting an audio file from the dropdown
-rec_folder = "rec"
-audio_files = [f for f in os.listdir(rec_folder) if f.endswith(('.wav', '.mp3'))]
-audio_files = [os.path.join(rec_folder, f) for f in audio_files]
-file_dropdown = gr.inputs.Dropdown(label="Select Preloaded Audio File", choices=audio_files)
-
-# Create the Gradio interface with both tabs
-iface = gr.Interface(
-    fn=predict_emotion,
-    inputs=inputs,
-    outputs=outputs,
-    title=title,
-    description=description,
-    tab_name="Upload Audio"
-)
-
-# Add the second tab to the interface
-iface.add_tab(
-    "Select Preloaded Audio",
-    gr.Interface(
-        fn=predict_emotion_from_file,
-        inputs=file_dropdown,
-        outputs=outputs,
-        tab_name="Select Preloaded Audio"
-    )
-)
-
-# Launch the Gradio app
-iface.launch()
+gr.Interface(fn=predict_emotion, inputs=inputs, outputs=outputs, title=title, description=description).launch()