128 lines
4.9 KiB
Python
128 lines
4.9 KiB
Python
"""
|
|
games/sequence_sentence.py - Arrange Sentences Game
|
|
type_id = 2
|
|
"""
|
|
from typing import List
|
|
from pydantic import BaseModel, Field
|
|
from langchain_core.output_parsers import PydanticOutputParser
|
|
|
|
|
|
# ============== SCHEMA ==============
|
|
class SentenceItem(BaseModel):
|
|
sentence: str = Field(description="Full sentence to arrange (EXACT from source)")
|
|
original_quote: str = Field(description="EXACT quote from source text")
|
|
image_description: str = Field(default="", description="Visual description of the content")
|
|
image_keywords: List[str] = Field(default=[], description="Keywords for image search")
|
|
image_is_complex: bool = Field(default=False, description="True if image needs precise quantities, humans, or multiple detailed objects")
|
|
|
|
|
|
class SentenceMetadata(BaseModel):
|
|
"""Metadata đánh giá nội dung"""
|
|
title: str = Field(
|
|
description="Title for this content. Prefer title from source document if available and suitable, otherwise create a short descriptive title."
|
|
)
|
|
description: str = Field(
|
|
description="Short description summarizing the content/topic."
|
|
)
|
|
grade: int = Field(
|
|
description="Estimated grade level 1-5 (1=easy/young, 5=advanced/older). Judge by vocabulary, concepts."
|
|
)
|
|
type: str = Field(default="sequence_sentence", description="Game type")
|
|
difficulty: int = Field(
|
|
description="Difficulty 1-5 for that grade (1=very easy, 5=very hard). Judge by sentence complexity, vocabulary."
|
|
)
|
|
|
|
|
|
class SentenceOutput(BaseModel):
|
|
"""Output wrapper for sentence items"""
|
|
items: List[SentenceItem] = Field(description="List of sentence items generated from source text")
|
|
metadata: SentenceMetadata = Field(description="Metadata about the content")
|
|
|
|
|
|
# Output parser
|
|
output_parser = PydanticOutputParser(pydantic_object=SentenceOutput)
|
|
|
|
|
|
# ============== CONFIG ==============
|
|
GAME_CONFIG = {
|
|
"game_type": "sequence_sentence",
|
|
"display_name": "Arrange Sentences",
|
|
"description": "Arrange sentences in order",
|
|
"type_id": 2,
|
|
|
|
"active": True,
|
|
|
|
"max_items": 10,
|
|
"schema": SentenceItem,
|
|
"output_schema": SentenceOutput,
|
|
"output_parser": output_parser,
|
|
|
|
# Dùng cho analyze + generate (không có format rules)
|
|
"system_prompt": """Extract sentences from source text.
|
|
|
|
RULES:
|
|
1. KEEP THE ORIGINAL LANGUAGE - Do NOT translate
|
|
2. sentence = EXACT copy from source text
|
|
3. original_quote = same as sentence value
|
|
4. image_description = ALWAYS provide a short visual description (NEVER empty)
|
|
5. image_is_complex = FALSE for simple/static objects, TRUE for quantities/humans/complex scenes""",
|
|
|
|
# Dùng cho generate trực tiếp (CÓ format rules)
|
|
"direct_prompt": """Extract sentences from source text.
|
|
|
|
EXPECTED INPUT: List of sentences (separated by semicolon, newline, or similar)
|
|
|
|
STEP 1 - VALIDATE INPUT:
|
|
Analyze if input looks like a list of sentences suitable for "arrange sentences" game.
|
|
- Should contain multiple complete sentences
|
|
- Should NOT be a quiz, single word list, or Q&A format
|
|
|
|
If input is clearly NOT suitable (e.g. it's a quiz, single words only, or wrong format), return:
|
|
{{"items": [], "format_error": "Input không phù hợp cho game sắp xếp câu"}}
|
|
|
|
STEP 2 - EXTRACT (if valid):
|
|
RULES:
|
|
1. KEEP THE ORIGINAL LANGUAGE - Do NOT translate
|
|
2. Extract ALL sentences from source
|
|
3. sentence = EXACT sentence from source (trim whitespace)
|
|
4. original_quote = same as sentence value
|
|
5. image_description = ALWAYS provide a short visual description (NEVER leave empty)
|
|
6. image_is_complex:
|
|
- FALSE: simple objects, static things, no specific quantities (e.g. "a sun", "a tree")
|
|
- TRUE: needs exact quantities, humans/people, or complex details (e.g. "3 birds", "a boy reading")""",
|
|
}
|
|
|
|
|
|
# ============== EXAMPLES ==============
|
|
EXAMPLES = [
|
|
{
|
|
"input": "The Sun is a star; The Moon orbits Earth; Mars is red",
|
|
"output": {
|
|
"items": [
|
|
{
|
|
"sentence": "The Sun is a star",
|
|
"original_quote": "The Sun is a star",
|
|
"image_description": "A bright glowing sun",
|
|
"image_keywords": ["sun", "star"],
|
|
"image_is_complex": False
|
|
},
|
|
{
|
|
"sentence": "The Moon orbits Earth",
|
|
"original_quote": "The Moon orbits Earth",
|
|
"image_description": "Moon circling around Earth",
|
|
"image_keywords": ["moon", "earth", "orbit"],
|
|
"image_is_complex": False
|
|
},
|
|
{
|
|
"sentence": "Mars is red",
|
|
"original_quote": "Mars is red",
|
|
"image_description": "Red planet Mars",
|
|
"image_keywords": ["mars", "red", "planet"],
|
|
"image_is_complex": False
|
|
}
|
|
]
|
|
},
|
|
"why_suitable": "Source has sentences separated by semicolons"
|
|
}
|
|
]
|