Update app.py

app.py

@@ -1,5 +1,5 @@
 from flask import Flask, render_template, request, redirect, url_for
-from translation_data import translation_dict_A, translation_dict_B, translation_dict_C
+from translation_data import translation_dict_A, translation_dict_B, translation_dict_C, translation_dict_D
 
 app = Flask(__name__)
 
@@ -10,6 +10,8 @@ english_sentences_B = list(translation_dict_B.keys())
 japanese_translations_B = list(translation_dict_B.values())
 english_sentences_C = list(translation_dict_C.keys())
 japanese_translations_C = list(translation_dict_C.values())
+english_sentences_D = list(translation_dict_D.keys())
+japanese_translations_D = list(translation_dict_D.values())
 
 @app.route('/')
 def portal():
@@ -27,10 +29,14 @@ def index():
         english = english_sentences_B[index]
         japanese = japanese_translations_B[index]
         total = len(english_sentences_B)
-    else:
+    elif set_name == 'C':
         english = english_sentences_C[index]
         japanese = japanese_translations_C[index]
         total = len(english_sentences_C)
+    elif set_name == 'D':
+        english = english_sentences_D[index]
+        japanese = japanese_translations_D[index]
+        total = len(english_sentences_D)
     return render_template('index.html', set_name=set_name, index=index, english=english, japanese=japanese, total=total)
 
 @app.route('/next')
@@ -43,6 +49,8 @@ def next_card():
         index = 0
     elif set_name == 'C' and index >= len(english_sentences_C):
         index = 0
+    elif set_name == 'D' and index >= len(english_sentences_D):
+        index = 0
     return redirect(url_for('index', set=set_name, index=index))
 
 @app.route('/prev')
@@ -56,6 +64,8 @@ def prev_card():
             index = len(english_sentences_B) - 1
         elif set_name == 'C':
             index = len(english_sentences_C) - 1
+        elif set_name == 'D':
+            index = len(english_sentences_D) - 1
     return redirect(url_for('index', set=set_name, index=index))
 
 if __name__ == '__main__':