Parse Mode Response (#3)

- Update (7ff0925c9a49c3df9bc66e1a886e21f20618fd76)


Co-authored-by: Dongwook Chang <[email protected]>

ice_breaking_challenge/background_task.py

@@ -27,11 +27,15 @@ def generate_quiz(data):
         input_text = template_input.format(question=question, answer=answer)
         
         response = model.generate(input_text, max_length=512)
-        # parsing response
-        # name + parsed response
+
+        lines = response.split('\n')
+        index = lines.index('<response>')
+        question_generated = lines[index+1]
+        multiple_choice_generated = lines[index + 2]
+        answer_generated = lines[index + 3]
         
-        generated_quizzes.append(response)
+        generated_quizzes.append([question_generated, multiple_choice_generated, answer_generated])
 
     return generated_quizzes
 
-  
+  

ice_breaking_challenge/google_sheets.py

@@ -73,9 +73,11 @@ def get_questions_and_answers() -> list[list[str]]:
 
     answers_values = get_range(f"Responses!B2:F{MAX_ROW}")
 
+    name_values = get_range(f"Responses!K2:K{MAX_ROW}")
+
     questions_and_answers: list[list[str]] = []
-    for answers in answers_values:
-        for question, answer in zip(questions, answers):
-            questions_and_answers.append([question, answer])
+    for answers, names in zip(answers_values, name_values):
+        for question, answer, name in zip(questions, answers, names):
+            questions_and_answers.append([question, answer, name])
 
     return questions_and_answers

ice_breaking_challenge/introduction.py

@@ -12,7 +12,7 @@ from threading import Thread
 from .auth import login_required
 from .db import get_db
 
-bp = Blueprint("introduction", __name__, url_prefix="introduction")
+bp = Blueprint("introduction", __name__, url_prefix="/introduction")
 
 
 @bp.route("/", methods=["GET", "POST"])

ice_breaking_challenge/qr.py

@@ -16,7 +16,7 @@ from .auth import login_required
 from .db import get_db
 from ice_breaking_challenge.google_sheets import get_team_numbers, get_questions_and_answers
 
-from background_task import background_task
+from ice_breaking_challenge import background_task
 
 
 bp = Blueprint("qr", __name__, url_prefix="/qr")
@@ -33,10 +33,12 @@ def qr() -> None:
             if session.get("team_number") not in counter:
                 flash("해당 팀 번호가 존재하지 않습니다.")
                 return render_template("index.html")
-            if counter[session.get("team_number")] == session["team_size"]:
+            if counter[session.get("team_number")] == int(session["team_size"]):
                 qna = get_questions_and_answers()
                 Thread(target=background_task, args=(session.sid, qna)).start()  
+                import time
+                time.sleep(100)
                 return render_template("introduction.html")
             else:
                 flash("설문을 완료하지 않은 팀원이 있습니다.")
-                return render_template("qr.html")
+                return render_template("qr.html")