Create app.py

app.py

@@ -0,0 +1,88 @@
+import streamlit as st
+import tempfile
+import json
+import random
+from crewai_flashcard import generate_flashcards
+
+# --- Set up the Streamlit app ---
+st.title("Study Companion: Flashcard Generator & Practice")
+
+# Create two tabs: one to generate flashcards from a PDF, and one to practice them.
+tab_gen, tab_practice = st.tabs(["Generate Flashcards", "Practice Flashcards"])
+
+# Initialize session state for flashcards and flashcard practice
+if "flashcards" not in st.session_state:
+    st.session_state.flashcards = []  # List of generated flashcards (JSON objects)
+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
+
+# --- Tab 1: Generate Flashcards from PDF ---
+with tab_gen:
+    st.header("Generate Flashcards from PDF")
+    uploaded_file = st.file_uploader("Upload PDF file", type="pdf")
+    page_range = st.text_input("Enter page range to extract (e.g., '1-5' or '1'):")
+    flashcard_count = st.number_input("Number of flashcards to generate:", min_value=1, max_value=20, value=5, step=1)
+    
+    if uploaded_file is not None and page_range:
+        # Save the uploaded file to a temporary location
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
+            tmp.write(uploaded_file.read())
+            pdf_file_path = tmp.name
+        
+        st.success("PDF uploaded successfully!")
+        
+        if st.button("Generate Flashcards"):
+            with st.spinner("Generating flashcards..."):
+                # This function runs the CrewAI system and writes flashcards to JSON output
+                flashcards_json = generate_flashcards(pdf_file_path, page_range, flashcard_count)
+            try:
+                # Parse the JSON output and save it in session state
+                st.session_state.flashcards = json.loads(flashcards_json)
+                st.success("Flashcards generated successfully!")
+                # Reset practice state
+                st.session_state.current_card = 0
+                st.session_state.score = 0
+                st.session_state.show_answer = False
+            except Exception as e:
+                st.error(f"Error parsing flashcards: {e}")
+
+# --- 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!")
+            
+            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.")