Update app.py

app.py

@@ -13,19 +13,18 @@ api_key = os.getenv("OPENAI_API_KEY")
 client = OpenAI(api_key = api_key)
 
 
-# Import the CrewAI flashcard module (modified below to remove page range)
-from crewai_flashcard import generate_flashcards
 
 # ---------------------------
 # Helper Function: Extract text from PDF
 # ---------------------------
 def extract_text(uploaded_file):
-    # Ensure file size is less than 10MB
+    # Check file size (max 10MB)
     uploaded_file.seek(0, os.SEEK_END)
-    if uploaded_file.tell() > 10 * 1024 * 1024:
-        st.error("File exceeds 10MB limit.")
-        return ""
+    file_size = uploaded_file.tell()
     uploaded_file.seek(0)
+    if file_size > 10 * 1024 * 1024:
+        st.error("File size exceeds 10MB limit.")
+        return ""
     pdf_reader = PdfReader(uploaded_file)
     text = ""
     for page in pdf_reader.pages:
@@ -35,7 +34,7 @@ def extract_text(uploaded_file):
     return text
 
 # ---------------------------
-# OpenAI Response Functions
+# OpenAI Response Functions (using new style)
 # ---------------------------
 def generate_summary_from_text(text):
     prompt = (
@@ -63,8 +62,8 @@ def chat_with_document(text, conversation_history, user_query):
 
 def generate_questions_from_text(text, num_questions):
     prompt = (
-        f"Generate {num_questions} study questions with answers based on the following document. "
-        "Return the output as a table in CSV format with two columns: 'Question' and 'Answer'.\n\nDocument:\n\n{text}"
+        f"Generate up to {num_questions} study questions with answers based on the following document.\n"
+        f"Return the output as a table with two columns: 'Question' and 'Answer'.\n\nDocument:\n\n{text}"
     )
     messages = [
         {"role": "system", "content": "You are an educational assistant that generates study questions."},
@@ -74,30 +73,48 @@ def generate_questions_from_text(text, num_questions):
         model="gpt-4o-mini",
         messages=messages
     )
-    # Expecting CSV output (with header: Question,Answer)
     return completion.choices[0].message.content.strip()
 
+def generate_flashcards_from_text(text, num_cards):
+    prompt = (
+        f"Generate {num_cards} flashcards based on the following document.\n\nDocument:\n\n{text}\n\n"
+        "Return a Python dictionary where each key is a flashcard question and its corresponding value is the answer. "
+        "Do not include any additional text."
+    )
+    messages = [
+        {"role": "system", "content": "You are an educational assistant that creates study flashcards."},
+        {"role": "user", "content": prompt}
+    ]
+    completion = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=messages
+    )
+    output = completion.choices[0].message.content.strip()
+    try:
+        flashcards = literal_eval(output)
+        if isinstance(flashcards, dict):
+            return flashcards
+        else:
+            return {}
+    except Exception as e:
+        st.error(f"Error parsing flashcards: {e}")
+        return {}
+
 # ---------------------------
 # Sidebar: File Upload & Mode Selection
 # ---------------------------
 st.sidebar.title("Study Companion Setup")
 
 uploaded_pdf = st.sidebar.file_uploader("Upload your study PDF (max 10MB)", type="pdf")
-mode = st.sidebar.radio("Select Mode", ("Chat", "Test Your Knowledge", "Flashcards"))
+mode = st.sidebar.radio("Select Mode", ("Chat", "Test Your Knowledge"))# , "Flashcards"))
 
-# For Test Your Knowledge: number of questions (max 50)
+# For Test Your Knowledge and Flashcards modes, allow number input.
 num_questions = None
+num_flashcards = None
 if mode == "Test Your Knowledge":
     num_questions = st.sidebar.number_input("Number of questions to generate (max 50):", min_value=1, max_value=50, value=10, step=1)
