Spaces:
Sleeping
Sleeping
| # app.py | |
| import streamlit as st | |
| import tempfile | |
| import json | |
| from pathlib import Path | |
| from crewai_flashcard import generate_flashcards | |
| st.title("📚 Flashcard Generator from PDF") | |
| st.write("Upload a PDF, specify a page range (e.g. '1' or '1-5'), and choose the number of flashcards to generate.") | |
| # PDF uploader | |
| 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 file | |
| 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 call will run the CrewAI crew, | |
| # which outputs flashcards.json to the working directory. | |
| generate_flashcards(pdf_file_path, page_range, flashcard_count) | |
| # # Read the flashcards.json file generated by the flashcard task | |
| # try: | |
| # with open("flashcards.json", "r") as f: | |
| # flashcards = json.load(f) | |
| # except Exception as e: | |
| # st.error(f"Error reading flashcards.json: {e}") | |
| # flashcards = [] | |
| # Alternative method: use pathlib to read the flashcards.json file | |
| try: | |
| flashcards_path = Path("flashcards.json") | |
| flashcards_data = flashcards_path.read_text(encoding="utf-8") | |
| flashcards = json.loads(flashcards_data) | |
| except Exception as e: | |
| st.error(f"Error reading flashcards.json: {e}") | |
| flashcards = [] | |
| st.subheader("Generated Flashcards") | |
| if flashcards: | |
| for idx, card in enumerate(flashcards): | |
| st.markdown(f"**Flashcard {idx+1}:**") | |
| st.write(f"**Question:** {card.get('question', '')}") | |
| st.write(f"**Answer:** {card.get('answer', '')}") | |
| else: | |
| st.write("No flashcards generated.") | |