using global var

ice_breaking_challenge/__init__.py

@@ -6,6 +6,8 @@ from ice_breaking_challenge.models.model_loader import load_model_with_lora
 from ice_breaking_challenge.google_sheets import load_google_sheets
 
 model = None
+quiz_generated = dict
+quiz_results = dict
 
 def create_app(test_config=None):
     """Create and configure an instance of the Flask application."""

ice_breaking_challenge/background_task.py

@@ -1,22 +1,15 @@
 from flask import session
 import time
 import random
-from ice_breaking_challenge import model
+from ice_breaking_challenge import model, quiz_generated, quiz_results
 
-def background_task():
+def background_task(sid, qna):
     """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
+    quiz_generated[sid] = False
+    generated_quizzes = generate_quiz(qna)
+    quiz_results[sid] = generated_quizzes
     # 퀴즈 생성 완료
-    session["quiz_generated"] = True
+    quiz_generated[sid] = True
 
 def generate_quiz(data):
     global model

ice_breaking_challenge/google_sheets.py

@@ -78,4 +78,4 @@ def get_questions_and_answers() -> list[list[str]]:
         for question, answer in zip(questions, answers):
             questions_and_answers.append([question, answer])
 
-    session['questions_and_answers'] = questions_and_answers
+    return questions_and_answers

ice_breaking_challenge/qr.py

@@ -34,8 +34,8 @@ 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()  
+                qna = get_questions_and_answers()
+                Thread(target=background_task, args=(session.sid, qna)).start()  
                 return render_template("introduction.html")
             else:
                 flash("설문을 완료하지 않은 팀원이 있습니다.")

ice_breaking_challenge/quiz.py

@@ -1,11 +1,13 @@
 from flask import Blueprint, render_template, session
 
+from ice_breaking_challenge import quiz_results
+
 bp = Blueprint("quiz", __name__, url_prefix="/quiz")
 
 @bp.route("/", methods=["GET"])
 def quiz():
     """quiz"""
-    quizzes = session.get("generated_quizzes", [])
+    quizzes = quiz_results[session.sid]
     return render_template("quiz.html", quizzes=quizzes, length=len(quizzes))
 
 @bp.route("/finish", methods=["GET"])

ice_breaking_challenge/templates/quiz.html

@@ -7,15 +7,15 @@
     <ul>
       {% for quiz in quizzes %}
         <li>
-          <h2>문제 주인: {{ quiz[0] }}</h2> <!-- 이름 부분 -->
-          <h3>질문: {{ quiz[1] }}</h3> <!-- 질문 부분 -->
+          <h2>문제 주인: {{ quiz[3] }}</h2> <!-- 이름 부분 -->
+          <h3>질문: {{ quiz[0] }}</h3> <!-- 질문 부분 -->
           <ul>
-            {% for option in quiz[2] %} <!-- 4지선다 보기 -->
+            {% for option in quiz[1] %} <!-- 4지선다 보기 -->
               <li>{{ option }}</li>
             {% endfor %}
           </ul>
           <button onclick="showAnswer('{{ loop.index0 }}')">정답 보기</button>
-          <p id="answer-{{ loop.index0 }}" style="display: none;">정답: {{ quiz[3] }}</p> <!-- 정답 -->
+          <p id="answer-{{ loop.index0 }}" style="display: none;">정답: {{ quiz[2] }}</p> <!-- 정답 -->
         </li>
       {% endfor %}
     </ul>

ice_breaking_challenge/topic.py

@@ -1,6 +1,7 @@
 from flask import Blueprint, render_template, redirect, url_for
 from flask import session
 import random
+from ice_breaking_challenge import quiz_generated
 
 import time
 
@@ -11,7 +12,7 @@ topics = ["반말 모드를 해보시는 건 어떠세요?", "오늘 낮에 한
 
 @bp.route("/", methods=["GET"])
 def topic() -> None:
-    if session.get("quiz_generated"):
+    if quiz_generated[session.sid]:
         return redirect(url_for("quiz.quiz"))
     else:
         selected_topic = random.choice(topics)