Catch exception related to search functionality

1 files changed, 9 insertions, 1 deletions

diff --git a/src/main/java/musichub/main/Main.java b/src/main/java/musichub/main/Main.java
index cd1fd5f..b0f98f4 100644
--- a/src/main/java/musichub/main/Main.java
+++ b/src/main/java/musichub/main/Main.java
@@ -246,7 +246,15 @@ public class Main {
                     break;
                 case 'r':
                     //search a music
-                    theHub.searchAudioElement();
+                    try {
+                        theHub.searchAudioElement();
+                    }
+                    catch (NoElementFoundException e){
+                        System.err.println(e.getMessage());
+                    }
+                    catch (java.io.FileNotFoundException e){
+                        System.err.println(e.getMessage()+" Please create a file with the extension .wav inside the song folder at root of the app");
+                    }
                     printAvailableCommands();
                 case 'o':
                     //consult the app logs