Connect to Google Sheets

credentials.json

@@ -0,0 +1 @@
+{"installed":{"client_id":"41519890488-4qdl7mkrl3i34oj1pr6hfn124v4mbng8.apps.googleusercontent.com","project_id":"ice-breaking-challenge","auth_uri":"https://accounts.google.com/o/oauth2/auth","token_uri":"https://oauth2.googleapis.com/token","auth_provider_x509_cert_url":"https://www.googleapis.com/oauth2/v1/certs","client_secret":"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","redirect_uris":["http://localhost"]}}

ice_breaking_challenge/__init__.py

@@ -3,12 +3,16 @@ import os
 from flask import Flask, session
 from flask_session import Session
 from ice_breaking_challenge.models.model_loader import load_model_with_lora
+from ice_breaking_challenge.google_sheets import load_google_sheets
 
 model = None
 
 def create_app(test_config=None):
     """Create and configure an instance of the Flask application."""
     global model
+
+    load_google_sheets()
+
     app = Flask(__name__, instance_relative_config=True)
     app.config.from_mapping(
         # a default secret that should be overridden by instance config

ice_breaking_challenge/google_sheets.py

@@ -0,0 +1,62 @@
+import os.path
+
+from google.auth.transport.requests import Request
+from google.oauth2.credentials import Credentials
+from google_auth_oauthlib.flow import InstalledAppFlow
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
+
+# If modifying these scopes, delete the file token.json.
+SCOPES = ["https://www.googleapis.com/auth/spreadsheets"]
+
+# The ID and range of a sample spreadsheet.
+# SAMPLE_SPREADSHEET_ID = "1Z3doUDpoUmXs7p4-g7f8q3XZIBhH_Lz2w51bnpUFigw"
+SAMPLE_SPREADSHEET_ID = "1NJGhpMsfUySHJcMBNpLTEnt-r5Co6qwRahenIa9D55k"
+SAMPLE_RANGE_NAME = "Responses!I2:I30"
+
+SERVICE = None
+
+
+def load_google_sheets():
+  global SERVICE
+  """Shows basic usage of the Sheets API.
+  Prints values from a sample spreadsheet.
+  """
+  creds = None
+  # The file token.json stores the user's access and refresh tokens, and is
+  # created automatically when the authorization flow completes for the first
+  # time.
+  if os.path.exists("token.json"):
+    creds = Credentials.from_authorized_user_file("token.json", SCOPES)
+  # If there are no (valid) credentials available, let the user log in.
+  if not creds or not creds.valid:
+    if creds and creds.expired and creds.refresh_token:
+      creds.refresh(Request())
+    else:
+      flow = InstalledAppFlow.from_client_secrets_file(
+        "credentials.json", SCOPES
+      )
+      creds = flow.run_local_server(port=0)
+    # Save the credentials for the next run
+    with open("token.json", "w") as token:
+      token.write(creds.to_json())
+
+  SERVICE = build("sheets", "v4", credentials=creds)
+
+def get_team_numbers() -> list[str]:
+    sheet = SERVICE.spreadsheets()
+    result = (
+        sheet.values()
+        .get(
+            spreadsheetId=SAMPLE_SPREADSHEET_ID,
+            range=SAMPLE_RANGE_NAME
+        )
+        .execute()
+    )
+    values = result.get("values", [])
+
+    if not values:
+        print("No data found.")
+        return
+
+    return [value[0] for value in values]

ice_breaking_challenge/models/model_loader.py

@@ -18,11 +18,16 @@ def load_model_with_lora(model_name:str = MODEL_NAME, lora_weight_path: str = LO
 
 
     model.backbone.load_lora_weights(lora_weight_path)
-    question_crawling="나의 이런 점은 일할 때 도움이 돼!?"
-    answer_crawling="나누고 싶어하는 마음? 주변 사람들은 그만 퍼주라고 하기도 하지만, 내가 클라이언트로부터 돈을 벌고자 하는 것이 아니라 조금이라도 더 챙겨주고자 하는 마음을 가졌을 때 결국 나의 브랜드가 훨씬 더 커질 수 있다는 믿음이 있다."
+    # question_crawling="나의 이런 점은 일할 때 도움이 돼!?"
+    question_crawling="즐겨보는 유튜버?"
+    # answer_crawling="나누고 싶어하는 마음? 주변 사람들은 그만 퍼주라고 하기도 하지만, 내가 클라이언트로부터 돈을 벌고자 하는 것이 아니라 조금이라도 더 챙겨주고자 하는 마음을 가졌을 때 결국 나의 브랜드가 훨씬 더 커질 수 있다는 믿음이 있다."
+    answer_crawling="선바, 드립치는 게 제 취향입니다."
+
+    input_text = f"""
+    <instruction>
+    Using the text: {question_crawling} {answer_crawling}, create a new multiple-choice question with 4 answer options.
+    """
 
-    input_text = f"{question_crawling} {answer_crawling}"
-    
     print(model.generate(input_text, max_length=512))
 
     return model

ice_breaking_challenge/qr.py

@@ -1,3 +1,5 @@
+from collections import Counter
+
 from flask import Blueprint
 from flask import flash
 from flask import g
@@ -10,6 +12,7 @@ from flask import session
 
 from .auth import login_required
 from .db import get_db
+from ice_breaking_challenge.google_sheets import get_team_numbers
 
 
 bp = Blueprint("qr", __name__, url_prefix="/qr")
@@ -21,4 +24,12 @@ def qr() -> None:
         case "GET":
             return render_template("qr.html")
         case "POST":   # qr.html에서 `다음` 버튼 눌렀을 때
-            return render_template("introduction.html")
+            counter = Counter(get_team_numbers())
+            if session.get("team_number") not in counter:
+                flash("해당 팀 번호가 존재하지 않습니다.")
+                return render_template("index.html")
+            if counter[session.get("team_number")] == session["team_size"]:
+                return render_template("introduction.html")
+            else:
+                flash("설문을 완료하지 않은 팀원이 있습니다.")
+                return render_template("qr.html")

requirements.txt

@@ -3,4 +3,11 @@ Flask-Session
 tensorflow
 keras
 keras-nlp
-tensorflow-text
+tensorflow-text
+google
+google-api-core
+google-api-python-client
+google-auth
+google-auth-httplib2
+google-auth-oauthlib
+google-common-protos

token.json

@@ -0,0 +1 @@
+{"token": "ya29.a0AcM612wxloLuLRCUhc53Pra8ugAHcUjNY6Qv1NEUZg0MZkDI3JU2JrI1_5b-tV0YeyoA5J8JgdFfTBjRhC9U_o36s1arZwIFJlGgc2yr0OVrfpen1NMU-jWIiuTAqA9FrfOIsMruNraxXR0y1dW8RNHdP4gwjcf5ukBgqLdVqAaCgYKAZASARMSFQHGX2MiE34UvuAvpMm3dk8CdQJqWg0177", "refresh_token": "1//0eNvRcc0OMNKPCgYIARAAGA4SNwF-L9Irom_jI7Oy6m71V1i05HPrPON8nR1N90q-fFcY0U-TxLK33r9K1qNSv4Jr1__o3hXxNEo", "token_uri": "https://oauth2.googleapis.com/token", "client_id": "41519890488-4qdl7mkrl3i34oj1pr6hfn124v4mbng8.apps.googleusercontent.com", "client_secret": "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", "scopes": ["https://www.googleapis.com/auth/spreadsheets"], "universe_domain": "googleapis.com", "account": "", "expiry": "2024-09-28T18:31:01.030411Z"}