Upload app.py

app.py

@@ -5,16 +5,15 @@ import openai
 import csv
 from sentence_transformers import SentenceTransformer, util
 
-# OpenAIキー
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-# ZIPを展開
 if not os.path.exists("articles"):
     with zipfile.ZipFile("articles.zip", "r") as zip_ref:
         zip_ref.extractall("articles")
 
-# CSVから記事とURLを読み込み
-documents, file_map = [], {}
+documents = []
+file_map = {}
+
 with open("articles/articles.csv", "r", encoding="cp932") as f:
     reader = csv.DictReader(f)
     for row in reader:
@@ -22,11 +21,9 @@ with open("articles/articles.csv", "r", encoding="cp932") as f:
         documents.append(content)
         file_map[content] = {"title": row["title"], "url": row["url"]}
 
-# 埋め込みモデル
 model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
 document_embeddings = model.encode(documents, convert_to_tensor=True)
 
-# 回答生成
 def generate_answer(query):
     query_embedding = model.encode(query, convert_to_tensor=True)
     hits = util.semantic_search(query_embedding, document_embeddings, top_k=2)[0]
@@ -39,8 +36,7 @@ def generate_answer(query):
 
     context = "\n\n".join(documents[hit["corpus_id"]][:300] for hit in hits)
     prompt = (
-        f"以下の内容をふまえて、転職面接に関する質問に自然で丁寧な日本語で答えてください。"
-        f"できるだけ共感から始めて、最後は応援の言葉で締めくくってください。\n"
+        f"以下の内容をふまえて、転職面接に関する質問に自然で丁寧な日本語で答えてください。\n"
         f"質問: {query}\n"
         f"---\n{context}\n---"
     )
@@ -52,12 +48,11 @@ def generate_answer(query):
             temperature=0.6,
         )
         answer = response.choices[0].message.content.strip()
-        html_links = "".join([f'<a href="{url}" target="_blank">{url}</a><br>' for url in sources])
-        return f"{answer}<br><br>もっと詳しく知りたい方はこちら：<br>{html_links}<br>応援しています！あなたらしく、面接に臨んでくださいね 😊"
+        reference = "\n".join(sources)
+        return f"{answer}\n\nもっと詳しく知りたい方は、以下もご覧ください：\n{reference}"
     except Exception as e:
         return f"エラーが発生しました: {str(e)}"
 
-# Gradio UI（Blocks形式）
 with gr.Blocks(title="面接で不安なことを入力して下さい") as demo:
     gr.Markdown("## 面接で不安なことを入力して下さい")
 
@@ -68,12 +63,12 @@ with gr.Blocks(title="面接で不安なことを入力して下さい") as demo
                 placeholder="例：面接で緊張してしまいます。どうすればよいですか？",
                 lines=2,
             )
-            submit_btn = gr.Button("転職面接ナビに回答してもらう", elem_id="submit-btn")
+            submit_btn = gr.Button("転職面接ナビに回答してもらう", elem_classes="submit-btn")
             clear_btn = gr.Button("リセット")
         with gr.Column():
-            answer = gr.HTML(label="回答")
+            answer = gr.Textbox(label="回答", lines=10)
 
     submit_btn.click(fn=generate_answer, inputs=question, outputs=answer)
     clear_btn.click(fn=lambda: ("", ""), inputs=None, outputs=[question, answer])
 
-demo.launch()
+demo.launch()