Add star wars 1 and exception

program.py

@@ -1,26 +1,30 @@
 #!/usr/bin/python
 
 import sys
+import matplotlib.pyplot as plt
 
 
-def sort(dictionnary):
+def sort(dic):
     new_dic = {}
-    while len(dictionnary) > 0:
+    while len(dic) > 0:
         tmp_val = 0
         tmp_key = ""
-        for key, val in dictionnary.items():
+        for key, val in dic.items():
             if val > tmp_val:
                 tmp_key = key
                 tmp_val = val
         new_dic[tmp_key] = tmp_val
-        dictionnary.__delitem__(tmp_key)
+        dic.__delitem__(tmp_key)
     return new_dic
 
 
 def get_words(script):
     words = []
+
+    try:
         file = open(script, "r")
         file_content = file.read()
+
         file_content_split = file_content.split(" ")
         for word_split_1 in file_content_split:
             if word_split_1 != "":
@@ -30,9 +34,21 @@ def get_words(script):
                         words.append(word)
 
         file.close()
+
+    except FileNotFoundError:
+        sys.stderr.write("Error: " + script + " does not exist!")
+        sys.exit(1)
+
     return words
 
 
+def get_movie_name(file_path):
+    files_name = file_path.split("/")
+    file_name = files_name.pop()
+    movie_name = (file_name.split('.')[0]).replace('_', ' ')
+    return movie_name
+
+
 def get_frequency(words):
     freq = {}
     for word in words:
@@ -43,10 +59,19 @@ def get_frequency(words):
     return sort(freq)
 
 
+def display_fig(words, movie_name):
+    plt.title("Word frequencies in " + movie_name)
+    plt.ylabel("Number of Occurrences")
+    plt.xlabel("Words")
+    plt.show()
+
+
 def main(script):
+    print("File loading and sorting ...")
     words = get_words(script)
     sorted_words = get_frequency(words)
-    print(sorted_words)
+    print("Display result")
+    display_fig(sorted_words, get_movie_name(script))
 
 
 if __name__ == "__main__":