change file

app.py

@@ -30,23 +30,23 @@ def generate_image(text):
     return image_path
 
 
-def calulate_similarity_score(ori_text, text):
+def cos_sim(v1, v2):
+    return np.dot(v1, v2) / (np.linalg.norm(v1) * np.linalg.norm(v2))
+
+
+def calculate_similarity_score(ori_text, text):
     if ori_text != text:
-        model_name = "text-embedding-3-small"
-        response = client.embeddings.create(input=[ori_text, text], model=model_name)
+        response = client.embeddings.create(
+            input=[ori_text, text], model="text-embedding-3-small"
+        )
         score = cos_sim(response.data[0].embedding, response.data[1].embedding)
         score = int(round(score, 2) * 100)
-        if score == 100:
-            score = 99
+        score = 99 if score == 100 else score
     else:
         score = 100
     return score
 
 
-def cos_sim(v1, v2):
-    return np.dot(v1, v2) / (np.linalg.norm(v1) * np.linalg.norm(v2))
-
-
 def tokenize_text(text):
     mecab = MeCab.Tagger(f"-Ochasen {ipadic.MECAB_ARGS}")
     return [t.split()[0] for t in mecab.parse(text).splitlines()[:-1]]
@@ -66,7 +66,7 @@ def create_hint_text(ori_text, text):
         if r[:2] == "- ":
             continue
         elif r[:2] == "+ ":
-            output += "^"
+            output += "X"
         else:
             output += r.strip()
     return output
@@ -80,7 +80,7 @@ def update_question(option):
 def main(text, option):
     ori_text = os.getenv(option)
     image_path = generate_image(text)
-    score = calulate_similarity_score(ori_text, text)
+    score = calculate_similarity_score(ori_text, text)
 
     if score < 80:
         match_words = create_match_words(ori_text, text)
@@ -92,6 +92,13 @@ def main(text, option):
     return image_path, f"{score}点", hint_text
 
 
+def auth(user_name, password):
+    if user_name == os.getenv("USER_NAME") and password == os.getenv("PASSWORD"):
+        return True
+    else:
+        return False
+
+
 questions = ["Q1", "Q2", "Q3"]
 for q in questions:
     image_path = generate_image(os.getenv(q))
@@ -100,7 +107,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             gr.Markdown(
-                "# プロンプトを当てるゲーム \n これは表示されている画像のプロンプトを当てるゲームです。プロンプトを入力するとそれに対応した画像とスコアとヒントが表示されます。スコア100点を目指して頑張ってください！ \n\nヒントは80点未満の場合は当たっている単語、80点以上の場合は足りない文字を「^」で示した文字列を表示しています。",
+                "# プロンプトを当てるゲーム \n これは表示されている画像のプロンプトを当てるゲームです。プロンプトを入力するとそれに対応した画像とスコアとヒントが表示されます。スコア100点を目指して頑張ってください！ \n\nヒントは80点未満の場合は当たっている単語（順番は合っているとは限らない）、80点以上の場合は足りない文字を「X」で示した文字列を表示しています。",
             )
             option = gr.components.Radio(
                 ["Q1", "Q2", "Q3"], label="問題を選んでください！"
@@ -113,7 +120,8 @@ with gr.Blocks() as demo:
             input_text = gr.components.Textbox(
                 lines=1, label="画像にマッチするテキストを入力して！"
             )
-            submit_button = gr.Button("Submit")
+            submit_button = gr.Button("Submit!")
+
         with gr.Column():
             output_image = gr.components.Image(type="filepath", label="生成画像")
             output_score = gr.components.Textbox(lines=1, label="スコア")