Spaces:
Sleeping
Sleeping
File size: 2,369 Bytes
99b596a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | import json
import re
import os
from huggingface_hub import InferenceClient
HF_TOKEN = os.getenv("HF_TOKEN", "")
MODEL_NAME = os.getenv("HF_MODEL", "Qwen/Qwen2.5-72B-Instruct")
_client = None
def _get_client() -> InferenceClient:
global _client
if _client is None:
_client = InferenceClient(token=HF_TOKEN or None)
return _client
def _call_hf(system: str, user: str, max_tokens: int = 1024, temperature: float = 0.3) -> str:
client = _get_client()
response = client.chat_completion(
model=MODEL_NAME,
messages=[
{"role": "system", "content": system},
{"role": "user", "content": user},
],
max_tokens=max_tokens,
temperature=temperature,
)
return response.choices[0].message.content.strip()
def _extract_json_array(raw: str) -> list:
cleaned = re.sub(r'```(?:json)?\s*|```', '', raw).strip()
try:
result = json.loads(cleaned)
if isinstance(result, list):
return result
except Exception:
pass
start = cleaned.find('[')
if start != -1:
depth = 0
for i, ch in enumerate(cleaned[start:], start):
if ch == '[': depth += 1
elif ch == ']':
depth -= 1
if depth == 0:
candidate = re.sub(r',\s*([}\]])', r'\1', cleaned[start:i+1])
try:
return json.loads(candidate)
except Exception:
pass
break
return []
def generate_flashcards(topic: str, num_cards: int = 10) -> list[dict]:
system = (
"Tu es un générateur de flashcards pédagogiques. "
"Tu réponds UNIQUEMENT avec un tableau JSON valide, sans texte avant ni après."
)
user = (
f"Génère {num_cards} flashcards sur : \"{topic}\".\n"
"Chaque objet : front (question/terme) et back (réponse/définition).\n"
"Réponds UNIQUEMENT avec le tableau JSON."
)
raw = _call_hf(system, user)
cards = _extract_json_array(raw)
if cards:
return [
{"front": str(c.get("front", "")), "back": str(c.get("back", ""))}
for c in cards
if isinstance(c, dict) and c.get("front") and c.get("back")
]
return [{"front": topic, "back": raw[:300]}] |