Update app.py

app.py

@@ -54,45 +54,85 @@ with tab_gen:
 with tab_practice:
     st.header("Practice Flashcards")
     if st.session_state.flashcards:
-        # Optionally, randomize flashcard order on first run
-        if "shuffled" not in st.session_state:
-            random.shuffle(st.session_state.flashcards)
-            st.session_state.shuffled = True
-        
-        current_card = st.session_state.flashcards[st.session_state.current_card]
-        st.write(f"**Question:** {current_card.get('question', 'No question provided')}")
-        
-        # Button to reveal the answer
-        if st.button("Show Answer"):
-            st.session_state.show_answer = True
-        
-        # If answer is revealed, show answer and feedback options
-        if st.session_state.show_answer:
-            st.write(f"**Answer:** {current_card.get('answer', 'No answer provided')}")
-            col1, col2 = st.columns(2)
-            with col1:
-                if st.button("Correct"):
-                    st.session_state.score += 1
-                    st.success("Correct!")
-            with col2:
-                if st.button("Wrong"):
-                    st.error("Incorrect!")
-
-            # Navigation: if this is the last card, show final score; otherwise, go to next card.
+        total_cards = len(st.session_state.flashcards)
+        # Check if we've completed all flashcards
+        if st.session_state.current_card >= total_cards:
+            st.success(f"You've completed all flashcards! Final Score: {st.session_state.score} / {total_cards}")
+            st.info("Restart the practice session or generate new flashcards.")
+        else:
+            # Safe access: ensure current_card is valid
+            current_card = st.session_state.flashcards[st.session_state.current_card]
+            st.write(f"**Question:** {current_card.get('question', 'No question provided')}")
+            
+            # Answer reveal button
+            if st.button("Show Answer"):
+                st.session_state.show_answer = True
+            
+            # If answer is revealed, show it and provide feedback options
+            if st.session_state.show_answer:
+                st.write(f"**Answer:** {current_card.get('answer', 'No answer provided')}")
+                col1, col2 = st.columns(2)
+                with col1:
+                    if st.button("Correct"):
+                        st.session_state.score += 1
+                        st.success("Correct!")
+                with col2:
+                    if st.button("Wrong"):
+                        st.error("Incorrect!")
+                
                 if st.button("Next Card"):
                     st.session_state.current_card += 1
                     st.session_state.show_answer = False
-                    #st.experimental_rerun()
+                    st.rerun()
             
-            st.write(f"**Current Score:** {st.session_state.score}/{len(st.session_state.flashcards)}")
+            st.write(f"**Current Score:** {st.session_state.score}/{total_cards}")
     else:
         st.info("No flashcards available. Please generate flashcards first.")
+
+
+# # --- Tab 2: Practice Flashcards ---
+# with tab_practice:
+#     st.header("Practice Flashcards")
+#     if st.session_state.flashcards:
+#         # Optionally, randomize flashcard order on first run
+#         if "shuffled" not in st.session_state:
+#             random.shuffle(st.session_state.flashcards)
+#             st.session_state.shuffled = True
+        
+#         current_card = st.session_state.flashcards[st.session_state.current_card]
+#         st.write(f"**Question:** {current_card.get('question', 'No question provided')}")
+        
+#         # Button to reveal the answer
+#         if st.button("Show Answer"):
+#             st.session_state.show_answer = True
+        
+#         # If answer is revealed, show answer and feedback options
+#         if st.session_state.show_answer:
+#             st.write(f"**Answer:** {current_card.get('answer', 'No answer provided')}")
+#             col1, col2 = st.columns(2)
+#             with col1:
+#                 if st.button("Correct"):
+#                     st.session_state.score += 1
+#                     st.success("Correct!")
+#             with col2:
+#                 if st.button("Wrong"):
+#                     st.error("Incorrect!")
+
+#             # Navigation: if this is the last card, show final score; otherwise, go to next card.
+#                 if st.button("Next Card"):
+#                     st.session_state.current_card += 1
+#                     st.session_state.show_answer = False
+#                     #st.experimental_rerun()
+            
+#             st.write(f"**Current Score:** {st.session_state.score}/{len(st.session_state.flashcards)}")
+#     else:
+#         st.info("No flashcards available. Please generate flashcards first.")
             
-    #         if st.button("Next Card"):
-    #             st.session_state.current_card = (st.session_state.current_card + 1) % len(st.session_state.flashcards)
-    #             st.session_state.show_answer = False
-    #             st.experimental_rerun()
+#     #         if st.button("Next Card"):
+#     #             st.session_state.current_card = (st.session_state.current_card + 1) % len(st.session_state.flashcards)
+#     #             st.session_state.show_answer = False
+#     #             st.experimental_rerun()
         
-    #     st.write(f"**Current Score:** {st.session_state.score}/{len(st.session_state.flashcards)}")
-    # else:
-    #     st.info("No flashcards available. Please generate flashcards first.")
+#     #     st.write(f"**Current Score:** {st.session_state.score}/{len(st.session_state.flashcards)}")
+#     # else:
+#     #     st.info("No flashcards available. Please generate flashcards first.")