Update app.py

app.py

@@ -1,15 +1,14 @@
 import streamlit as st
-import os
 import tempfile
 import json
 import random
+from pathlib import Path
 from PyPDF2 import PdfReader
-import openai
+from openai import OpenAI
 
-# ---------------------------
-# Set up OpenAI API Key
-# ---------------------------
-openai.api_key = os.getenv("OPENAI_API_KEY")  # Ensure this is set in your environment/Space secrets
+# Initialize the OpenAI client
+api_key = os.getenv("OPENAI_API_KEY")
+client = OpenAI(api_key = api_key)
 
 # ---------------------------
 # Helper Function: Extract text from PDF
@@ -18,28 +17,39 @@ def extract_text(uploaded_file):
     pdf_reader = PdfReader(uploaded_file)
     text = ""
     for page in pdf_reader.pages:
-        text += page.extract_text()
+        page_text = page.extract_text()
+        if page_text:
+            text += page_text
     return text
 
 # ---------------------------
-# OpenAI Helper Functions
+# 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}"
+    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}
     ]
-    response = openai.ChatCompletion.create(model="gpt-4o-mini", messages=messages)
-    return response.choices[0].message.content.strip()
+    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):
-    # Append the new query to the conversation history.
+    # Build a message list that includes the conversation history plus the new query with context.
     messages = conversation_history + [
-        {"role": "user", "content": f"Document content:\n\n{text}\n\nQuestion: {user_query}"}
+        {"role": "user", "content": f"Based on the following document:\n\n{text}\n\nQuestion: {user_query}"}
     ]
-    response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
-    return response.choices[0].message.content.strip()
+    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 = (
@@ -51,10 +61,12 @@ def generate_flashcards_from_text(text, num_cards):
         {"role": "system", "content": "You are an educational assistant that creates study flashcards."},
         {"role": "user", "content": prompt}
     ]
-    response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
-    output = response.choices[0].message.content.strip()
+    completion = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=messages
+    )
+    output = completion.choices[0].message.content.strip()
     try:
-        # Try to parse the output as JSON and convert it into a Python dict.
         flashcards = json.loads(output)
         if isinstance(flashcards, dict):
             return flashcards
@@ -65,14 +77,14 @@ def generate_flashcards_from_text(text, num_cards):
         return {}
 
 # ---------------------------
-# Sidebar: Upload and Mode Selection
+# 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 mode, ask how many flashcards to generate.
+# 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)
@@ -102,7 +114,7 @@ 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 the full text from the PDF and store it in session state.
+    # 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!")
 
@@ -116,7 +128,6 @@ if st.session_state.pdf_text is None:
 else:
     if mode == "Summary":
         st.header("Summary & Key Points")
-        # Generate summary only once
         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)
@@ -124,12 +135,11 @@ else:
 
     elif mode == "Chat":
         st.header("Chat with Your Study Companion")
-        # Display chat history
+        # 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:
-            # Preserve current history and then update
             st.session_state.chat_history.append({"role": "user", "content": user_question})
             st.chat_message("user").write(user_question)
             with st.spinner("Processing your question..."):
@@ -152,14 +162,12 @@ else:
         if not st.session_state.flashcards:
             st.info("No flashcards available. Click the button above to generate flashcards.")
         else:
-            # Loop through flashcards stored as a dictionary.
-            flashcards = st.session_state.flashcards
-            total_cards = len(flashcards)
-            # Check if all flashcards have been practiced.
+            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]