|
import random |
|
|
|
import gradio as gr |
|
|
|
demo = gr.Blocks() |
|
|
|
def is_data_empty(flashcards): |
|
if isinstance(flashcards, dict): |
|
return all(item == '' for sublist in flashcards['data'] for item in sublist) |
|
elif isinstance(flashcards, list): |
|
return all(all(item == '' for item in sublist) for sublist in flashcards) |
|
else: |
|
return True |
|
|
|
with demo: |
|
gr.Markdown( |
|
"Load the flashcards in the table below, then use the Practice tab to practice." |
|
) |
|
|
|
with gr.Tabs() as tabs: |
|
with gr.Tab("Word Bank"): |
|
flashcards_table = gr.Dataframe(headers=["front", "back"], type="array") |
|
flashcards_table.change(fn=lambda: print(flashcards_table.value)) |
|
practice_btn = gr.Button("Start Practice") |
|
with gr.Tab("Practice", interactive=False, id=1) as practice_tab: |
|
with gr.Row(): |
|
with gr.Column(): |
|
front = gr.Textbox(label="Prompt") |
|
with gr.Row(): |
|
new_btn = gr.Button("New Card") |
|
flip_btn = gr.Button("Flip Card") |
|
with gr.Column(visible=False) as answer_col: |
|
back = gr.Textbox(label="Answer") |
|
selected_card = gr.State() |
|
with gr.Row(): |
|
correct_btn = gr.Button("Correct") |
|
incorrect_btn = gr.Button("Incorrect") |
|
|
|
def start_practice(flashcards): |
|
|
|
if is_data_empty(flashcards): |
|
practice_tab = gr.Tab("Practice", interactive=False, id=1) |
|
raise gr.Error("Please enter word prompts into the table.") |
|
return [practice_tab, tabs] |
|
else: |
|
practice_tab = gr.Tab("Practice", interactive=True, id=1) |
|
new_tabs = gr.Tabs(selected=1) |
|
return [practice_tab, new_tabs] |
|
|
|
with gr.Tab("Results", visible=False) as results_tab: |
|
results = gr.State(value={}) |
|
correct_field = gr.Markdown("# Correct: 0") |
|
incorrect_field = gr.Markdown("# Incorrect: 0") |
|
gr.Markdown("Card Statistics: ") |
|
results_table = gr.Dataframe(headers=["Card", "Correct", "Incorrect"]) |
|
practice_btn.click(start_practice, inputs=[flashcards_table], outputs=[practice_tab, tabs]) |
|
|
|
def load_new_card(flashcards): |
|
card = random.choice(flashcards) |
|
return ( |
|
card, |
|
card[0], |
|
gr.Column(visible=False), |
|
) |
|
|
|
new_btn.click( |
|
load_new_card, |
|
[flashcards_table], |
|
[selected_card, front, answer_col], |
|
) |
|
|
|
def flip_card(card): |
|
return card[1], gr.Column(visible=True) |
|
|
|
flip_btn.click(flip_card, [selected_card], [back, answer_col]) |
|
|
|
def mark_correct(card, results): |
|
if card[0] not in results: |
|
results[card[0]] = [0, 0] |
|
results[card[0]][0] += 1 |
|
correct_count = sum(result[0] for result in results.values()) |
|
return ( |
|
results, |
|
f"# Correct: {correct_count}", |
|
[[front, scores[0], scores[1]] for front, scores in results.items()], |
|
) |
|
|
|
def mark_incorrect(card, results): |
|
if card[0] not in results: |
|
results[card[0]] = [ |
|
0, 0] |
|
results[card[0]][1] += 1 |
|
incorrect_count = sum(result[1] for result in results.values()) |
|
return ( |
|
results, |
|
f"# Inorrect: {incorrect_count}", |
|
[[front, scores[0], scores[1]] for front, scores in results.items()], |
|
) |
|
|
|
def toggle_results_tab(): |
|
return gr.Tab("Results", visible=True) |
|
|
|
correct_btn.click( |
|
mark_correct, |
|
[selected_card, results], |
|
[results, correct_field, results_table], |
|
) |
|
|
|
incorrect_btn.click(mark_incorrect, [selected_card, results], [results, incorrect_field, results_table]) |
|
|
|
|
|
correct_btn.click(fn=toggle_results_tab, outputs=[results_tab]) |
|
incorrect_btn.click(fn=toggle_results_tab, outputs=[results_tab]) |
|
|
|
if __name__ == "__main__": |
|
demo.launch() |
|
|
