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, translation_dict_D
+from translation_data import translation_dict_A, translation_dict_B, translation_dict_C, translation_dict_D, translation_dict_F, translation_dict_G
 
 app = Flask(__name__)
 
@@ -12,6 +12,10 @@ 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())
+english_sentences_F = list(translation_dict_F.keys())
+japanese_translations_F = list(translation_dict_F.values())
+english_sentences_G = list(translation_dict_G.keys())
+japanese_translations_G = list(translation_dict_G.values())
 
 @app.route('/')
 def portal():
@@ -37,6 +41,14 @@ def index():
         english = english_sentences_D[index]
         japanese = japanese_translations_D[index]
         total = len(english_sentences_D)
+    elif set_name == 'F':
+        english = english_sentences_F[index]
+        japanese = japanese_translations_F[index]
+        total = len(english_sentences_F)
+    elif set_name == 'G':
+        english = english_sentences_G[index]
+        japanese = japanese_translations_G[index]
+        total = len(english_sentences_G)
     return render_template('index.html', set_name=set_name, index=index, english=english, japanese=japanese, total=total)
 
 @app.route('/next')
@@ -51,6 +63,10 @@ def next_card():
         index = 0
     elif set_name == 'D' and index >= len(english_sentences_D):
         index = 0
+    elif set_name == 'F' and index >= len(english_sentences_F):
+        index = 0
+    elif set_name == 'G' and index >= len(english_sentences_G):
+        index = 0
     return redirect(url_for('index', set=set_name, index=index))
 
 @app.route('/prev')
@@ -66,6 +82,10 @@ def prev_card():
             index = len(english_sentences_C) - 1
         elif set_name == 'D':
             index = len(english_sentences_D) - 1
+        elif set_name == 'F':
+            index = len(english_sentences_F) - 1
+        elif set_name == 'G':
+            index = len(english_sentences_G) - 1
     return redirect(url_for('index', set=set_name, index=index))
 
 if __name__ == '__main__':