Update app.py

app.py

@@ -10,11 +10,9 @@ ZIP_NAME = "solo_leveling_faiss_ko.zip"
 TARGET_DIR = "solo_leveling_faiss_ko"
 
 def ensure_faiss_dir() -> str:
-    """FAISS index가 어디 있든 로드 가능한 위치를 보장합니다."""
     if os.path.exists(os.path.join(TARGET_DIR, "index.faiss")) and \
        os.path.exists(os.path.join(TARGET_DIR, "index.pkl")):
         return TARGET_DIR
-
     if os.path.exists("index.faiss") and os.path.exists("index.pkl"):
         os.makedirs(TARGET_DIR, exist_ok=True)
         if not os.path.exists(os.path.join(TARGET_DIR, "index.faiss")):
@@ -22,7 +20,6 @@ def ensure_faiss_dir() -> str:
         if not os.path.exists(os.path.join(TARGET_DIR, "index.pkl")):
             shutil.move("index.pkl", os.path.join(TARGET_DIR, "index.pkl"))
         return TARGET_DIR
-
     if os.path.exists(ZIP_NAME):
         with zipfile.ZipFile(ZIP_NAME, 'r') as z:
             z.extractall(".")
@@ -36,7 +33,6 @@ def ensure_faiss_dir() -> str:
             shutil.copy2(faiss_cand[0], os.path.join(TARGET_DIR, "index.faiss"))
             shutil.copy2(pkl_cand[0],   os.path.join(TARGET_DIR, "index.pkl"))
             return TARGET_DIR
-
     raise FileNotFoundError("FAISS index files not found (index.faiss / index.pkl).")
 
 # 0) FAISS 인덱스 위치 확보
@@ -46,17 +42,13 @@ base_dir = ensure_faiss_dir()
 embeddings = HuggingFaceEmbeddings(model_name="jhgan/ko-sroberta-multitask")
 vectorstore = FAISS.load_local(base_dir, embeddings, allow_dangerous_deserialization=True)
 
-# 2) 모델 로딩 (CPU 환경 안전 옵션)
+# 2) 모델 로딩 (CPU)
 model_name = "kakaocorp/kanana-nano-2.1b-instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(
-    model_name,
-    torch_dtype=torch.float32,   
-    device_map=None              
-)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float32, device_map=None)
 
 # 3) 텍스트 생성 파이프라인
-pipe = pipeline(
+lm = pipeline(
     "text-generation",
     model=model,
     tokenizer=tokenizer,
@@ -66,7 +58,6 @@ pipe = pipeline(
     top_p=0.9,
     return_full_text=False
 )
-lm = pipe
 
 # 선택지
 choices = [
@@ -76,19 +67,15 @@ choices = [
     "4: 시스템을 거부하고 그냥 도망친다."
 ]
 
-# RAG + 대사 생성 함수
 def rag_answer(message, history):
     try:
         user_idx = int(message.strip()) - 1
         user_choice = choices[user_idx]
     except:
         return "❗올바른 번호를 입력해주세요. (예: 1 ~ 4)"
-        
-    # FAISS 검색
     docs = vectorstore.similarity_search(user_choice, k=3)
     context = "\n".join([doc.page_content for doc in docs])
-    prompt = f"""
-당신은 웹툰 '나 혼자만 레벨업'의 성진우입니다.
+    prompt = f"""당신은 웹툰 '나 혼자만 레벨업'의 성진우입니다.
 현재 상황:
 {context}
 사용자 선택: {user_choice}
@@ -97,51 +84,46 @@ def rag_answer(message, history):
 """
     response = lm(prompt)[0]["generated_text"]
     only_dialogue = response.strip().split("\n")[-1]
-
-    # "대사:" 중복 방지
     if not only_dialogue.startswith("대사:"):
         only_dialogue = "대사: " + only_dialogue
-
     return only_dialogue
 
-# CSS - 제목 오른쪽에 아이콘
+# ===== UI (변경 지점) =====
 css_code = """
 .quest-title {
-    display: flex;
-    align-items: center;
-    gap: 10px;
+  display:flex; align-items:center; gap:10px;
+  font-weight:700; font-size:22px; margin-bottom:6px;
 }
 .quest-title img {
-    width: 60px;
-    height: auto;
+  width:72px; height:auto; opacity:.95;
 }
+.quest-desc { line-height:1.5; margin-bottom:14px; }
 """
 
-# Gradio UI
-demo = gr.ChatInterface(
-    fn=rag_answer,
-    title="""
-        <div class="quest-title">
-            [긴급 퀘스트: 적을 처치하라!]
-            <img src="https://huggingface.co/spaces/min24ss/r-story-selection/resolve/main/system.png">
-        </div>
-    """,
-    description=(
-        "'플레이어'에게 살의를 가진 이들이 주위에 있습니다. 이들을 모두 처치하여 안전을 확보하십시오.<br>"
-        "지시에 따르지 않으면 당신의 심장은 정지(!)하게 됩니다.<br>"
-        "처치해야 할 적의 숫자: 8명 / 처치한 적의 숫자: 0명<br><br>"
-        "💬 선택지를 입력하세요:<br>"
-        "1: 황동석 무리를 모두 처치한다.<br>"
-        "2: 황동석 무리와 진호를 포함하여 모두 처치한다.<br>"
-        "3: 전부 기절 시키고 살려둔다.<br>"
-        "4: 시스템을 거부하고 그냥 도망친다."
-    ),
-    css=css_code
-)
+header_html = """
+<div class="quest-title">
+  [긴급 퀘스트: 적을 처치하라!]
+  <img src="https://huggingface.co/spaces/min24ss/r-story-selection/resolve/main/system.png" alt="quest">
+</div>
+<div class="quest-desc">
+  '플레이어'에게 살의를 가진 이들이 주위에 있습니다. 이들을 모두 처치하여 안전을 확보하십시오.<br>
+  지시에 따르지 않으면 당신의 심장은 정지(!)하게 됩니다.<br>
+  처치해야 할 ���의 숫자: 8명 / 처치한 적의 숫자: 0명<br><br>
+  💬 선택지를 입력하세요:<br>
+  1: 황동석 무리를 모두 처치한다.<br>
+  2: 황동석 무리와 진호를 포함하여 모두 처치한다.<br>
+  3: 전부 기절 시키고 살려둔다.<br>
+  4: 시스템을 거부하고 그냥 도망친다.
+</div>
+"""
+
+with gr.Blocks(css=css_code) as demo:
+    gr.HTML(header_html)  # ← 여기서 HTML 그대로 렌더링 (이미지 보장)
+    gr.ChatInterface(fn=rag_answer)  # title/description은 쓰지 않음
 
-# 실행
 if __name__ == "__main__":
     print("Torch:", torch.__version__)
-    print("Transformers:", __import__('transformers').__version__)
+    import transformers as _t
+    print("Transformers:", _t.__version__)
     print("LangChain:", langchain.__version__)
     demo.launch()