ice_breaking_challenge/background_task.py

from flask import session
import time
import random
from ice_breaking_challenge import model, quiz_generated, quiz_results

def background_task(sid, qna):
    """Session에 저장된 응답을 바탕으로 Gemma 모델을 사용한 퀴즈 생성"""
    quiz_generated[sid] = False
    generated_quizzes = generate_quiz(qna)
    quiz_results[sid] = generated_quizzes
    # 퀴즈 생성 완료
    quiz_generated[sid] = True

def generate_quiz(data):
    global model
    generated_quizzes = []

    selected_data = random.sample(data, min(10, len(data)))

    template_input="""
        <instruction>
        Using the text: {question} {answer}, create a new multiple-choice question with 4 answer options.
        """
    for row in selected_data:
        question, answer, name = row
        
        input_text = template_input.format(question=question, answer=answer)
        
        response = model.generate(input_text, max_length=512)
        # parsing response
        # name + parsed response
        
        generated_quizzes.append(response)

    return generated_quizzes