|
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()
|
|
|
