Update app.py

app.py

@@ -2,18 +2,23 @@ import gradio as gr
 import json
 import os
 
+# Function to get available JSON files in the working directory
+def get_available_weeks():
+    files = [f for f in os.listdir() if f.startswith('week-') and f.endswith('.json')]
+    return files
+
 # Function to load questions from the specified week file
-def load_questions(week):
+def load_questions(week_file):
     try:
-        with open(f"week-{week}.json", "r") as f:
+        with open(week_file, "r") as f:
             data = json.load(f)
         return data, None
     except FileNotFoundError:
-        return None, f"Week {week} file not found."
+        return None, f"File {week_file} not found."
 
 # Flashcard UI function
-def flashcard_ui(week, index):
-    data, error = load_questions(week)
+def flashcard_ui(week_file, index):
+    data, error = load_questions(week_file)
     if error:
         return f"Error: {error}", None
     question = data[index]["question"]
@@ -21,37 +26,48 @@ def flashcard_ui(week, index):
     return f"Question {index + 1}/{total}: {question}", ""
 
 # Reveal answer function
-def reveal_answer(week, index):
-    data, error = load_questions(week)
+def reveal_answer(week_file, index):
+    data, error = load_questions(week_file)
     if error:
         return None
     answer = data[index]["answer"]
     return answer
 
 # Function to handle navigation
-def change_question(week, index, direction):
-    data, _ = load_questions(week)
+def change_question(week_file, index, direction):
+    data, _ = load_questions(week_file)
     total = len(data)
     index = (index + direction) % total
-    return index, *flashcard_ui(week, index)
+    return index, *flashcard_ui(week_file, index)
 
 # Gradio interface
 with gr.Blocks() as demo:
-    week = gr.Number(label="Enter CAM Week to Study", value=1)
-    index = gr.State(0)  # State to keep track of the current index
-    question_output = gr.Textbox(label="Flashcard", interactive=False)
-    answer_output = gr.Textbox(label="Answer", interactive=False)
-
+    available_weeks = get_available_weeks()
+    
+    # Splash page to select the week
     with gr.Row():
-        prev_btn = gr.Button("Previous")
-        reveal_btn = gr.Button("Reveal Answer")
-        next_btn = gr.Button("Next")
-
-    # Event handling
-    week.change(flashcard_ui, inputs=[week, index], outputs=[question_output, answer_output])
-    reveal_btn.click(reveal_answer, inputs=[week, index], outputs=answer_output)
-    prev_btn.click(change_question, inputs=[week, index, gr.Number(-1, visible=False)], outputs=[index, question_output, answer_output])
-    next_btn.click(change_question, inputs=[week, index, gr.Number(1, visible=False)], outputs=[index, question_output, answer_output])
+        week_file = gr.Dropdown(choices=available_weeks, label="Select CAM Week to Study")
+        start_button = gr.Button("Start Studying")
+    
+    # Flashcard UI (hidden until week is selected)
+    with gr.Column(visible=False) as flashcard_section:
+        index = gr.State(0)
+        question_output = gr.Textbox(label="Flashcard", interactive=False)
+        answer_output = gr.Textbox(label="Answer", interactive=False)
+
+        with gr.Row():
+            prev_btn = gr.Button("Previous")
+            reveal_btn = gr.Button("Reveal Answer")
+            next_btn = gr.Button("Next")
+
+    # Start button action to reveal the flashcard section
+    start_button.click(lambda: gr.update(visible=True), inputs=[], outputs=flashcard_section)
+
+    # Event handling for flashcards
+    week_file.change(flashcard_ui, inputs=[week_file, index], outputs=[question_output, answer_output])
+    reveal_btn.click(reveal_answer, inputs=[week_file, index], outputs=answer_output)
+    prev_btn.click(change_question, inputs=[week_file, index, gr.Number(-1, visible=False)], outputs=[index, question_output, answer_output])
+    next_btn.click(change_question, inputs=[week_file, index, gr.Number(1, visible=False)], outputs=[index, question_output, answer_output])
 
 # Launch the app
 demo.launch()