Create app.py

app.py

@@ -0,0 +1,45 @@
+# app.py
+import streamlit as st
+import tempfile
+import json
+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 = []
+
+        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.")