Flash cards artifact added.

backend/api/artifacts.py

@@ -3,8 +3,9 @@ from fastapi.responses import FileResponse
 
 from backend.models.schemas import ArtifactGenerateOut, ArtifactGenerateRequest, ArtifactListOut
 from backend.modules.artifacts import (
-    generate_podcast,
-    generate_quiz,
+    generate_flashcards,
+    generate_podcast,
+    generate_quiz,
     generate_report,
     list_artifacts,
     resolve_artifact_path,
@@ -72,8 +73,31 @@ def create_quiz_artifact(
         raise HTTPException(status_code=404, detail="Notebook not found")
     except ValueError as exc:
         raise HTTPException(status_code=400, detail=str(exc)) from exc
-    except Exception as exc:
-        raise HTTPException(status_code=500, detail=f"Quiz generation failed: {exc}") from exc
+    except Exception as exc:
+        raise HTTPException(status_code=500, detail=f"Quiz generation failed: {exc}") from exc
+
+
+@router.post("/{notebook_id}/artifacts/flashcards", response_model=ArtifactGenerateOut)
+def create_flashcards_artifact(
+    notebook_id: str,
+    payload: ArtifactGenerateRequest,
+    current_user: User = Depends(get_current_user),
+) -> ArtifactGenerateOut:
+    try:
+        enforce_user_match(current_user, payload.user_id)
+        return generate_flashcards(
+            store,
+            user_id=current_user.user_id,
+            notebook_id=notebook_id,
+            prompt=payload.prompt,
+            num_questions=payload.num_questions,
+        )
+    except FileNotFoundError:
+        raise HTTPException(status_code=404, detail="Notebook not found")
+    except ValueError as exc:
+        raise HTTPException(status_code=400, detail=str(exc)) from exc
+    except Exception as exc:
+        raise HTTPException(status_code=500, detail=f"Flashcards generation failed: {exc}") from exc
 
 
 @router.post("/{notebook_id}/artifacts/podcast", response_model=ArtifactGenerateOut)

backend/models/schemas.py

@@ -99,6 +99,7 @@ class PodcastArtifactOut(BaseModel):
 class ArtifactListOut(BaseModel):
     reports: List[ArtifactFileOut]
     quizzes: List[ArtifactFileOut]
+    flashcards: List[ArtifactFileOut]
     podcasts: List[PodcastArtifactOut]
 
 

backend/modules/artifacts.py

@@ -59,11 +59,12 @@ def _now() -> str:
 def _artifact_dirs(store: NotebookStore, user_id: str, notebook_id: str) -> Dict[str, Path]:
     notebook_dir = store.require_notebook_dir(user_id, notebook_id)
     root = notebook_dir / "artifacts"
-    dirs = {
-        "report": root / "reports",
-        "quiz": root / "quizzes",
-        "podcast": root / "podcasts",
-    }
+    dirs = {
+        "report": root / "reports",
+        "quiz": root / "quizzes",
+        "flashcards": root / "flashcards",
+        "podcast": root / "podcasts",
+    }
     for d in dirs.values():
         d.mkdir(parents=True, exist_ok=True)
     return dirs
@@ -167,7 +168,7 @@ def _quiz_fallback(sources: List[Dict[str, str]], num_questions: int) -> str:
     )
 
 
-def _podcast_transcript_fallback(sources: List[Dict[str, str]], extra_prompt: Optional[str]) -> str:
+def _podcast_transcript_fallback(sources: List[Dict[str, str]], extra_prompt: Optional[str]) -> str:
     focus = extra_prompt.strip() if extra_prompt else "core concepts from the notebook"
     lines = [
         "# Podcast Transcript",
@@ -182,7 +183,20 @@ def _podcast_transcript_fallback(sources: List[Dict[str, str]], extra_prompt: Op
         lines.append(f"**Host:** Great, and why does that matter in practice?")
     lines.append("**Co-Host:** That wraps the study summary. Review the report and quiz next.")
     lines.append("")
-    return "\n".join(lines)
+    return "\n".join(lines)
+
+
+def _flashcards_fallback(sources: List[Dict[str, str]], num_cards: int) -> str:
+    cards = max(3, min(20, int(num_cards)))
+    lines = ["# Flashcards", ""]
+    for i in range(1, cards + 1):
+        src = sources[(i - 1) % len(sources)]
+        snippet = src["text"].replace("\n", " ")[:200].strip()
+        lines.append(f"## Card {i}")
+        lines.append(f"Q: What key point appears in {src['source_name']}?")
+        lines.append(f"A: {snippet}")
+        lines.append("")
+    return "\n".join(lines).strip() + "\n"
 
 
 def _encode_pcm_to_mp3(pcm_16le: bytes, sample_rate: int, channels: int) -> bytes:
@@ -286,7 +300,7 @@ def generate_report(
     )
 
 
-def generate_quiz(
+def generate_quiz(
     store: NotebookStore,
     *,
     user_id: str,
@@ -310,13 +324,47 @@ def generate_quiz(
     )
     content = _llm_or_fallback(llm_prompt, _quiz_fallback(sources, questions))
     out_path = _write_markdown_artifact(dirs["quiz"], "quiz", content)
-    return ArtifactGenerateOut(
-        artifact_type="quiz",
+    return ArtifactGenerateOut(
+        artifact_type="quiz",
         message=f"Generated {out_path.name}",
         markdown_path=str(out_path.as_posix()),
         audio_path=None,
-        created_at=_now(),
-    )
+        created_at=_now(),
+    )
+
+
+def generate_flashcards(
+    store: NotebookStore,
+    *,
+    user_id: str,
+    notebook_id: str,
+    prompt: Optional[str] = None,
+    num_questions: int = 8,
+) -> ArtifactGenerateOut:
+    dirs = _artifact_dirs(store, user_id, notebook_id)
+    sources = _collect_source_texts(store, user_id, notebook_id)
+    if not sources:
+        raise ValueError("No ingested sources available. Upload or ingest sources first.")
+
+    cards = max(3, min(20, int(num_questions)))
+    focus = prompt.strip() if prompt else "Create concise study flashcards."
+    llm_prompt = (
+        "Create markdown flashcards from SOURCES.\n"
+        f"Include exactly {cards} cards in format:\n"
+        "## Card N\nQ: ...\nA: ...\n"
+        "Keep answers concise and grounded in source content.\n\n"
+        f"FOCUS:\n{focus}\n\n"
+        f"SOURCES:\n{_sources_block(sources)}\n"
+    )
+    content = _llm_or_fallback(llm_prompt, _flashcards_fallback(sources, cards))
+    out_path = _write_markdown_artifact(dirs["flashcards"], "flashcards", content)
+    return ArtifactGenerateOut(
+        artifact_type="flashcards",
+        message=f"Generated {out_path.name}",
+        markdown_path=str(out_path.as_posix()),
+        audio_path=None,
+        created_at=_now(),
+    )
 
 
 def generate_podcast(
@@ -361,8 +409,9 @@ def generate_podcast(
 def list_artifacts(store: NotebookStore, *, user_id: str, notebook_id: str) -> ArtifactListOut:
     dirs = _artifact_dirs(store, user_id, notebook_id)
 
-    reports = [_artifact_file_out(p) for p in sorted(dirs["report"].glob("report_*.md"))]
-    quizzes = [_artifact_file_out(p) for p in sorted(dirs["quiz"].glob("quiz_*.md"))]
+    reports = [_artifact_file_out(p) for p in sorted(dirs["report"].glob("report_*.md"))]
+    quizzes = [_artifact_file_out(p) for p in sorted(dirs["quiz"].glob("quiz_*.md"))]
+    flashcards = [_artifact_file_out(p) for p in sorted(dirs["flashcards"].glob("flashcards_*.md"))]
 
     podcast_indices: set[int] = set()
     for path in dirs["podcast"].glob("podcast_*.*"):
@@ -381,7 +430,7 @@ def list_artifacts(store: NotebookStore, *, user_id: str, notebook_id: str) -> A
             )
         )
 
-    return ArtifactListOut(reports=reports, quizzes=quizzes, podcasts=podcasts)
+    return ArtifactListOut(reports=reports, quizzes=quizzes, flashcards=flashcards, podcasts=podcasts)
 
 
 def resolve_artifact_path(

frontend/app.py

@@ -417,16 +417,18 @@ def send_message(message: str, history, user_id: str, notebook_id: str):
 
 
 def _empty_artifacts_payload() -> dict[str, Any]:
-    return {"reports": [], "quizzes": [], "podcasts": []}
+    return {"reports": [], "quizzes": [], "flashcards": [], "podcasts": []}
 
 
 def _artifact_outputs_from_payload(payload: dict[str, Any]):
     reports = payload.get("reports") or []
     quizzes = payload.get("quizzes") or []
+    flashcards = payload.get("flashcards") or []
     podcasts = payload.get("podcasts") or []
 
     latest_report = reports[-1].get("path") if reports else None
     latest_quiz = quizzes[-1].get("path") if quizzes else None
+    latest_flashcards = flashcards[-1].get("path") if flashcards else None
 
     latest_transcript = None
     latest_audio = None
@@ -436,14 +438,14 @@ def _artifact_outputs_from_payload(payload: dict[str, Any]):
         audio = last.get("audio") or {}
         latest_transcript = transcript.get("path")
         latest_audio = audio.get("path")
-    return latest_report, latest_quiz, latest_transcript, latest_audio
+    return latest_report, latest_quiz, latest_flashcards, latest_transcript, latest_audio
 
 
 def refresh_artifacts(user_id: str, notebook_id: str):
     user_id = (user_id or "").strip()
     if not user_id or not notebook_id:
         payload = _empty_artifacts_payload()
-        return gr.JSON(value=payload), None, None, None, None, "Select a notebook first."
+        return gr.JSON(value=payload), None, None, None, None, None, "Select a notebook first."
     try:
         payload = _api_request(
             "GET",
@@ -451,23 +453,23 @@ def refresh_artifacts(user_id: str, notebook_id: str):
             params={"user_id": user_id},
             headers=_auth_headers(user_id),
         )
-        report, quiz, transcript, audio = _artifact_outputs_from_payload(payload)
-        return gr.JSON(value=payload), report, quiz, transcript, audio, ""
+        report, quiz, flashcards, transcript, audio = _artifact_outputs_from_payload(payload)
+        return gr.JSON(value=payload), report, quiz, flashcards, transcript, audio, ""
     except Exception as exc:
         payload = _empty_artifacts_payload()
-        return gr.JSON(value=payload), None, None, None, None, str(exc)
+        return gr.JSON(value=payload), None, None, None, None, None, str(exc)
 
 
 def sync_artifacts_on_notebook_change(user_id: str, notebook_id: str):
-    payload_json, report, quiz, transcript, audio, _status = refresh_artifacts(user_id, notebook_id)
-    return payload_json, report, quiz, transcript, audio
+    payload_json, report, quiz, flashcards, transcript, audio, _status = refresh_artifacts(user_id, notebook_id)
+    return payload_json, report, quiz, flashcards, transcript, audio
 
 
 def generate_report_artifact(user_id: str, notebook_id: str, artifact_prompt: str):
     user_id = (user_id or "").strip()
     if not user_id or not notebook_id:
         payload = _empty_artifacts_payload()
-        return gr.JSON(value=payload), None, None, None, None, "Select a notebook first."
+        return gr.JSON(value=payload), None, None, None, None, None, "Select a notebook first."
     try:
         resp = _api_request(
             "POST",
@@ -476,18 +478,18 @@ def generate_report_artifact(user_id: str, notebook_id: str, artifact_prompt: st
             headers=_auth_headers(user_id),
             timeout=180,
         )
-        payload_json, report, quiz, transcript, audio, _ = refresh_artifacts(user_id, notebook_id)
-        return payload_json, report, quiz, transcript, audio, str(resp.get("message", "Generated report."))
+        payload_json, report, quiz, flashcards, transcript, audio, _ = refresh_artifacts(user_id, notebook_id)
+        return payload_json, report, quiz, flashcards, transcript, audio, str(resp.get("message", "Generated report."))
     except Exception as exc:
         payload = _empty_artifacts_payload()
-        return gr.JSON(value=payload), None, None, None, None, str(exc)
+        return gr.JSON(value=payload), None, None, None, None, None, str(exc)
 
 
 def generate_quiz_artifact(user_id: str, notebook_id: str, artifact_prompt: str, num_questions: float):
     user_id = (user_id or "").strip()
     if not user_id or not notebook_id:
         payload = _empty_artifacts_payload()
-        return gr.JSON(value=payload), None, None, None, None, "Select a notebook first."
+        return gr.JSON(value=payload), None, None, None, None, None, "Select a notebook first."
     try:
         resp = _api_request(
             "POST",
@@ -500,18 +502,42 @@ def generate_quiz_artifact(user_id: str, notebook_id: str, artifact_prompt: str,
             headers=_auth_headers(user_id),
             timeout=180,
         )
-        payload_json, report, quiz, transcript, audio, _ = refresh_artifacts(user_id, notebook_id)
-        return payload_json, report, quiz, transcript, audio, str(resp.get("message", "Generated quiz."))
+        payload_json, report, quiz, flashcards, transcript, audio, _ = refresh_artifacts(user_id, notebook_id)
+        return payload_json, report, quiz, flashcards, transcript, audio, str(resp.get("message", "Generated quiz."))
     except Exception as exc:
         payload = _empty_artifacts_payload()
-        return gr.JSON(value=payload), None, None, None, None, str(exc)
+        return gr.JSON(value=payload), None, None, None, None, None, str(exc)
+
+
+def generate_flashcards_artifact(user_id: str, notebook_id: str, artifact_prompt: str, num_questions: float):
+    user_id = (user_id or "").strip()
+    if not user_id or not notebook_id:
+        payload = _empty_artifacts_payload()
+        return gr.JSON(value=payload), None, None, None, None, None, "Select a notebook first."
+    try:
+        resp = _api_request(
+            "POST",
+            f"/api/notebooks/{notebook_id}/artifacts/flashcards",
+            json_body={
+                "user_id": user_id,
+                "prompt": (artifact_prompt or "").strip() or None,
+                "num_questions": int(num_questions),
+            },
+            headers=_auth_headers(user_id),
+            timeout=180,
+        )
+        payload_json, report, quiz, flashcards, transcript, audio, _ = refresh_artifacts(user_id, notebook_id)
+        return payload_json, report, quiz, flashcards, transcript, audio, str(resp.get("message", "Generated flashcards."))
+    except Exception as exc:
+        payload = _empty_artifacts_payload()
+        return gr.JSON(value=payload), None, None, None, None, None, str(exc)
 
 
 def generate_podcast_artifact(user_id: str, notebook_id: str, artifact_prompt: str):
     user_id = (user_id or "").strip()
     if not user_id or not notebook_id:
         payload = _empty_artifacts_payload()
-        return gr.JSON(value=payload), None, None, None, None, "Select a notebook first."
+        return gr.JSON(value=payload), None, None, None, None, None, "Select a notebook first."
     try:
         resp = _api_request(
             "POST",
@@ -520,11 +546,11 @@ def generate_podcast_artifact(user_id: str, notebook_id: str, artifact_prompt: s
             headers=_auth_headers(user_id),
             timeout=240,
         )
-        payload_json, report, quiz, transcript, audio, _ = refresh_artifacts(user_id, notebook_id)
-        return payload_json, report, quiz, transcript, audio, str(resp.get("message", "Generated podcast."))
+        payload_json, report, quiz, flashcards, transcript, audio, _ = refresh_artifacts(user_id, notebook_id)
+        return payload_json, report, quiz, flashcards, transcript, audio, str(resp.get("message", "Generated podcast."))
     except Exception as exc:
         payload = _empty_artifacts_payload()
-        return gr.JSON(value=payload), None, None, None, None, str(exc)
+        return gr.JSON(value=payload), None, None, None, None, None, str(exc)
     
 def greet_user(profile: gr.OAuthProfile | None) -> tuple[str | None, str]:
     if profile is None:
@@ -593,10 +619,12 @@ with gr.Blocks(title="MemoriaLM") as demo:
                         report_btn = gr.Button("Generate Report")
                     with gr.Row():
                         quiz_btn = gr.Button("Generate Quiz")
+                        flashcards_btn = gr.Button("Generate Flashcards")
                         podcast_btn = gr.Button("Generate Podcast")
                     artifacts_json = gr.JSON(label="Generated Artifacts", value=_empty_artifacts_payload())
                     latest_report_file = gr.File(label="Latest Report (.md)", interactive=False)
                     latest_quiz_file = gr.File(label="Latest Quiz (.md)", interactive=False)
+                    latest_flashcards_file = gr.File(label="Latest Flashcards (.md)", interactive=False)
                     latest_podcast_transcript_file = gr.File(label="Latest Podcast Transcript (.md)", interactive=False)
                     latest_podcast_audio = gr.Audio(label="Latest Podcast Audio (.mp3)", type="filepath", interactive=False)
 
@@ -666,6 +694,7 @@ with gr.Blocks(title="MemoriaLM") as demo:
             artifacts_json,
             latest_report_file,
             latest_quiz_file,
+            latest_flashcards_file,
             latest_podcast_transcript_file,
             latest_podcast_audio,
         ],
@@ -678,6 +707,7 @@ with gr.Blocks(title="MemoriaLM") as demo:
             artifacts_json,
             latest_report_file,
             latest_quiz_file,
+            latest_flashcards_file,
             latest_podcast_transcript_file,
             latest_podcast_audio,
         ],
@@ -689,6 +719,7 @@ with gr.Blocks(title="MemoriaLM") as demo:
             artifacts_json,
             latest_report_file,
             latest_quiz_file,
+            latest_flashcards_file,
             latest_podcast_transcript_file,
             latest_podcast_audio,
         ],
@@ -736,6 +767,7 @@ with gr.Blocks(title="MemoriaLM") as demo:
             artifacts_json,
             latest_report_file,
             latest_quiz_file,
+            latest_flashcards_file,
             latest_podcast_transcript_file,
             latest_podcast_audio,
             status_box,
@@ -748,6 +780,7 @@ with gr.Blocks(title="MemoriaLM") as demo:
             artifacts_json,
             latest_report_file,
             latest_quiz_file,
+            latest_flashcards_file,
             latest_podcast_transcript_file,
             latest_podcast_audio,
             status_box,
@@ -760,6 +793,20 @@ with gr.Blocks(title="MemoriaLM") as demo:
             artifacts_json,
             latest_report_file,
             latest_quiz_file,
+            latest_flashcards_file,
+            latest_podcast_transcript_file,
+            latest_podcast_audio,
+            status_box,
+        ],
+    )
+    flashcards_btn.click(
+        generate_flashcards_artifact,
+        inputs=[user_id, notebook_selector, artifact_prompt, quiz_questions],
+        outputs=[
+            artifacts_json,
+            latest_report_file,
+            latest_quiz_file,
+            latest_flashcards_file,
             latest_podcast_transcript_file,
             latest_podcast_audio,
             status_box,
@@ -772,6 +819,7 @@ with gr.Blocks(title="MemoriaLM") as demo:
             artifacts_json,
             latest_report_file,
             latest_quiz_file,
+            latest_flashcards_file,
             latest_podcast_transcript_file,
             latest_podcast_audio,
             status_box,

tests/test_api_artifacts.py

@@ -45,6 +45,12 @@ def test_artifact_endpoints_generate_list_and_download(monkeypatch, tmp_path: Pa
         headers=AUTH_U1,
     )
     assert report.status_code == 200
+    flashcards = client.post(
+        f"/api/notebooks/{nb.notebook_id}/artifacts/flashcards",
+        json={"user_id": "u1", "num_questions": 6},
+        headers=AUTH_U1,
+    )
+    assert flashcards.status_code == 200
 
     podcast = client.post(
         f"/api/notebooks/{nb.notebook_id}/artifacts/podcast",
@@ -62,6 +68,7 @@ def test_artifact_endpoints_generate_list_and_download(monkeypatch, tmp_path: Pa
     assert listed.status_code == 200
     payload = listed.json()
     assert len(payload["reports"]) == 1
+    assert len(payload["flashcards"]) == 1
     assert len(payload["podcasts"]) == 1
 
     dl = client.get(

tests/test_artifacts.py

@@ -34,15 +34,19 @@ def test_generate_report_quiz_and_list(monkeypatch, tmp_path: Path):
 
     report = artifacts.generate_report(store, user_id="u1", notebook_id=nb.notebook_id, prompt="Focus")
     quiz = artifacts.generate_quiz(store, user_id="u1", notebook_id=nb.notebook_id, num_questions=5)
+    flashcards = artifacts.generate_flashcards(store, user_id="u1", notebook_id=nb.notebook_id, num_questions=6)
 
     assert report.artifact_type == "report"
     assert quiz.artifact_type == "quiz"
+    assert flashcards.artifact_type == "flashcards"
     assert Path(report.markdown_path).exists()
     assert Path(quiz.markdown_path).exists()
+    assert Path(flashcards.markdown_path).exists()
 
     listed = artifacts.list_artifacts(store, user_id="u1", notebook_id=nb.notebook_id)
     assert len(listed.reports) == 1
     assert len(listed.quizzes) == 1
+    assert len(listed.flashcards) == 1
     assert listed.podcasts == []