background job

ice_breaking_challenge/background_task.py

@@ -1,2 +1,42 @@
+from flask import session
+import time
+import random
+from ice_breaking_challenge import model
+
 def background_task():
-    "continue"
+    """Session에 저장된 응답을 바탕으로 Gemma 모델을 사용한 퀴즈 생성"""
+    # 세션에 퀴즈 생성 상태 초기화
+    session["quiz_generated"] = False
+
+    data = session.get('questions_and_answers', [])
+    if not data:
+        print("No data found in session.")
+        return
+    
+    generated_quizzes = generate_quiz(data)
+    session['generated_quizzes'] = generated_quizzes
+    # 퀴즈 생성 완료
+    session["quiz_generated"] = 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 = row
+        
+        input_text = template_input.format(question=question, answer=answer)
+        
+        response = model.generate(input_text, max_length=512)
+        
+        generated_quizzes.append(response)
+
+    return generated_quizzes
+
+  

ice_breaking_challenge/qr.py

@@ -14,7 +14,7 @@ from threading import Thread
 
 from .auth import login_required
 from .db import get_db
-from ice_breaking_challenge.google_sheets import get_team_numbers
+from ice_breaking_challenge.google_sheets import get_team_numbers, get_questions_and_answers
 
 from background_task import background_task
 
@@ -34,6 +34,7 @@ def qr() -> None:
                 flash("해당 팀 번호가 존재하지 않습니다.")
                 return render_template("index.html")
             if counter[session.get("team_number")] == session["team_size"]:
+                get_questions_and_answers()
                 Thread(target=background_task).start()  
                 return render_template("introduction.html")
             else:

ice_breaking_challenge/topic.py

@@ -1,10 +1,11 @@
 from flask import Blueprint, render_template, redirect, url_for
+from flask import session
 import random
 
 import time
 
 bp = Blueprint("topic", __name__, url_prefix="topic")
-topics = ["MBTI", ""]
+topics = ["MBTI", "직업"]
 
 @bp.route("/", methods=["GET"])
 def topic() -> None: