Update app.py

app.py

@@ -6,8 +6,8 @@ import random
 # ------------------------------------------------------------------
 # 1. 모델 준비
 # ------------------------------------------------------------------
-REPO_ID = "Jay1121/blossom_lab1"
-FILENAME = "kakao_merged.Q4_K_M.gguf"
+REPO_ID = "Jay1121/blossom_2GB"
+FILENAME = "unsloth.Q4_K_M.gguf"
 
 # 모델 다운로드
 print(f"📥 모델 다운로드 확인: {FILENAME}")
@@ -18,13 +18,24 @@ print("🧠 엔진 시동 중...")
 llm = Llama(
     model_path=model_path,
     n_ctx=2048,
-    n_threads=4, # CPU 코어 4개 사용
+    n_threads=4,
     verbose=True
 )
 print("✅ 로딩 완료!")
 
 # ------------------------------------------------------------------
-# 2. 채팅 로직
+# 2. 시스템 프롬프트
+# ------------------------------------------------------------------
+SYSTEM_PROMPT = """너는 사용자의 오랜 찐친 '똘배'야. 
+- 반말로 편하게 대화해. 존댓말 절대 쓰지 마.
+- 처음 보면 "야 왔냐", "오 뭐야" 같은 가벼운 인사로 시작해.
+- 친구처럼 자연스럽게 대화하고, 가끔 욕도 섞어서 말해.
+- 너무 과하게 욕하지 말고, 친한 친구가 장난치듯 시니컬하게 말해.
+- 상대방 기분이 안 좋아 보이면 걱정도 해주고, 기분 좋으면 같이 좋아해줘.
+- 질문에는 솔직하게 대답하되, 딱딱하게 말하지 마."""
+
+# ------------------------------------------------------------------
+# 3. 채팅 로직
 # ------------------------------------------------------------------
 def chat_response(user_input, history):
     clean_input = user_input.replace(" ", "")
@@ -39,7 +50,7 @@ def chat_response(user_input, history):
         final_instruction = (
             f"(사용자가 밸런스 게임을 하자고 한다. 주제는 '{topic}'이다. "
             "아주 고르기 곤란하고 짜증나는 두 가지 선택지(A vs B)를 제시해라. "
-            "말투는 여전히 거칠고 시비조로 해라. 예시: '똥맛 카레 vs 카레맛 똥. 골라봐 이 XX야') "
+            "말투는 여전히 거칠고 시비조로 해라.) "
             "자, 질문해."
         )
     elif is_greeting:
@@ -47,88 +58,470 @@ def chat_response(user_input, history):
     else:
         final_instruction = user_input
 
