Spaces:
Runtime error
Runtime error
Refactor VideoTemplateEditor component: remove AI editing features and related state management, streamline template loading, and update styles for improved layout. Enhance CSS for responsive design and visual consistency across various components, including dropdowns, sync stats, and quiz elements. Remove deprecated premium badge styles and adjust layout for chat components.
66de9c8 | """Flashcards: front and back cards written from a notebook's posts or from notebook chats, flipped in the browser.""" | |
| import uuid | |
| from sqlalchemy.orm import Session | |
| from app.llm import run | |
| from app.llm.provider import fast_lm | |
| from app.llm.signatures import GenerateFlashcards | |
| from app.models import Artifact, Job | |
| from app.pipeline import idea_pool | |
| from app.pipeline.generate import NothingToDo | |
| from app.pipeline.material import Material, load_material | |
| from app.services.jobs import update_job | |
| SAME_LANGUAGE = "the same language as the sources" # no language picked: write as the material is written | |
| CARD_COUNTS = {"fewer": 8, "standard": 16, "more": 24} | |
| def _card(c: dict, refs) -> dict | None: | |
| """One generated card as the deck plays it, or None when it has no front or no back.""" | |
| front, back = (c.get("front") or "").strip(), (c.get("back") or "").strip() | |
| if not front or not back: | |
| return None | |
| return {"front": front[:300], "back": back[:800], "sources": refs(c.get("sources"))} | |
| def make_cards(db: Session, workspace_id: uuid.UUID, material: Material, *, topic: str = "", count: int = 16, | |
| difficulty: str = "medium", language: str = SAME_LANGUAGE, job: Job | None = None) -> list[dict]: | |
| """Cards about the material: a front, a back and verified source lines. Repeats and empty cards are dropped.""" | |
| inputs = dict(title=material.title, topic=topic or "(none)", difficulty=difficulty, language=language.strip() or SAME_LANGUAGE) | |
| # Posts with their ideas stored (the Mind Constellation's): one call over a random sample of those ideas, not the posts' text. | |
| text = idea_pool.sample(db, workspace_id, material.docs, topic, chats=material.text) if material.mode == "ideas" else material.text | |
| out = run.predict(GenerateFlashcards, db=db, workspace_id=workspace_id, job=job, material=text, count=count, | |
| lm=fast_lm(), **inputs) | |
| cards, seen = [], set() | |
| for raw in out.get("cards") or []: | |
| card = _card(raw, material.refs) | |
| if not card or card["front"].lower() in seen: | |
| continue | |
| seen.add(card["front"].lower()) | |
| cards.append(card) | |
| return cards[:count] | |
| def build_flashcards(db: Session, job: Job, artifact: Artifact) -> dict: | |
| params = job.params | |
| count = CARD_COUNTS.get(params.get("count") or "standard", CARD_COUNTS["standard"]) | |
| topic = (params.get("topic") or "").strip() | |
| difficulty = params.get("difficulty") if params.get("difficulty") in ("easy", "medium", "hard") else "medium" | |
| material = load_material(db, artifact.workspace_id, artifact.notebook_id, params, job=job, what="flashcard set", | |
| large_ok=True, ideas_always=True) | |
| update_job(db, job, progress=0.4, message="Writing the cards") | |
| cards = make_cards(db, artifact.workspace_id, material, topic=topic, count=count, difficulty=difficulty, | |
| language=params.get("language") or SAME_LANGUAGE, job=job) | |
| if not cards: | |
| raise NothingToDo("The material did not give enough to make cards from. Try other posts or another topic.") | |
| artifact.content_json = { | |
| "title": f"Flashcards: {topic[:60] or material.title}", "topic": topic, "difficulty": difficulty, | |
| "language": params.get("language") or SAME_LANGUAGE, "source": material.source, "cards": cards, | |
| "card_count": len(cards), | |
| } | |
| artifact.status = "ready" | |
| db.commit() | |
| return {"artifact_id": str(artifact.id)} | |