Update app.py

app.py

@@ -26,7 +26,6 @@ df['text'] = df.apply(
     lambda x: f"[{x['에피소드']}] #{x['row_id']} {x['type']} {x['scene_text']}",
     axis=1
 )
-texts = df['text'].tolist()
 
 # ====== FAISS 안전 로드 ======
 embedding_model = HuggingFaceEmbeddings(model_name='jhgan/ko-sroberta-multitask')
@@ -47,7 +46,7 @@ if load_path:
     vectorstore = FAISS.load_local(load_path, embedding_model, allow_dangerous_deserialization=True)
     print(f"[INFO] FAISS 인덱스 로드 완료 → {load_path}")
 else:
-    raise FileNotFoundError("FAISS index.faiss 파일을 찾을 수 없습니다. 압축 구조를 확인하세요.")
+    raise FileNotFoundError("FAISS index.faiss 파일을 찾을 수 없습니다.")
 
 # ====== 모델 로드 (CPU 전용) ======
 model_name = "kakaocorp/kanana-nano-2.1b-instruct"
@@ -70,17 +69,23 @@ qa_chain = RetrievalQA.from_chain_type(
     chain_type_kwargs={"prompt": custom_prompt}
 )
 
-# ====== 선택지 ======
+# ====== 대화형 응답 함수 ======
 choices = [
-    "황동석 무리를 모두 처치한다.",
-    "진호를 포함한 황동석 무리를 모두 처치한다.",
-    "전부 기절 시키고 살려둔다.",
-    "시스템을 거부하고 그냥 도망친다."
+    "1. 황동석 무리를 모두 처치한다.",
+    "2. 진호를 포함한 황동석 무리를 모두 처치한다.",
+    "3. 전부 기절 시키고 살려둔다.",
+    "4. 시스템을 거부하고 그냥 도망친다."
 ]
 
-# ====== Gradio 함수 ======
-def run_episode(selection):
-    user_choice = choices[int(selection) - 1]
+def respond(message, history):
+    try:
+        sel_num = int(message.strip())
+        if sel_num < 1 or sel_num > len(choices):
+            return "❌ 올바른 번호를 입력하세요. (1~4)"
+    except ValueError:
+        return "❌ 번호를 입력하세요. (예: 1, 2, 3, 4)"
+
+    user_choice = choices[sel_num - 1]
     result = qa_chain({"query": user_choice})
     retrieved_context = "\n".join([doc.page_content for doc in result["source_documents"]])
 
@@ -94,15 +99,13 @@ def run_episode(selection):
     """
 
     response = llm_pipeline(prompt)[0]["generated_text"]
-    return f"[성진우 응답]\n{response}"
-
-# ====== Gradio UI ======
-demo = gr.Interface(
-    fn=run_episode,
-    inputs=gr.Dropdown(choices=["1", "2", "3", "4"], label="선택 번호", type="value"),
-    outputs="text",
-    title="성진우 선택 시뮬레이션",
-    description="번호를 선택하면 성진우의 응답이 생성됩니다."
+    return f"[성진우]\n{response}"
+
+# ====== Gradio ChatInterface ======
+demo = gr.ChatInterface(
+    respond,
+    title="성진우 선택 시뮬레이션 (카카오톡 스타일)",
+    description="1~4번 중 하나를 입력하면 성진우의 응답이 대화 형식으로 나타납니다."
 )
 
 if __name__ == "__main__":