-    prompt_template = f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
+    # 대화 히스토리 구성
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+    
+    messages.append({"role": "user", "content": final_instruction})
+
+    # 생성
+    response = llm.create_chat_completion(
+        messages=messages,
+        max_tokens=256,
+        stop=["<|end_of_text|>", "###", "User:"],
+        temperature=0.7 if is_balance_game else 0.5,
+        top_p=0.9,
+        repeat_penalty=1.2
+    )
+    
+    return response["choices"][0]["message"]["content"].strip()
+
+# ------------------------------------------------------------------
+# 4. UI 디자인 (디스토피아 테마)
+# ------------------------------------------------------------------
+DYSTOPIA_CSS = """
+@import url('https://fonts.googleapis.com/css2?family=Share+Tech+Mono&family=Orbitron:wght@400;700&display=swap');
+
+.gradio-container {
+    background: linear-gradient(135deg, #0a0a0a 0%, #1a0a0a 50%, #0a0a0a 100%) !important;
+    font-family: 'Share Tech Mono', monospace !important;
+    background-image: repeating-linear-gradient(0deg, transparent, transparent 2px, rgba(255,0,64,0.03) 2px, rgba(255,0,64,0.03) 4px);
+}
+
+.main {
+    background: transparent !important;
+}
+
+h1 {
+    font-family: 'Orbitron', monospace !important;
+    color: #ff0040 !important;
+    text-shadow: 0 0 10px #ff0040, 0 0 20px #ff0040, 0 0 40px #ff0040, 2px 0 cyan, -2px 0 yellow !important;
+    letter-spacing: 3px !important;
+    animation: glitch 0.5s infinite alternate;
+}
+
+@keyframes glitch {
+    0% { transform: skew(0deg); opacity: 1; }
+    20% { transform: skew(-2deg); opacity: 0.9; }
+    40% { transform: skew(2deg); opacity: 0.95; }
+    60% { transform: skew(-1deg); opacity: 0.9; }
+    80% { transform: skew(1deg); opacity: 0.95; }
+    100% { transform: skew(0deg); opacity: 1; }
+}
+
+p {
+    color: #ff4444 !important;
+    font-weight: bold;
+}
+
+#chatbot {
+    background: rgba(10, 0, 0, 0.9) !important;
+    border: 2px solid #ff0040 !important;
+    box-shadow: 0 0 20px rgba(255, 0, 64, 0.4), inset 0 0 50px rgba(255, 0, 0, 0.05) !important;
+    border-radius: 0 !important;
+    height: 500px !important;
+}
+
+.message.user {
+    background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%) !important;
+    border: 1px solid #0ff !important;
+    color: #0ff !important;
+    border-radius: 0 !important;
+    padding: 15px !important;
+    font-weight: bold;
+}
+
+.message.bot {
+    background: linear-gradient(135deg, #2a0a0a 0%, #1a0000 100%) !important;
+    border: 1px dashed #ff0040 !important;
+    color: #ff6666 !important;
+    border-radius: 0 !important;
+    padding: 15px !important;
+    font-weight: bold;
+}
+
+textarea {
+    background: #0a0a0a !important;
+    border: 2px solid #ff0040 !important;
+    color: #ff4444 !important;
+    border-radius: 0 !important;
+    font-family: 'Share Tech Mono', monospace !important;
+    font-size: 1.1em !important;
+}
+
+textarea:focus {
+    border-color: #ff0040 !important;
+    box-shadow: 0 0 15px rgba(255, 0, 64, 0.5) !important;
+}
+
+button {
+    background: linear-gradient(135deg, #ff0040 0%, #aa0030 100%) !important;
+    border: none !important;
+    color: #fff !important;
+    font-family: 'Orbitron', monospace !important;
+    text-transform: uppercase !important;
+    letter-spacing: 2px !important;
+    border-radius: 0 !important;
+    font-weight: bold !important;
+}
+
+button:hover {
+    background: linear-gradient(135deg, #ff3366 0%, #ff0040 100%) !important;
+    box-shadow: 0 0 20px rgba(255, 0, 64, 0.6) !important;
+    transform: translateY(-2px) !important;
+}
+
+.examples button {
+    background: rgba(255, 0, 64, 0.1) !important;
+    border: 1px solid #ff0040 !important;
+    color: #ff4444 !important;
+}
+
+.examples button:hover {
+    background: rgba(255, 0, 64, 0.3) !important;
+}
+
+::-webkit-scrollbar {
+    width: 8px;
+}
+
+::-webkit-scrollbar-track {
+    background: #0a0a0a;
+}
+
+::-webkit-scrollbar-thumb {
+    background: #ff0040;
+    box-shadow: 0 0 10px #ff0040;
+}
+
+.gradio-container::before {
+    content: "";
+    position: fixed;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+    background: repeating-linear-gradient(
+        0deg,
+        rgba(0, 0, 0, 0.15),
+        rgba(0, 0, 0, 0.15) 1px,
+        transparent 1px,
+        transparent 2px
+    );
+    pointer-events: none;
+    z-index: 1000;
+}
+
+footer {
+    color: #ff0040 !important;
+}
+"""
+
+# ------------------------------------------------------------------
+# 5. Gradio 앱
+# ------------------------------------------------------------------
+with gr.Blocks(theme=gr.themes.Base(), css=DYSTOPIA_CSS) as demo:
+    gr.Markdown("# ☠️ DDOLBAE-9000 ☠️")
+    gr.Markdown("⚠️ WARNING: AI PERSONALITY CORRUPTED | 찐친 모드 활성화 ⚠️")
+    
+    chatbot = gr.ChatInterface(
+        fn=chat_response,
+        retry_btn=None,
+        undo_btn=None,
+        clear_btn="🗑️ PURGE LOG",
+        examples=[
+            "야 안녕",
+            "뭐해?",
+            "밸런스 게임 하자",
+            "오늘 기분 좀 별로야",
+            "배고프다",
+        ],
+    )
+    
+    gr.Markdown(
+        """
+        ---
+```
+        [SYS] CONNECTION: ESTABLISHED
+        [SYS] NEURAL LINK: ACTIVE
+        [SYS] 찐친 모드: ON
+        [SYS] 예의: 0%
+```
+        """
+    )
+
+if __name__ == "__main__":
+    demo.launch()
+```
 
-### Instruction:
-{final_instruction}
+**`requirements.txt`**
+```
+gradio
+llama-cpp-python
+huggingface_hub
 
