Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import json
 import os
+import random
 import google.generativeai as genai
 
 # Function to get available JSON files in the working directory
@@ -41,83 +42,104 @@ def change_question(week_file, index, direction):
     index = (index + direction) % total
     return index, *flashcard_ui(week_file, index)
 
-with open("aa2.txt", "r") as f:
-  prompt = f.read()
+# Function to load mock test questions
+def load_mocktest():
+    try:
+        with open("mocktest.json", "r") as f:
+            data = json.load(f)
+        return data
+    except FileNotFoundError:
+        return None
 
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+# Function to display the mock test question and answers
+def display_mock_question(index, score, incorrect_questions):
+    data = load_mocktest()
+    if not data:
+        return "Mocktest file not found.", None, None, None, None, None
+    
+    question_data = data[index]
+    question = f"Question {index + 1}: {question_data['question']}"
+    correct_answer = question_data["correct-answer"]
+    all_answers = question_data["wrong-answers"] + [correct_answer]
+    random.shuffle(all_answers)
+    
+    return question, all_answers[0], all_answers[1], all_answers[2], all_answers[3], correct_answer
+
+# Function to handle answer submission
+def handle_answer(index, answer, score, incorrect_questions):
+    data = load_mocktest()
+    correct_answer = data[index]["correct-answer"]
+    
+    if answer == correct_answer:
+        score += 1
+    else:
+        incorrect_questions.append(data[index]["question"])
     
-# Create the model configuration
-generation_config = {
-    "temperature": 0.5,
-    "top_p": 0.98,
-    "top_k": 64,
-    "max_output_tokens": 8192,
-    "response_mime_type": "text/plain",
-}
-
-model = genai.GenerativeModel(
-    model_name="gemini-1.5-flash-8b-exp-0827",
-    generation_config=generation_config,
-)
-
-# AI Chatbot function
-def chat_with_ai(user_input):
-    chat_session = model.start_chat(
-      history=[
-        {
-          "role": "user",
-          "parts": [ prompt ],
-        },
-        {
-          "role": "model",
-          "parts": [
-            "Sure, I can answer your question. \n",
-          ],
-        },
-      ]
-    )
+    index += 1
+    if index >= len(data):
+        return gr.update(visible=False), f"Your score: {score}/{len(data)}", incorrect_questions, gr.update(visible=True)
     
-    response = chat_session.send_message(user_input)
-    return response.text
+    return *display_mock_question(index, score, incorrect_questions), score, incorrect_questions, gr.update(visible=True)
+
+# Function to restart the test
+def restart_test():
+    return 0, "", [], *display_mock_question(0, 0, []), gr.update(visible=False), gr.update(visible=True)
 
 # Gradio interface
 with gr.Blocks() as demo:
     available_weeks = get_available_weeks()
     
-    # Splash page to select the week
-    with gr.Row():
-        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")
-    
-    with gr.Column(visible=True):
-        chat_input = gr.Textbox(label="Ask a question to the A.I.")
-        with gr.Row():
-            gr.Markdown(">Note: This A.I. has only been given the NBAA Management guide. Use the flashcards for specific week information.")
-        chat_output = gr.Markdown(label="AI Response")
-        chat_button = gr.Button("Ask AI")
-
-    # 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])
-
-    # Chatbot button action
-    chat_button.click(chat_with_ai, inputs=[chat_input], outputs=[chat_output])
+    # Tabs for Flashcards and Mock Test
+    with gr.Tabs():
+        with gr.TabItem("Flashcards"):
+            # Splash page to select the week
+            with gr.Row():
+                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)
+            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])
+        
+        with gr.TabItem("Mock Test"):
+            mock_index = gr.State(0)
+            score = gr.State(0)
+            incorrect_questions = gr.State([])
+
+            question_output = gr.Markdown(label="Question")
+            answer_1 = gr.Button("Answer 1")
+            answer_2 = gr.Button("Answer 2")
+            answer_3 = gr.Button("Answer 3")
+            answer_4 = gr.Button("Answer 4")
+
+            result_output = gr.Markdown(visible=False)
+            restart_button = gr.Button("Restart Test", visible=False)
+            
+            # Initialize the first question
+            demo.load(display_mock_question, inputs=[mock_index, score, incorrect_questions], outputs=[question_output, answer_1, answer_2, answer_3, answer_4])
+
+            # Handle answer selection
+            answer_1.click(handle_answer, inputs=[mock_index, answer_1, score, incorrect_questions], outputs=[question_output, answer_1, answer_2, answer_3, answer_4, score, incorrect_questions, result_output])
+            answer_2.click(handle_answer, inputs=[mock_index, answer_2, score, incorrect_questions], outputs=[question_output, answer_1, answer_2, answer_3, answer_4, score, incorrect_questions, result_output])
+            answer_3.click(handle_answer, inputs=[mock_index, answer_3, score, incorrect_questions], outputs=[question_output, answer_1, answer_2, answer_3, answer_4, score, incorrect_questions, result_output])
+            answer_4.click(handle_answer, inputs=[mock_index, answer_4, score, incorrect_questions], outputs=[question_output, answer_1, answer_2, answer_3, answer_4, score, incorrect_questions, result_output])
+
+            # Restart test button
+            restart_button.click(restart_test, inputs=[], outputs=[mock_index, score, incorrect_questions, question_output, answer_1, answer_2, answer_3, answer_4, result_output, restart_button])
 
 # Launch the app
 demo.launch()