Spaces:
Sleeping
Sleeping
File size: 2,474 Bytes
77e56ff 2832da8 77e56ff 2832da8 77e56ff 2832da8 77e56ff 2832da8 77e56ff 2832da8 |
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 |
from fastapi import HTTPException
from fastapi.responses import JSONResponse
from typing import Callable, Dict, Any
from backend import config
from backend.content_generator import content_generator
from backend.utils import generate_completions
async def handle_generation_request(
data: Any,
mode: str,
instructions_template: str
) -> JSONResponse:
"""
Shared handler for all generation endpoints (curriculum, flashcards, exercises, simulation).
Args:
data: The GenerationRequest object containing query and metadata
mode: The type of generation (curriculum, flashcards, exercises, simulation)
instructions_template: The template string from config to use
Returns:
JSONResponse with the generated content
Raises:
HTTPException: If required metadata is missing or other errors occur
"""
# Validate required metadata
if not (data.native_language and data.target_language and data.proficiency):
raise HTTPException(
status_code=400,
detail="native_language, target_language, and proficiency are required. Please extract metadata first."
)
# Format instructions with metadata
instructions = (
instructions_template
.replace("{native_language}", data.native_language)
.replace("{target_language}", data.target_language)
.replace("{proficiency}", data.proficiency)
)
# Generate new content
response = await generate_completions.get_completions(
data.query,
instructions
)
# Save generated content to database
content_id = await content_generator.save_content(
query=str(data.query),
content=response,
content_type=mode,
user_id=data.user_id,
native_language=data.native_language,
target_language=data.target_language,
proficiency=data.proficiency
)
return JSONResponse(
content={
"data": response,
"type": mode,
"status": "success",
"content_id": content_id,
"saved": True
},
status_code=200
)
# Mapping of modes to their instruction templates
INSTRUCTION_TEMPLATES: Dict[str, str] = {
"curriculum": config.curriculum_instructions,
"flashcards": config.flashcard_mode_instructions,
"exercises": config.exercise_mode_instructions,
"simulation": config.simulation_mode_instructions
} |