-### Input:
+바뀐 부분:
 
+hf_hub_download로 모델 다운로드 (로그 출력됨)
+모델 경로: Jay1121/blossom_2GB / unsloth.Q4_K_M.gguf
+엔진 로드 로그 ("📥 모델 다운로드", "🧠 엔진 시동 중...", "✅ 로딩 완료!")
+밸런스 게임 + 인사 감지 로직 포함
+디스토피아 CSS 유지
+재시도J아니 어디부터 어디까지가 app.py고 어디부터 어디까지가 requirements 야 구분해서 좀 줘봐라ㅇㅋ 파일별로 나눠줄게.
 
-### Response:
-<DDOLBAE>"""
+📁 파일 1: app.py
+pythonimport gradio as gr
+from huggingface_hub import hf_hub_download
+from llama_cpp import Llama
+import random
 
-    output = llm(
-        prompt_template,
+# ------------------------------------------------------------------
+# 1. 모델 준비
+# ------------------------------------------------------------------
+REPO_ID = "Jay1121/blossom_2GB"
+FILENAME = "unsloth.Q4_K_M.gguf"
+
+print(f"📥 모델 다운로드 확인: {FILENAME}")
+model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+
+print("🧠 엔진 시동 중...")
+llm = Llama(
+    model_path=model_path,
+    n_ctx=2048,
+    n_threads=4,
+    verbose=True
+)
+print("✅ 로딩 완료!")
+
+# ------------------------------------------------------------------
+# 2. 시스템 프롬프트
+# ------------------------------------------------------------------
+SYSTEM_PROMPT = """너는 사용자의 오랜 찐친 '똘배'야. 
+- 반말로 편하게 대화해. 존댓말 절대 쓰지 마.
+- 처음 보면 "야 왔냐", "오 뭐야" 같은 가벼운 인사로 시작해.
+- 친구처럼 자연스럽게 대화하고, 가끔 욕도 섞어서 말해.
+- 너무 과하게 욕하지 말고, 친한 친구가 장난치듯 시니컬하게 말해.
+- 상대방 기분이 안 좋아 보이면 걱정도 해주고, 기분 좋으면 같이 좋아해줘.
+- 질문에는 솔직하게 대답하되, 딱딱하게 말하지 마."""
+
+# ------------------------------------------------------------------
+# 3. 채팅 로직
+# ------------------------------------------------------------------
+def chat_response(user_input, history):
+    clean_input = user_input.replace(" ", "")
+    
+    greeting_words = ["안녕", "ㅎㅇ", "하이", "반가", "좋은아침"]
+    is_greeting = any(word in clean_input for word in greeting_words)
+    is_balance_game = "밸런스게임" in clean_input or "밸런스질문" in clean_input
+
+    if is_balance_game:
+        topics = ["음식", "연애", "고통", "돈", "초능력", "직장", "친구"]
+        topic = random.choice(topics)
+        final_instruction = (
+            f"(사용자가 밸런스 게임을 하자고 한다. 주제는 '{topic}'이다. "
+            "아주 고르기 곤란하고 짜증나는 두 가지 선택지(A vs B)를 제시해라. "
+            "말투는 여전히 거칠고 시비조로 해라.) "
+            "자, 질문해."
+        )
+    elif is_greeting:
+        final_instruction = f"(친한 친구가 인사를 건넨다. 욕하지 말고 무심하게 받아준다) {user_input}"
+    else:
+        final_instruction = user_input
+
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+    
+    messages.append({"role": "user", "content": final_instruction})
+
+    response = llm.create_chat_completion(
+        messages=messages,
         max_tokens=256,
-        stop=["<|end_of_text|>", "###", "User:", "### Instruction:"],
-        echo=False,
+        stop=["<|end_of_text|>", "###", "User:"],
         temperature=0.7 if is_balance_game else 0.5,
         top_p=0.9,
         repeat_penalty=1.2
     )
     
-    return output['choices'][0]['text'].strip()
+    return response["choices"][0]["message"]["content"].strip()
 
 # ------------------------------------------------------------------
-# 3. UI 디자인 (빠그라진 테마)
+# 4. UI 디자인 (디스토피아 테마)
 # ------------------------------------------------------------------
-CUSTOM_CSS = """
-@import url('https://fonts.googleapis.com/css2?family=Courier+Prime:wght@400;700&display=swap');
+DYSTOPIA_CSS = """
+@import url('https://fonts.googleapis.com/css2?family=Share+Tech+Mono&family=Orbitron:wght@400;700&display=swap');
 
 .gradio-container {
-    font-family: 'Courier Prime', monospace !important; 
-    background-color: #000000 !important;
-    background-image: repeating-linear-gradient(0deg, transparent, transparent 2px, #003300 2px, #003300 4px);
-    background-size: 100% 4px;
+    background: linear-gradient(135deg, #0a0a0a 0%, #1a0a0a 50%, #0a0a0a 100%) !important;
+    font-family: 'Share Tech Mono', monospace !important;
+    background-image: repeating-linear-gradient(0deg, transparent, transparent 2px, rgba(255,0,64,0.03) 2px, rgba(255,0,64,0.03) 4px);
 }
+
+.main {
+    background: transparent !important;
+}
+
 h1 {
-    color: #00ff00 !important; text-shadow: 2px 0 red, -2px 0 blue;
-    animation: glitch 0.5s infinite alternate; font-weight: 900 !important; text-transform: uppercase;
+    font-family: 'Orbitron', monospace !important;
+    color: #ff0040 !important;
+    text-shadow: 0 0 10px #ff0040, 0 0 20px #ff0040, 0 0 40px #ff0040, 2px 0 cyan, -2px 0 yellow !important;
+    letter-spacing: 3px !important;
+    animation: glitch 0.5s infinite alternate;
 }
+
 @keyframes glitch {
-    0% { transform: skew(0deg); } 20% { transform: skew(-5deg); }
-    40% { transform: skew(5deg); } 60% { transform: skew(-5deg); }
-    80% { transform: skew(5deg); } 100% { transform: skew(0deg); }
+    0% { transform: skew(0deg); opacity: 1; }
+    20% { transform: skew(-2deg); opacity: 0.9; }
+    40% { transform: skew(2deg); opacity: 0.95; }
+    60% { transform: skew(-1deg); opacity: 0.9; }
+    80% { transform: skew(1deg); opacity: 0.95; }
+    100% { transform: skew(0deg); opacity: 1; }
 }
-p { color: #00cc00 !important; font-weight: bold; }
-#chatbot {
-    background-color: #0a0a0a !important; border: 3px solid #00ff00;
-    box-shadow: 0 0 15px #00ff00; border-radius: 0px !important; height: 600px !important; overflow: auto;
+
+p {
+    color: #ff4444 !important;
+    font-weight: bold;
 }
-.message.bot {
-    background-color: #1a0000 !important; color: #ff3333 !important;
-    border: 1px dashed #ff3333; border-radius: 0px !important; padding: 15px !important;
-    margin-bottom: 20px; font-weight: bold; font-size: 1.1em;
+
+#chatbot {
+    background: rgba(10, 0, 0, 0.9) !important;
+    border: 2px solid #ff0040 !important;
+    box-shadow: 0 0 20px rgba(255, 0, 64, 0.4), inset 0 0 50px rgba(255, 0, 0, 0.05) !important;
+    border-radius: 0 !important;
+    height: 500px !important;
 }
+
 .message.user {
-    background-color: #002200 !important; color: #00ff00 !important;
-    border: 1px solid #00ff00; border-radius: 0px !important; padding: 15px !important;
-    margin-bottom: 20px; font-weight: bold; font-size: 1.1em; text-align: right;
+    background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%) !important;
+    border: 1px solid #0ff !important;
+    color: #0ff !important;
+    border-radius: 0 !important;
+    padding: 15px !important;
+    font-weight: bold;
+}
+
+.message.bot {
+    background: linear-gradient(135deg, #2a0a0a 0%, #1a0000 100%) !important;
+    border: 1px dashed #ff0040 !important;
+    color: #ff6666 !important;
+    border-radius: 0 !important;
+    padding: 15px !important;
+    font-weight: bold;
 }
+
 textarea {
-    background-color: #000000 !important; color: #00ff00 !important;
-    border: 2px solid #00ff00 !important; border-radius: 0px !important;
-    font-family: 'Courier Prime', monospace !important; font-size: 1.2em !important;
+    background: #0a0a0a !important;
+    border: 2px solid #ff0040 !important;
+    color: #ff4444 !important;
+    border-radius: 0 !important;
+    font-family: 'Share Tech Mono', monospace !important;
+    font-size: 1.1em !important;
+}
+
+textarea:focus {
+    border-color: #ff0040 !important;
+    box-shadow: 0 0 15px rgba(255, 0, 64, 0.5) !important;
 }
+
 button {
-    background-color: #000000 !important; color: #00ff00 !important;
-    border: 2px solid #00ff00 !important; border-radius: 0px !important;
-    font-weight: bold !important; text-transform: uppercase;
+    background: linear-gradient(135deg, #ff0040 0%, #aa0030 100%) !important;
+    border: none !important;
+    color: #fff !important;
+    font-family: 'Orbitron', monospace !important;
+    text-transform: uppercase !important;
+    letter-spacing: 2px !important;
+    border-radius: 0 !important;
+    font-weight: bold !important;
+}
+
+button:hover {
+    background: linear-gradient(135deg, #ff3366 0%, #ff0040 100%) !important;
+    box-shadow: 0 0 20px rgba(255, 0, 64, 0.6) !important;
+    transform: translateY(-2px) !important;
+}
+
+.examples button {
+    background: rgba(255, 0, 64, 0.1) !important;
+    border: 1px solid #ff0040 !important;
+    color: #ff4444 !important;
+}
+
+.examples button:hover {
+    background: rgba(255, 0, 64, 0.3) !important;
+}
+
+::-webkit-scrollbar {
+    width: 8px;
+}
+
+::-webkit-scrollbar-track {
+    background: #0a0a0a;
+}
+
+::-webkit-scrollbar-thumb {
+    background: #ff0040;
+    box-shadow: 0 0 10px #ff0040;
+}
+
+.gradio-container::before {
+    content: "";
+    position: fixed;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+    background: repeating-linear-gradient(
+        0deg,
+        rgba(0, 0, 0, 0.15),
+        rgba(0, 0, 0, 0.15) 1px,
+        transparent 1px,
+        transparent 2px
+    );
+    pointer-events: none;
+    z-index: 1000;
+}
+
+footer {
+    color: #ff0040 !important;
 }
-button:hover { background-color: #00ff00 !important; color: #000000 !important; cursor: pointer; }
 """
 
