Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,1144 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import tempfile
+import json
+import random
+from pathlib import Path
+from PyPDF2 import PdfReader
+from openai import OpenAI
+import os
+from ast import literal_eval
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# Initialize the OpenAI client
+api_key = os.getenv("OPENAI_API_KEY")
+client = OpenAI(api_key = api_key)
+
+
+
+# ---------------------------
+# Helper Function: Extract text from PDF
+# ---------------------------
+def extract_text(uploaded_file):
+    # Check file size (max 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.")
+        return ""
+    pdf_reader = PdfReader(uploaded_file)
+    text = ""
+    for page in pdf_reader.pages:
+        page_text = page.extract_text()
+        if page_text:
+            text += page_text + "\n"
+    return text
+
+# ---------------------------
+# OpenAI Response Functions (using new style)
+# ---------------------------
+def generate_summary_from_text(text):
+    prompt = (
+        f"Summarize the following document in a concise manner, highlighting the key points that a student should know:\n\n{text}"
+    )
+    messages = [
+        {"role": "system", "content": "You are an educational assistant."},
+        {"role": "user", "content": prompt}
+    ]
+    completion = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=messages
+    )
+    return completion.choices[0].message.content.strip()
+
+def chat_with_document(text, conversation_history, user_query):
+    messages = conversation_history + [
+        {"role": "user", "content": f"Based on the following document:\n\n{text}\n\nQuestion: {user_query}"}
+    ]
+    completion = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=messages
+    )
+    return completion.choices[0].message.content.strip()
+
+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}"
+    )
+    messages = [
+        {"role": "system", "content": "You are an educational assistant that generates study questions."},
+        {"role": "user", "content": prompt}
+    ]
+    completion = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=messages
+    )
+    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.markdown("""
+<style>
+.sidebar-title {
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    padding: 20px;
+    border-radius: 10px;
+    color: white;
+    text-align: center;
+    margin-bottom: 20px;
+    font-size: 1.3em;
+    font-weight: bold;
+}
+</style>
+<div class="sidebar-title">
+    ⚙️ Study Companion Setup
+</div>
+""", unsafe_allow_html=True)
+
+uploaded_pdf = st.sidebar.file_uploader("📄 Upload your study PDF (max 10MB)", type="pdf")
+st.sidebar.markdown("---")
+mode = st.sidebar.radio("🎯 Select Mode", ("Chat", "Test Your Knowledge", "Flashcards"))
+
+# 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)
+elif mode == "Flashcards":
+    num_flashcards = st.sidebar.number_input("Number of flashcards to generate (max 20):", min_value=1, max_value=20, value=5, step=1)
+
+# ---------------------------
+# Session State Initialization
+# ---------------------------
+if "pdf_text" not in st.session_state:
+    st.session_state.pdf_text = None
+if "summary" not in st.session_state:
+    st.session_state.summary = None
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = [{"role": "assistant", "content": "Hi, how can I help you with your study material?"}]
+if "questions_table" not in st.session_state:
+    st.session_state.questions_table = None
+if "flashcards" not in st.session_state:
+    st.session_state.flashcards = {}
+if "current_card" not in st.session_state:
+    st.session_state.current_card = 0
+if "score" not in st.session_state:
+    st.session_state.score = 0
+if "show_answer" not in st.session_state:
+    st.session_state.show_answer = False
+if "flashcard_keys" not in st.session_state:
+    st.session_state.flashcard_keys = []
+if "user_answers" not in st.session_state:
+    st.session_state.user_answers = {}
+if "shuffle_cards" not in st.session_state:
+    st.session_state.shuffle_cards = False
+
+# ---------------------------
+# Process PDF Upload
+# ---------------------------
+if uploaded_pdf is not None:
+    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.")
+
+# ---------------------------
+# Main Area: Mode-Based Display (all functions via side menu)
+# ---------------------------
+# Page configuration and custom styling
+st.markdown("""
+<style>
+/* Global styling */
+.main {
+    background-color: #f5f7fa;
+}
+
+/* Header styling */
+.main-header {
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    padding: 30px;
+    border-radius: 15px;
+    text-align: center;
+    color: white;
+    margin-bottom: 30px;
+    box-shadow: 0 5px 20px rgba(0,0,0,0.2);
+}
+
+.main-header h1 {
+    margin: 0;
+    font-size: 2.5em;
+}
+
+.main-header p {
+    margin: 10px 0 0 0;
+    font-size: 1.1em;
+    opacity: 0.9;
+}
+
+/* Chat message styling */
+.stChatMessage {
+    background-color: white;
+    border-radius: 10px;
+    padding: 15px;
+    margin: 10px 0;
+    box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+}
+
+/* Info box styling */
+.stInfo {
+    background-color: #e3f2fd;
+    border-left: 5px solid #2196f3;
+}
+
+/* Success box styling */
+.stSuccess {
+    background-color: #e8f5e9;
+    border-left: 5px solid #4caf50;
+}
+
+/* Questions table styling */
+table {
+    background-color: white;
+    border-radius: 10px;
+    overflow: hidden;
+    box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+}
+
+th {
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    color: white;
+    padding: 15px;
+}
+
+td {
+    padding: 12px;
+    border-bottom: 1px solid #e0e0e0;
+}
+
+/* Sidebar styling */
+.css-1d391kg {
+    background-color: #f8f9fa;
+}
+</style>
+""", unsafe_allow_html=True)
+
+st.markdown("""
+<div class="main-header">
+    <h1>📚 Study Companion</h1>
+    <p>Your AI-powered learning assistant</p>
+</div>
+""", unsafe_allow_html=True)
+
+if st.session_state.pdf_text is None:
+    st.info("Please upload a PDF from the sidebar to begin.")
+else:
+    if mode == "Chat":
+        st.header("💬 Chat with Your Study Companion")
+        st.markdown("""
+        <style>
+        .chat-container {
+            background: white;
+            border-radius: 10px;
+            padding: 20px;
+            margin: 15px 0;
+            box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+        }
+        </style>
+        """, unsafe_allow_html=True)
+        
+        # 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...")
+        if user_question:
+            st.session_state.chat_history.append({"role": "user", "content": user_question})
+            st.chat_message("user").write(user_question)
+            with st.spinner("🤔 Thinking..."):
+                response = chat_with_document(st.session_state.pdf_text, st.session_state.chat_history, user_question)
+            st.session_state.chat_history.append({"role": "assistant", "content": response})
+            st.chat_message("assistant").write(response)
+        
+        # Add a clear chat button
+        if len(st.session_state.chat_history) > 1:
+            if st.button("🗑️ Clear Chat History"):
+                st.session_state.chat_history = [{"role": "assistant", "content": "Hi, how can I help you with your study material?"}]
+                st.rerun()
+
+    elif mode == "Test Your Knowledge":
+        st.header("📝 Test Your Knowledge")
+        st.markdown("""
+        <style>
+        .question-container {
+            background: white;
+            border-radius: 10px;
+            padding: 20px;
+            margin: 15px 0;
+            box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+        }
+        </style>
+        """, unsafe_allow_html=True)
+        
+        if num_questions is None:
+            st.info("Please specify the number of questions in the sidebar.")
+        else:
+            if st.button("📋 Generate Questions", use_container_width=True, type="primary"):
+                with st.spinner("✨ Generating questions..."):
+                    questions_output = generate_questions_from_text(st.session_state.pdf_text, num_questions)
+                st.session_state.questions_table = questions_output
+            
+            if st.session_state.questions_table:
+                st.markdown('<div class="question-container">', unsafe_allow_html=True)
+                st.markdown(st.session_state.questions_table)
+                st.markdown('</div>', unsafe_allow_html=True)
+                
+                # Download button for questions
+                st.download_button(
+                    label="📥 Download Questions",
+                    data=st.session_state.questions_table,
+                    file_name="study_questions.md",
+                    mime="text/markdown"
+                )
+            else:
+                st.info("💡 Click 'Generate Questions' to create a quiz based on your document.")
+
+    elif mode == "Flashcards":
+        st.header("🎴 Interactive Flashcards")
+        
+        # Custom CSS for flashcard styling
+        st.markdown("""
+        <style>
+        .flashcard-container {
+            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+            border-radius: 15px;
+            padding: 40px;
+            margin: 20px 0;
+            box-shadow: 0 10px 30px rgba(0,0,0,0.3);
+            min-height: 250px;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            text-align: center;
+            transition: transform 0.3s ease;
+        }
+        .flashcard-container:hover {
+            transform: translateY(-5px);
+        }
+        .flashcard-question {
+            color: white;
+            font-size: 24px;
+            font-weight: bold;
+            margin-bottom: 20px;
+        }
+        .flashcard-answer {
+            background: rgba(255,255,255,0.95);
+            border-radius: 10px;
+            padding: 25px;
+            margin-top: 20px;
+            color: #333;
+            font-size: 18px;
+            box-shadow: 0 4px 15px rgba(0,0,0,0.2);
+        }
+        .score-card {
+            background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%);
+            border-radius: 10px;
+            padding: 20px;
+            color: white;
+            font-size: 20px;
+            font-weight: bold;
+            text-align: center;
+            margin: 20px 0;
+        }
+        .progress-bar-container {
+            background: #e0e0e0;
+            border-radius: 10px;
+            height: 20px;
+            margin: 20px 0;
+            overflow: hidden;
+        }
+        .progress-bar {
+            background: linear-gradient(90deg, #4facfe 0%, #00f2fe 100%);
+            height: 100%;
+            transition: width 0.3s ease;
+        }
+        .card-number {
+            color: white;
+            font-size: 16px;
+            margin-bottom: 10px;
+            opacity: 0.9;
+        }
+        .stButton > button {
+            border-radius: 8px;
+            font-weight: 600;
+            padding: 10px 24px;
+            transition: all 0.3s ease;
+        }
+        .stButton > button:hover {
+            transform: scale(1.05);
+            box-shadow: 0 4px 12px rgba(0,0,0,0.2);
+        }
+        </style>
+        """, unsafe_allow_html=True)
+        
+        # Sidebar controls for flashcards
+        with st.sidebar:
+            st.markdown("---")
+            st.subheader("🎴 Flashcard Options")
+            shuffle_option = st.checkbox("Shuffle cards", value=st.session_state.shuffle_cards)
+            if shuffle_option != st.session_state.shuffle_cards:
+                st.session_state.shuffle_cards = shuffle_option
+                if st.session_state.flashcards and st.session_state.flashcard_keys:
+                    if shuffle_option:
+                        st.session_state.flashcard_keys = list(st.session_state.flashcards.keys())
+                        random.shuffle(st.session_state.flashcard_keys)
+                    else:
+                        st.session_state.flashcard_keys = list(st.session_state.flashcards.keys())
+        
+        if num_flashcards is None:
+            st.info("Please specify the number of flashcards in the sidebar.")
+        else:
+            col1, col2 = st.columns([3, 1])
+            with col1:
+                if st.button("🎯 Generate Flashcards", use_container_width=True):
+                    with st.spinner("✨ Creating your flashcards..."):
+                        flashcards = generate_flashcards_from_text(st.session_state.pdf_text, num_flashcards)
+                    if flashcards:
+                        st.session_state.flashcards = flashcards
+                        st.session_state.flashcard_keys = list(flashcards.keys())
+                        if st.session_state.shuffle_cards:
+                            random.shuffle(st.session_state.flashcard_keys)
+                        st.session_state.current_card = 0
+                        st.session_state.score = 0
+                        st.session_state.show_answer = False
+                        st.session_state.user_answers = {}
+                        st.success("✅ Flashcards generated successfully!")
+                        st.rerun()
+                    else:
+                        st.error("Failed to generate flashcards. Please try again.")
+            
+            with col2:
+                if st.session_state.flashcards and st.button("🔄 Reset", use_container_width=True):
+                    st.session_state.current_card = 0
+                    st.session_state.score = 0
+                    st.session_state.show_answer = False
+                    st.session_state.user_answers = {}
+                    st.rerun()
+            
+            if not st.session_state.flashcards:
+                st.info("💡 No flashcards available. Click 'Generate Flashcards' to create a study set based on your document.")
+            else:
+                total_cards = len(st.session_state.flashcards)
+                
+                if st.session_state.current_card >= total_cards:
+                    # Completion screen
+                    st.balloons()
+                    st.markdown(f"""
+                    <div class="score-card">
+                        🎉 Congratulations! You've completed all flashcards!<br>
+                        Final Score: {st.session_state.score} / {total_cards} ({int(st.session_state.score/total_cards*100)}%)
+                    </div>
+                    """, unsafe_allow_html=True)
+                    
+                    # Show review of incorrect answers
+                    if st.session_state.user_answers:
+                        st.subheader("📝 Review Your Answers")
+                        for i, (q, is_correct) in enumerate(st.session_state.user_answers.items(), 1):
+                            if not is_correct:
+                                with st.expander(f"❌ Card {i}: {q[:50]}..."):
+                                    st.write(f"**Question:** {q}")
+                                    st.write(f"**Answer:** {st.session_state.flashcards[q]}")
+                    
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        if st.button("🔄 Practice Again", use_container_width=True):
+                            st.session_state.current_card = 0
+                            st.session_state.score = 0
+                            st.session_state.show_answer = False
+                            st.session_state.user_answers = {}
+                            if st.session_state.shuffle_cards:
+                                random.shuffle(st.session_state.flashcard_keys)
+                            st.rerun()
+                    with col2:
+                        if st.button("📚 New Set", use_container_width=True):
+                            st.session_state.flashcards = {}
+                            st.session_state.flashcard_keys = []
+                            st.session_state.current_card = 0
+                            st.session_state.score = 0
+                            st.session_state.show_answer = False
+                            st.session_state.user_answers = {}
+                            st.rerun()
+                else:
+                    # Progress bar
+                    progress = (st.session_state.current_card / total_cards) * 100
+                    st.markdown(f"""
+                    <div class="progress-bar-container">
+                        <div class="progress-bar" style="width: {progress}%"></div>
+                    </div>
+                    """, unsafe_allow_html=True)
+                    
+                    # Display current card
+                    current_key = st.session_state.flashcard_keys[st.session_state.current_card]
+                    current_answer = st.session_state.flashcards[current_key]
+                    
+                    st.markdown(f"""
+                    <div class="flashcard-container">
+                        <div>
+                            <div class="card-number">Card {st.session_state.current_card + 1} of {total_cards}</div>
+                            <div class="flashcard-question">{current_key}</div>
+                        </div>
+                    </div>
+                    """, unsafe_allow_html=True)
+                    
+                    # Show/Hide answer button
+                    if not st.session_state.show_answer:
+                        if st.button("🔍 Show Answer", use_container_width=True, key="show_btn"):
+                            st.session_state.show_answer = True
+                            st.rerun()
+                    else:
+                        st.markdown(f"""
+                        <div class="flashcard-answer">
+                            <strong>Answer:</strong><br><br>
+                            {current_answer}
+                        </div>
+                        """, unsafe_allow_html=True)
+                        
+                        st.markdown("### Did you get it right?")
+                        col1, col2, col3 = st.columns([1, 1, 1])
+                        
+                        with col1:
+                            if st.button("✅ Got it!", use_container_width=True, type="primary"):
+                                st.session_state.score += 1
+                                st.session_state.user_answers[current_key] = True
+                                st.session_state.current_card += 1
+                                st.session_state.show_answer = False
+                                st.rerun()
+                        
+                        with col2:
+                            if st.button("❌ Missed it", use_container_width=True):
+                                st.session_state.user_answers[current_key] = False
+                                st.session_state.current_card += 1
+                                st.session_state.show_answer = False
+                                st.rerun()
+                        
+                        with col3:
+                            if st.button("⏭️ Skip", use_container_width=True):
+                                st.session_state.current_card += 1
+                                st.session_state.show_answer = False
+                                st.rerun()
+                    
+                    # Current score display
+                    st.markdown(f"""
+                    <div class="score-card">
+                        Current Score: {st.session_state.score} / {st.session_state.current_card} 
+                        {f"({int(st.session_state.score/st.session_state.current_card*100)}%)" if st.session_state.current_card > 0 else ""}
+                    </div>
+                    """, unsafe_allow_html=True)
+
+
+
+
+
+# # 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
+#     uploaded_file.seek(0, os.SEEK_END)
+#     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:
+#         page_text = page.extract_text()
+#         if page_text:
+#             text += page_text + "\n"
+#     return text
+
+# # ---------------------------
+# # OpenAI Response Functions
+# # ---------------------------
+# def generate_summary_from_text(text):
+#     prompt = (
+#         f"Summarize the following document in a concise manner, highlighting the key points that a student should know:\n\n{text}"
+#     )
+#     messages = [
+#         {"role": "system", "content": "You are an educational assistant."},
+#         {"role": "user", "content": prompt}
+#     ]
+#     completion = client.chat.completions.create(
+#         model="gpt-4o-mini",
+#         messages=messages
+#     )
+#     return completion.choices[0].message.content.strip()
+
+# def chat_with_document(text, conversation_history, user_query):
+#     messages = conversation_history + [
+#         {"role": "user", "content": f"Based on the following document:\n\n{text}\n\nQuestion: {user_query}"}
+#     ]
+#     completion = client.chat.completions.create(
+#         model="gpt-4o-mini",
+#         messages=messages
+#     )
+#     return completion.choices[0].message.content.strip()
+
+# 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}"
+#     )
+#     messages = [
+#         {"role": "system", "content": "You are an educational assistant that generates study questions."},
+#         {"role": "user", "content": prompt}
+#     ]
+#     completion = client.chat.completions.create(
+#         model="gpt-4o-mini",
+#         messages=messages
+#     )
+#     # Expecting CSV output (with header: Question,Answer)
+#     return completion.choices[0].message.content.strip()
+
+# # ---------------------------
+# # 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"))
+
+# # For Test Your Knowledge: number of questions (max 50)
+# num_questions = 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":
+#     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
+# # ---------------------------
+# if "pdf_text" not in st.session_state:
+#     st.session_state.pdf_text = None
+# if "summary" not in st.session_state:
+#     st.session_state.summary = None
+# if "chat_history" not in st.session_state:
+#     st.session_state.chat_history = [{"role": "assistant", "content": "Hi, how can I help you with your study material?"}]
+# if "questions_table" not in st.session_state:
+#     st.session_state.questions_table = None
+# if "flashcards" not in st.session_state:
+#     st.session_state.flashcards = {}
+# if "current_card" not in st.session_state:
+#     st.session_state.current_card = 0
+# if "score" not in st.session_state:
+#     st.session_state.score = 0
+# if "show_answer" not in st.session_state:
+#     st.session_state.show_answer = False
+
+# # ---------------------------
+# # 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.")
+
+# # ---------------------------
+# # Main Area: Mode-Based Display (using side menu)
+# # ---------------------------
+# st.title("Study Companion: PDF-based Learning")
+
+# if st.session_state.pdf_text is None:
+#     st.info("Please upload a PDF from the sidebar to begin.")
+# else:
+#     if mode == "Chat":
+#         st.header("Chat with Your Study Companion")
+#         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:")
+#         if user_question:
+#             st.session_state.chat_history.append({"role": "user", "content": user_question})
+#             st.chat_message("user").write(user_question)
+#             with st.spinner("Processing your question..."):
+#                 response = chat_with_document(st.session_state.pdf_text, st.session_state.chat_history, user_question)
+#             st.session_state.chat_history.append({"role": "assistant", "content": response})
+#             st.chat_message("assistant").write(response)
+
+#     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.")
+#         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}")
+
+#     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.")
+#         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.")
+#             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}")
+
+
+
+
+######################################################################################################
+
+
+# # ---------------------------
+# # Helper Function: Extract text from PDF
+# # ---------------------------
+# def extract_text(uploaded_file):
+#     pdf_reader = PdfReader(uploaded_file)
+#     text = ""
+#     for page in pdf_reader.pages:
+#         page_text = page.extract_text()
+#         if page_text:
+#             text += page_text
+#     return text
+
+# # ---------------------------
+# # OpenAI Response Functions (using new style)
+# # ---------------------------
+# def generate_summary_from_text(text):
+#     prompt = (
+#         f"Summarize the following document in a concise manner, "
+#         "highlighting the key points that a student should know:\n\n{text}"
+#     )
+#     messages = [
+#         {"role": "system", "content": "You are an educational assistant."},
+#         {"role": "user", "content": prompt}
+#     ]
+#     completion = client.chat.completions.create(
+#         model="gpt-4o-mini",
+#         messages=messages
+#     )
+#     return completion.choices[0].message.content.strip()
+
+# def chat_with_document(text, conversation_history, user_query):
+#     messages = conversation_history + [
+#         {"role": "user", "content": f"Based on the following document:\n\n{text}\n\nQuestion: {user_query}"}
+#     ]
+#     completion = client.chat.completions.create(
+#         model="gpt-4o-mini",
+#         messages=messages
+#     )
+#     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} "
+#         "Return a Python dictionary where each key is a flashcard question and its corresponding value is the answer. "
+#         #"Do not include any additional text.\n\nDocument:\n\n{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:
+#         # Use literal_eval to safely evaluate the string as a Python dictionary.
+#         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", type="pdf")
+# mode = st.sidebar.radio("Select Mode", ("Summary", "Chat", "Flashcards"))
+
+# num_flashcards = None
+# if mode == "Flashcards":
+#     num_flashcards = st.sidebar.number_input("Number of flashcards to generate:", min_value=1, max_value=20, value=5, step=1)
+
+# # ---------------------------
+# # Session State Initialization
+# # ---------------------------
+# if "pdf_text" not in st.session_state:
+#     st.session_state.pdf_text = None
+# if "summary" not in st.session_state:
+#     st.session_state.summary = None
+# if "chat_history" not in st.session_state:
+#     st.session_state.chat_history = [{"role": "assistant", "content": "Hi, how can I help you with your study material?"}]
+# if "flashcards" not in st.session_state:
+#     st.session_state.flashcards = {}
+# if "current_card" not in st.session_state:
+#     st.session_state.current_card = 0
+# if "score" not in st.session_state:
+#     st.session_state.score = 0
+# if "show_answer" not in st.session_state:
+#     st.session_state.show_answer = False
+
+# # ---------------------------
+# # 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)
+#     st.sidebar.success("PDF uploaded and processed successfully!")
+
+# # ---------------------------
+# # Main Area: Mode-Based Display
+# # ---------------------------
+# st.title("Study Companion: PDF-based Learning")
+
+# if st.session_state.pdf_text is None:
+#     st.info("Please upload a PDF from the sidebar to begin.")
+# else:
+#     if mode == "Summary":
+#         st.header("Summary & Key Points")
+#         if st.session_state.summary is None:
+#             with st.spinner("Generating summary..."):
+#                 st.session_state.summary = generate_summary_from_text(st.session_state.pdf_text)
+#         st.write(st.session_state.summary)
+
+#     elif mode == "Chat":
+#         st.header("Chat with Your Study Companion")
+#         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:")
+#         if user_question:
+#             st.session_state.chat_history.append({"role": "user", "content": user_question})
+#             st.chat_message("user").write(user_question)
+#             with st.spinner("Processing your question..."):
+#                 response = chat_with_document(st.session_state.pdf_text, st.session_state.chat_history, user_question)
+#             st.session_state.chat_history.append({"role": "assistant", "content": response})
+#             st.chat_message("assistant").write(response)
+
+#     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.")
+#         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}")
+
+
+
+
+# # ---------------------------
+# # Helper Function: Extract text from PDF
+# # ---------------------------
+# def extract_text(uploaded_file):
+#     pdf_reader = PdfReader(uploaded_file)
+#     text = ""
+#     for page in pdf_reader.pages:
+#         page_text = page.extract_text()
+#         if page_text:
+#             text += page_text
+#     return text
+
+# # ---------------------------
+# # OpenAI Response Functions (using new style)
+# # ---------------------------
+# def generate_summary_from_text(text):
+#     prompt = (
+#         f"Summarize the following document in a concise manner, "
+#         "highlighting the key points that a student should know:\n\n{text}"
+#     )
+#     messages = [
+#         {"role": "system", "content": "You are an educational assistant."},
+#         {"role": "user", "content": prompt}
+#     ]
+#     completion = client.chat.completions.create(
+#         model="gpt-4o-mini",
+#         messages=messages
+#     )
+#     return completion.choices[0].message.content.strip()
+
+# def chat_with_document(text, conversation_history, user_query):
+#     # Build a message list that includes the conversation history plus the new query with context.
+#     messages = conversation_history + [
+#         {"role": "user", "content": f"Based on the following document:\n\n{text}\n\nQuestion: {user_query}"}
+#     ]
+#     completion = client.chat.completions.create(
+#         model="gpt-4o-mini",
+#         messages=messages
+#     )
+#     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. "
+#         "Return a Python dictionary (in valid JSON format) where each key is a flashcard question and its value is the corresponding answer. "
+#         f"Document:\n\n{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 = json.loads(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", type="pdf")
+# mode = st.sidebar.radio("Select Mode", ("Summary", "Chat", "Flashcards"))
+
+# # For Flashcards, allow user to input number of flashcards
+# num_flashcards = None
+# if mode == "Flashcards":
+#     num_flashcards = st.sidebar.number_input("Number of flashcards to generate:", min_value=1, max_value=20, value=5, step=1)
+
+# # ---------------------------
+# # Session State Initialization
+# # ---------------------------
+# if "pdf_text" not in st.session_state:
+#     st.session_state.pdf_text = None
+# if "summary" not in st.session_state:
+#     st.session_state.summary = None
+# if "chat_history" not in st.session_state:
+#     st.session_state.chat_history = [{"role": "assistant", "content": "Hi, how can I help you with your study material?"}]
+# if "flashcards" not in st.session_state:
+#     st.session_state.flashcards = {}
+# if "current_card" not in st.session_state:
+#     st.session_state.current_card = 0
+# if "score" not in st.session_state:
+#     st.session_state.score = 0
+# if "show_answer" not in st.session_state:
+#     st.session_state.show_answer = False
+
+# # ---------------------------
+# # Process PDF Upload
+# # ---------------------------
+# if uploaded_pdf is not None:
+#     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+#         tmp.write(uploaded_pdf.read())
+#         pdf_file_path = tmp.name
+#     # Extract text from the PDF (all pages)
+#     st.session_state.pdf_text = extract_text(pdf_file_path)
+#     st.sidebar.success("PDF uploaded and processed successfully!")
+
+# # ---------------------------
+# # Main Area: Mode-Based Display
+# # ---------------------------
+# st.title("Study Companion: PDF-based Learning")
+
+# if st.session_state.pdf_text is None:
+#     st.info("Please upload a PDF from the sidebar to begin.")
+# else:
+#     if mode == "Summary":
+#         st.header("Summary & Key Points")
+#         if st.session_state.summary is None:
+#             with st.spinner("Generating summary..."):
+#                 st.session_state.summary = generate_summary_from_text(st.session_state.pdf_text)
+#         st.write(st.session_state.summary)
+
+#     elif 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:")
+#         if user_question:
+#             st.session_state.chat_history.append({"role": "user", "content": user_question})
+#             st.chat_message("user").write(user_question)
+#             with st.spinner("Processing your question..."):
+#                 response = chat_with_document(st.session_state.pdf_text, st.session_state.chat_history, user_question)
+#             st.session_state.chat_history.append({"role": "assistant", "content": response})
+#             st.chat_message("assistant").write(response)
+
+#     elif mode == "Flashcards":
+#         st.header("Practice Flashcards")
+#         # Provide a button to generate flashcards on demand.
+#         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:
+#             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
+#                 # Get the current flashcard key-value pair.
+#                 current_keys = list(flashcards.keys())
+#                 current_key = current_keys[st.session_state.current_card]
+#                 current_answer = flashcards[current_key]
+#                 st.write(f"**Question:** {current_key}")
+#                 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}")