-    if st.sidebar.button("Generate Questions"):
-        st.session_state.gen_questions = True
-
-# For Flashcards: number of flashcards (max 5)
-num_flashcards = None
-if mode == "Flashcards":
+elif mode == "Flashcards":
     num_flashcards = st.sidebar.number_input("Number of flashcards to generate (max 5):", min_value=1, max_value=5, value=3, step=1)
-    if st.sidebar.button("Generate Flashcards"):
-        st.session_state.gen_flashcards = True
 
 # ---------------------------
 # Session State Initialization
@@ -123,17 +140,14 @@ if "show_answer" not in st.session_state:
 # Process PDF Upload
 # ---------------------------
 if uploaded_pdf is not None:
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
-        tmp.write(uploaded_pdf.read())
-        st.session_state.pdf_file_path = tmp.name
     st.session_state.pdf_text = extract_text(uploaded_pdf)
     if st.session_state.pdf_text:
         st.sidebar.success("PDF uploaded and processed successfully!")
     else:
-        st.sidebar.error("Failed to extract text from the PDF.")
+        st.sidebar.error("Failed to extract text. Please check your PDF file.")
 
 # ---------------------------
-# Main Area: Mode-Based Display (using side menu)
+# Main Area: Mode-Based Display (all functions via side menu)
 # ---------------------------
 st.title("Study Companion: PDF-based Learning")
 
@@ -142,6 +156,7 @@ if st.session_state.pdf_text is None:
 else:
     if mode == "Chat":
         st.header("Chat with Your Study Companion")
+        # Display persistent chat history
         for msg in st.session_state.chat_history:
             st.chat_message(msg["role"]).write(msg["content"])
         user_question = st.chat_input("Ask a question about the document:")
@@ -155,90 +170,75 @@ else:
 
     elif mode == "Test Your Knowledge":
         st.header("Test Your Knowledge")
-        if num_questions is None or not st.session_state.get("gen_questions", False):
-            st.info("Enter the number of questions and press 'Generate Questions' from the sidebar.")
+        if num_questions is None:
+            st.info("Please specify the number of questions in the sidebar.")
         else:
             with st.spinner("Generating questions..."):
-                questions_csv = generate_questions_from_text(st.session_state.pdf_text, num_questions)
-            # Convert CSV output into a table (assuming header row "Question,Answer")
-            try:
-                lines = questions_csv.splitlines()
-                if len(lines) < 2:
-                    st.error("Failed to generate questions properly.")
-                else:
-                    header = lines[0].split(",")
-                    data = [line.split(",") for line in lines[1:]]
-                    st.table(data, headers=header)
-                    st.session_state.questions_table = data
-            except Exception as e:
-                st.error(f"Error processing questions: {e}")
+                questions_output = generate_questions_from_text(st.session_state.pdf_text, num_questions)
+            # Assume the output is a table in markdown format
+            st.markdown("### Generated Questions")
+            st.markdown(questions_output)
+            # Optionally, you can parse the table and display it with st.table if it's in a CSV-like format.
 
     elif mode == "Flashcards":
         st.header("Practice Flashcards")
-        if not st.session_state.get("gen_flashcards", False):
-            st.info("Enter the number of flashcards and press 'Generate Flashcards' from the sidebar.")
+        if st.button("Generate Flashcards"):
+            with st.spinner("Generating flashcards..."):
+                flashcards = generate_flashcards_from_text(st.session_state.pdf_text, num_flashcards)
+            st.session_state.flashcards = flashcards
+            st.session_state.current_card = 0
+            st.session_state.score = 0
+            st.session_state.show_answer = False
+            st.success("Flashcards generated successfully!")
+        
+        if not st.session_state.flashcards:
+            st.info("No flashcards available. Click the button above to generate flashcards.")
         else:
-            if st.button("Reset Flashcards"):
-                st.session_state.flashcards = {}
-                st.session_state.current_card = 0
-                st.session_state.score = 0
-                st.session_state.show_answer = False
-                st.session_state.gen_flashcards = False
-            if st.session_state.get("gen_flashcards", False):
-                # Generate flashcards using the CrewAI module (which returns a Python dictionary)
-                flashcards = generate_flashcards(st.session_state.pdf_file_path, num_flashcards)
-                st.session_state.flashcards = flashcards
-                st.session_state.current_card = 0
-                st.session_state.score = 0
-                st.session_state.show_answer = False
-                st.success("Flashcards generated successfully!")
-                st.session_state.gen_flashcards = False  # reset flag after generation
-            
-            if not st.session_state.flashcards:
-                st.info("No flashcards available. Click the 'Generate Flashcards' button in the sidebar.")
+            total_cards = len(st.session_state.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 session or generate new flashcards from the sidebar.")
             else:
-                total_cards = len(st.session_state.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 session or generate new flashcards from the sidebar.")
-                else:
-                    flashcards = st.session_state.flashcards
-                    current_keys = list(flashcards.keys())
-                    current_question = current_keys[st.session_state.current_card]
-                    current_answer = flashcards[current_question]
-                    st.write(f"**Question:** {current_question}")
-                    if st.button("Show Answer"):
-                        st.session_state.show_answer = True
-                    if st.session_state.show_answer:
-                        st.write(f"**Answer:** {current_answer}")
-                        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.rerun()
-                    st.write(f"**Current Score:** {st.session_state.score} / {total_cards}")
+                flashcards = st.session_state.flashcards
+                current_keys = list(flashcards.keys())
+                current_question = current_keys[st.session_state.current_card]
+                current_answer = flashcards[current_question]
+                st.write(f"**Question:** {current_question}")
+                if st.button("Show Answer"):
+                    st.session_state.show_answer = True
+                if st.session_state.show_answer:
+                    st.write(f"**Answer:** {current_answer}")
+                    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.rerun()
+                st.write(f"**Current Score:** {st.session_state.score} / {total_cards}")
 
 
 
 
+
+# # Import the CrewAI flashcard module (modified below to remove page range)
+# from crewai_flashcard import generate_flashcards
+
 # # ---------------------------
 # # Helper Function: Extract text from PDF
 # # ---------------------------
 # def extract_text(uploaded_file):
-#     # Check file size (max 10MB)
+#     # Ensure file size is less than 10MB
 #     uploaded_file.seek(0, os.SEEK_END)
-#     file_size = uploaded_file.tell()
-#     uploaded_file.seek(0)
-#     if file_size > 10 * 1024 * 1024:
-#         st.error("File size exceeds 10MB limit.")
+#     if uploaded_file.tell() > 10 * 1024 * 1024:
+#         st.error("File exceeds 10MB limit.")
 #         return ""
+#     uploaded_file.seek(0)
 #     pdf_reader = PdfReader(uploaded_file)
 #     text = ""
 #     for page in pdf_reader.pages:
@@ -248,7 +248,7 @@ else:
 #     return text
 
 # # ---------------------------
-# # OpenAI Response Functions (using new style)
+# # OpenAI Response Functions
 # # ---------------------------
 # def generate_summary_from_text(text):
 #     prompt = (
@@ -276,8 +276,8 @@ else:
 
 # def generate_questions_from_text(text, num_questions):
 #     prompt = (
-#         f"Generate up to {num_questions} study questions with answers based on the following document.\n"
-#         f"Return the output as a table with two columns: 'Question' and 'Answer'.\n\nDocument:\n\n{text}"
+#         f"Generate {num_questions} study questions with answers based on the following document. "
+#         "Return the output as a table in CSV format with two columns: 'Question' and 'Answer'.\n\nDocument:\n\n{text}"
 #     )
 #     messages = [
 #         {"role": "system", "content": "You are an educational assistant that generates study questions."},
@@ -287,33 +287,9 @@ else:
 #         model="gpt-4o-mini",
 #         messages=messages
 #     )
+#     # Expecting CSV output (with header: Question,Answer)
 #     return completion.choices[0].message.content.strip()
 
-# def generate_flashcards_from_text(text, num_cards):
-#     prompt = (
-#         f"Generate {num_cards} flashcards based on the following document.\n\nDocument:\n\n{text}\n\n"
-#         "Return a Python dictionary where each key is a flashcard question and its corresponding value is the answer. "
-#         "Do not include any additional text."
-#     )
-#     messages = [
-#         {"role": "system", "content": "You are an educational assistant that creates study flashcards."},
-#         {"role": "user", "content": prompt}
-#     ]
-#     completion = client.chat.completions.create(
-#         model="gpt-4o-mini",
-#         messages=messages
-#     )
-#     output = completion.choices[0].message.content.strip()
-#     try:
-#         flashcards = literal_eval(output)
-#         if isinstance(flashcards, dict):
-#             return flashcards
-#         else:
-#             return {}
-#     except Exception as e:
-#         st.error(f"Error parsing flashcards: {e}")
-#         return {}
-
 # # ---------------------------
 # # Sidebar: File Upload & Mode Selection
 # # ---------------------------
@@ -322,13 +298,19 @@ else:
 # uploaded_pdf = st.sidebar.file_uploader("Upload your study PDF (max 10MB)", type="pdf")
 # mode = st.sidebar.radio("Select Mode", ("Chat", "Test Your Knowledge", "Flashcards"))
 
-# # For Test Your Knowledge and Flashcards modes, allow number input.
+# # For Test Your Knowledge: number of questions (max 50)
 # num_questions = None
-# num_flashcards = None
 # if mode == "Test Your Knowledge":
 #     num_questions = st.sidebar.number_input("Number of questions to generate (max 50):", min_value=1, max_value=50, value=10, step=1)
-# elif mode == "Flashcards":
+#     if st.sidebar.button("Generate Questions"):
+#         st.session_state.gen_questions = True
+
+# # For Flashcards: number of flashcards (max 5)
+# num_flashcards = None
+# if mode == "Flashcards":
 #     num_flashcards = st.sidebar.number_input("Number of flashcards to generate (max 5):", min_value=1, max_value=5, value=3, step=1)
+#     if st.sidebar.button("Generate Flashcards"):
+#         st.session_state.gen_flashcards = True
 
 # # ---------------------------
 # # Session State Initialization
@@ -354,14 +336,17 @@ else:
 # # Process PDF Upload
 # # ---------------------------
 # if uploaded_pdf is not None:
+#     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+#         tmp.write(uploaded_pdf.read())
+#         st.session_state.pdf_file_path = tmp.name
 #     st.session_state.pdf_text = extract_text(uploaded_pdf)
 #     if st.session_state.pdf_text:
 #         st.sidebar.success("PDF uploaded and processed successfully!")
 #     else:
-#         st.sidebar.error("Failed to extract text. Please check your PDF file.")
+#         st.sidebar.error("Failed to extract text from the PDF.")
 
 # # ---------------------------
-# # Main Area: Mode-Based Display (all functions via side menu)
+# # Main Area: Mode-Based Display (using side menu)
 # # ---------------------------
 # st.title("Study Companion: PDF-based Learning")
 
@@ -370,7 +355,6 @@ else:
 # else:
 #     if mode == "Chat":
 #         st.header("Chat with Your Study Companion")
-#         # Display persistent chat history
 #         for msg in st.session_state.chat_history:
 #             st.chat_message(msg["role"]).write(msg["content"])
 #         user_question = st.chat_input("Ask a question about the document:")
@@ -384,57 +368,78 @@ else:
 
 #     elif mode == "Test Your Knowledge":
 #         st.header("Test Your Knowledge")
-#         if num_questions is None:
-#             st.info("Please specify the number of questions in the sidebar.")
+#         if num_questions is None or not st.session_state.get("gen_questions", False):
+#             st.info("Enter the number of questions and press 'Generate Questions' from the sidebar.")
 #         else:
 #             with st.spinner("Generating questions..."):
-#                 questions_output = generate_questions_from_text(st.session_state.pdf_text, num_questions)
-#             # Assume the output is a table in markdown format
-#             st.markdown("### Generated Questions")
-#             st.markdown(questions_output)
-#             # Optionally, you can parse the table and display it with st.table if it's in a CSV-like format.
+#                 questions_csv = generate_questions_from_text(st.session_state.pdf_text, num_questions)
+#             # Convert CSV output into a table (assuming header row "Question,Answer")
+#             try:
+#                 lines = questions_csv.splitlines()
+#                 if len(lines) < 2:
+#                     st.error("Failed to generate questions properly.")
+#                 else:
+#                     header = lines[0].split(",")
+#                     data = [line.split(",") for line in lines[1:]]
+#                     st.table(data, headers=header)
+#                     st.session_state.questions_table = data
+#             except Exception as e:
+#                 st.error(f"Error processing questions: {e}")
 
 #     elif mode == "Flashcards":
 #         st.header("Practice Flashcards")
-#         if st.button("Generate Flashcards"):
-#             with st.spinner("Generating flashcards..."):
-#                 flashcards = generate_flashcards_from_text(st.session_state.pdf_text, num_flashcards)
-#             st.session_state.flashcards = flashcards
-#             st.session_state.current_card = 0
-#             st.session_state.score = 0
-#             st.session_state.show_answer = False
-#             st.success("Flashcards generated successfully!")
-        
-#         if not st.session_state.flashcards:
-#             st.info("No flashcards available. Click the button above to generate flashcards.")
+#         if not st.session_state.get("gen_flashcards", False):
+#             st.info("Enter the number of flashcards and press 'Generate Flashcards' from the sidebar.")
 #         else:
-#             total_cards = len(st.session_state.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 session or generate new flashcards from the sidebar.")
+#             if st.button("Reset Flashcards"):
+#                 st.session_state.flashcards = {}
+#                 st.session_state.current_card = 0
+#                 st.session_state.score = 0
+#                 st.session_state.show_answer = False
+#                 st.session_state.gen_flashcards = False
+#             if st.session_state.get("gen_flashcards", False):
+#                 # Generate flashcards using the CrewAI module (which returns a Python dictionary)
+#                 flashcards = generate_flashcards(st.session_state.pdf_file_path, num_flashcards)
+#                 st.session_state.flashcards = flashcards
+#                 st.session_state.current_card = 0
+#                 st.session_state.score = 0
+#                 st.session_state.show_answer = False
+#                 st.success("Flashcards generated successfully!")
+#                 st.session_state.gen_flashcards = False  # reset flag after generation
+            
+#             if not st.session_state.flashcards:
+#                 st.info("No flashcards available. Click the 'Generate Flashcards' button in the sidebar.")
 #             else:
-#                 flashcards = st.session_state.flashcards
-#                 current_keys = list(flashcards.keys())
-#                 current_question = current_keys[st.session_state.current_card]
-#                 current_answer = flashcards[current_question]
-#                 st.write(f"**Question:** {current_question}")
-#                 if st.button("Show Answer"):
-#                     st.session_state.show_answer = True
-#                 if st.session_state.show_answer:
-#                     st.write(f"**Answer:** {current_answer}")
-#                     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.rerun()
-#                 st.write(f"**Current Score:** {st.session_state.score} / {total_cards}")
+#                 total_cards = len(st.session_state.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 session or generate new flashcards from the sidebar.")
+#                 else:
+#                     flashcards = st.session_state.flashcards
+#                     current_keys = list(flashcards.keys())
+#                     current_question = current_keys[st.session_state.current_card]
+#                     current_answer = flashcards[current_question]
+#                     st.write(f"**Question:** {current_question}")
+#                     if st.button("Show Answer"):
+#                         st.session_state.show_answer = True
+#                     if st.session_state.show_answer:
+#                         st.write(f"**Answer:** {current_answer}")
+#                         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.rerun()
+#                     st.write(f"**Current Score:** {st.session_state.score} / {total_cards}")
+
+
+
 
 ######################################################################################################