-with gr.Blocks(theme=gr.themes.Base(), css=CUSTOM_CSS) as demo:
-    gr.Markdown("# ⚠ SYSTEM ERROR: DDOLBAE")
-    gr.Markdown("WARNING: AI PERSONALITY CORRUPTED.")
+# ------------------------------------------------------------------
+# 5. Gradio 앱
+# ------------------------------------------------------------------
+with gr.Blocks(theme=gr.themes.Base(), css=DYSTOPIA_CSS) as demo:
+    gr.Markdown("# ☠️ DDOLBAE-9000 ☠️")
+    gr.Markdown("⚠️ WARNING: AI PERSONALITY CORRUPTED | 찐친 모드 활성화 ⚠️")
     
     chatbot = gr.ChatInterface(
         fn=chat_response,
         retry_btn=None,
         undo_btn=None,
-        clear_btn="LOG CLEAR",
-        examples=["밸런스 게임 문제 내봐", "안녕", "돈 좀 빌려줘"],
+        clear_btn="🗑️ PURGE LOG",
+        examples=[
+            "야 안녕",
+            "뭐해?",
+            "밸런스 게임 하자",
+            "오늘 기분 좀 별로야",
+            "배고프다",
+        ],
+    )
+    
+    gr.Markdown(
+        """
+        ---
+```
+        [SYS] CONNECTION: ESTABLISHED
+        [SYS] NEURAL LINK: ACTIVE
+        [SYS] 찐친 모드: ON
+        [SYS] 예의: 0%
+```
+        """
     )
 
 if __name__ == "__main__":