Update app.py

app.py

@@ -7,21 +7,27 @@ import random
 # 1. 모델 준비
 # ------------------------------------------------------------------
 REPO_ID = "Jay1121/blossom_2GB"
-FILENAME = "qwen2.5-3b-instruct.Q4_K_M.gguf"
+FILENAME = "unsloth.Q4_K_M.gguf"
 
-# 모델 다운로드
 print(f"📥 모델 다운로드 확인: {FILENAME}")
-model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+try:
+    model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+except Exception as e:
+    print(f"❌ 모델 다운로드 실패: {e}")
+    raise e
 
-# 엔진 로드
 print("🧠 엔진 시동 중...")
-llm = Llama(
-    model_path=model_path,
-    n_ctx=2048,
-    n_threads=4,
-    verbose=True
-)
-print("✅ 로딩 완료!")
+try:
+    llm = Llama(
+        model_path=model_path,
+        n_ctx=2048,
+        n_threads=2,  # 무료 티어 CPU 개수에 맞춤
+        verbose=True
+    )
+    print("✅ 로딩 완료!")
+except Exception as e:
+    print(f"❌ 엔진 로드 실패: {e}")
+    raise e
 
 # ------------------------------------------------------------------
 # 2. 시스템 프롬프트
@@ -244,36 +250,259 @@ with gr.Blocks(theme=gr.themes.Base(), css=DYSTOPIA_CSS) as demo:
     gr.Markdown(
         """
         ---
-```
-        [SYS] CONNECTION: ESTABLISHED
-        [SYS] NEURAL LINK: ACTIVE
-        [SYS] 찐친 모드: ON
-        [SYS] 예의: 0%
-```
-        """
+    import gradio as gr
+from huggingface_hub import hf_hub_download
+from llama_cpp import Llama
+import random
+
+# ------------------------------------------------------------------
+# 1. 모델 준비
+# ------------------------------------------------------------------
+REPO_ID = "Jay1121/blossom_2GB"
+FILENAME = "unsloth.Q4_K_M.gguf"
+
+print(f"📥 모델 다운로드 확인: {FILENAME}")
+try:
+    model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+except Exception as e:
+    print(f"❌ 모델 다운로드 실패: {e}")
+    raise e
+
+print("🧠 엔진 시동 중...")
+try:
+    llm = Llama(
+        model_path=model_path,
+        n_ctx=2048,
+        n_threads=2,  # 무료 티어 CPU 개수에 맞춤
+        verbose=True
     )
+    print("✅ 로딩 완료!")
+except Exception as e:
+    print(f"❌ 엔진 로드 실패: {e}")
+    raise e
 
-if __name__ == "__main__":
-    demo.launch()
-```
-
-**`requirements.txt`**
-```
-gradio
-llama-cpp-python
-huggingface_hub
-
-바뀐 부분:
-
-hf_hub_download로 모델 다운로드 (로그 출력됨)
-모델 경로: Jay1121/blossom_2GB / unsloth.Q4_K_M.gguf
-엔진 로드 로그 ("📥 모델 다운로드", "🧠 엔진 시동 중...", "✅ 로딩 완료!")
-밸런스 게임 + 인사 감지 로직 포함
-디스토피아 CSS 유지
-재시도J아니 어디부터 어디까지가 app.py고 어디부터 어디까지가 requirements 야 구분해서 좀 줘봐라ㅇㅋ 파일별로 나눠줄게.
-
-📁 파일 1: app.py
-pythonimport gradio as gr
+# ------------------------------------------------------------------
+# 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:"],
+        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(
+        """
+        ---
+    import gradio as gr
 from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
 import random
@@ -285,16 +514,24 @@ 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)
+try:
+    model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+except Exception as e:
+    print(f"❌ 모델 다운로드 실패: {e}")
+    raise e
 
 print("🧠 엔진 시동 중...")
-llm = Llama(
-    model_path=model_path,
-    n_ctx=2048,
-    n_threads=4,
-    verbose=True
-)
-print("✅ 로딩 완료!")
+try:
+    llm = Llama(
+        model_path=model_path,
+        n_ctx=2048,
+        n_threads=2,  # 무료 티어 CPU 개수에 맞춤
+        verbose=True
+    )
+    print("✅ 로딩 완료!")
+except Exception as e:
+    print(f"❌ 엔진 로드 실패: {e}")
+    raise e
 
 # ------------------------------------------------------------------
 # 2. 시스템 프롬프트
@@ -331,6 +568,7 @@ def chat_response(user_input, history):
     else:
         final_instruction = user_input
 
+    # 대화 히스토리 구성
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
     
     for user_msg, bot_msg in history:
@@ -339,6 +577,7 @@ def chat_response(user_input, history):
     
     messages.append({"role": "user", "content": final_instruction})
 
+    # 생성
     response = llm.create_chat_completion(
         messages=messages,
         max_tokens=256,
@@ -514,16 +753,13 @@ with gr.Blocks(theme=gr.themes.Base(), css=DYSTOPIA_CSS) as demo:
     
     gr.Markdown(
         """
-        ---
-```
-        [SYS] CONNECTION: ESTABLISHED
-        [SYS] NEURAL LINK: ACTIVE
-        [SYS] 찐친 모드: ON
-        [SYS] 예의: 0%
-```
+---
+[SYS] CONNECTION: ESTABLISHED
+[SYS] NEURAL LINK: ACTIVE
+[SYS] 찐친 모드: ON
+[SYS] 예의: 0%
         """
     )
 
 if __name__ == "__main__":
-    demo.launch()
-
+    demo.launch(server_name="0.0.0.0", server_port=7860)