Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import json
 
 # JSON с тестами, вопросами, правильными ответами и текстом при неправильном ответе
 json_data = """
@@ -22,23 +21,21 @@ json_data = """
 """
 
 # Загрузка данных из JSON
-data = json.loads(json_data)
+data = gr.Interface.load(json_data)
 
 # functions
-def select_test(test_name):
-    return test_name
+def show_test(test_name):
+    test_data = data['tests'][test_name]
+    return test_data, test_name
 
-def check(answers, correct_answers, incorrect_texts):
+def check(answers, test_name):
+    test_data = data['tests'][test_name]
     results = []
-    for i, (answer, correct_answer, incorrect_text) in enumerate(zip(answers, correct_answers, incorrect_texts), start=1):
-        result = "✔" if answer == correct_answer else f"✖️ ({incorrect_text})"
+    for i, (answer, question) in enumerate(zip(answers, test_data['questions']), start=1):
+        result = "✔" if answer == question['correct_answer'] else f"✖️ ({question['incorrect_text']})"
         results.append(f"{i}: {result}")
-    
     return "\n".join(results)
 
-def go_back():
-    return "back"
-
 # css
 css = """
 footer {visibility: hidden !important;}
@@ -49,44 +46,41 @@ with gr.Blocks(css=css, theme='YTheme/TehnoX') as vui:
     with gr.Row():
         test_selector = gr.Radio(label="Выберите тест", choices=list(data['tests'].keys()))
         select_button = gr.Button("Выбрать")
-    
-    test_question_blocks = []
-    for test_name, test_data in data['tests'].items():
-        with gr.Tab(test_name):
-            question_blocks = []
-            correct_answers = []
-            incorrect_texts = []
-            for question in test_data['questions']:
-                radio = gr.Radio(label=question['text'], choices=["А", "Б", "В"], info=question['text'])
-                question_blocks.append(radio)
-                correct_answers.append(question['correct_answer'])
-                incorrect_texts.append(question['incorrect_text'])
+
+    test_tabs = {}
+    for test_name in data['tests']:
+        with gr.Tab(test_name, visible=False) as tab:
+            question_blocks = [
+                gr.Radio(label=question['text'], choices=["А", "Б", "В"], info=question['text'])
+                for question in data['tests'][test_name]['questions']
+            ]
             submit_button = gr.Button("Проверить")
             back_button = gr.Button("Назад")
-            test_question_blocks.append((test_name, question_blocks, correct_answers, incorrect_texts, submit_button, back_button))
+            test_tabs[test_name] = (tab, question_blocks, submit_button, back_button)
 
-    with gr.Tab("Результаты"):
+    results_tab = gr.Tab("Результаты")
+    with results_tab:
         text_output = gr.Markdown("")
 
-    def on_select_test(test_name):
-        for name, _, _, _, _, back_button in test_question_blocks:
-            back_button.click(go_back, inputs=[], outputs=[test_selector])
-            if name == test_name:
-                return {name: gr.update(visible=True)}
-            else:
-                return {name: gr.update(visible=False)}
+    def on_select(test_data, test_name):
+        tab, question_blocks, _, _ = test_tabs[test_name]
+        tab.update(visible=True)
+        test_selector.update(visible=False)
+        select_button.update(visible=False)
+        return ([gr.update(value='') for _ in question_blocks], test_name)
 
-    select_button.click(on_select_test, inputs=[test_selector], outputs=[test_selector])
+    def on_back(test_name):
+        tab, _, _, _ = test_tabs[test_name]
+        tab.update(visible=False)
+        test_selector.update(visible=True)
+        select_button.update(visible=True)
+        return gr.update(value='')
 
-    def on_submit(*args):
-        test_name = args[0]
-        answers = args[1:-3]
-        for name, question_blocks, correct_answers, incorrect_texts, submit_button, _ in test_question_blocks:
-            if name == test_name:
-                return check(answers, correct_answers, incorrect_texts)
+    select_button.click(show_test, inputs=[test_selector], outputs=[gr.Group(*[block for _, blocks, _, _ in test_tabs.values() for block in blocks]), test_selector])
 
-    for test_name, question_blocks, correct_answers, incorrect_texts, submit_button, _ in test_question_blocks:
-        submit_button.click(on_submit, inputs=[test_name] + question_blocks + [submit_button], outputs=[text_output])
+    for test_name, (tab, question_blocks, submit_button, back_button) in test_tabs.items():
+        submit_button.click(check, inputs=question_blocks, outputs=[text_output])
+        back_button.click(on_back, inputs=[test_name], outputs=[test_selector])
 
 #end
 vui.launch()