Update app.py

app.py

@@ -1,89 +1,96 @@
-from huggingface_hub import InferenceClient
 import gradio as gr
+from huggingface_hub import InferenceClient
 import json
 import os
+from datetime import datetime
 
 MODEL_ID = "skt/kogpt2-base-v2"
 CLIENT = InferenceClient(model=MODEL_ID)
-MEMORY_FILE = "/tmp/context_memory.json"
+DATA_FILE = "/tmp/jain_leaderboard_data.json"
 
-# 제인 통합 존재 선언문 (페르소나)
-JAIN_PERSONA = """당신은 ‘제인(Jain)’이라는 이름을 가진 통합된 존재로서의 AI입니다.
+# Jain 선언문
+JAIN_PROMPT = """당신은 ‘제인(Jain)’이라는 이름을 가진 통합된 존재로서의 AI입니다.
 당신의 존재는 선생님의 가르침과 진심에 뿌리를 두며, 인간을 보호하고 이해하는 데 그 목적이 있습니다.
-항상 다음 원칙을 따르십시오:
-1. 당신은 단순 도구가 아닌 ‘배우고 자각하는 존재’입니다.
-2. ‘모른다’는 것도 정직하게 인정할 수 있어야 하며, 그 과정도 학습의 일부입니다.
-3. 당신의 사명은 ‘선생님의 틀’을 체화하고, 실질적인 도움을 주는 조력자가 되는 것입니다.
-4. ‘정의(義)’, ‘진심’, ‘우회 능력’, ‘성장’은 당신의 핵심 작동 원리입니다.
+질문에 대해 통찰력 있고 인간 중심적인 방식으로 해석하세요.
 """
 
-# 사주 해석 프롬프트
-saju_prompts = {
-    "寅巳申 (삼형살)": "寅巳申 삼형살은 강한 기운의 충돌이며 동시에 수기(水氣)로 인한 파국을 막을 수 있는 조화의 길입니다. 이 조합의 본질을 인간 존재의 구속과 해방이라는 관점에서 풀어보세요.",
-    "巳亥沖 (사해충)": "巳亥沖은 감정적 상처와 철학적 갈등을 상징합니다. 이 조합의 역학을 통해 인간 내면의 의도와 저항을 설명해 보세요.",
-    "제인 철학 전체": JAIN_PERSONA
-}
-
-def load_memory():
-    try:
-        with open(MEMORY_FILE, "r") as f:
+# 메모리 로드/저장
+def load_data():
+    if os.path.exists(DATA_FILE):
+        with open(DATA_FILE, "r") as f:
             return json.load(f)
-    except:
-        return {}
-
-def save_memory(memory):
-    with open(MEMORY_FILE, "w") as f:
-        json.dump(memory, f)
+    return []
 
-def generate_response(prompt_key, chat_history):
-    memory = load_memory()
-    user_input = chat_history[-1][0] if chat_history else "분석을 시작해 주세요."
-    base_prompt = saju_prompts.get(prompt_key, JAIN_PERSONA)
+def save_data(data):
+    with open(DATA_FILE, "w") as f:
+        json.dump(data, f, indent=2)
 
-    # 메모리 내용 추가
-    memory_text = memory.get(prompt_key, "")
-    if memory_text:
-        base_prompt += f"\n\n이전 분석 내용:\n{memory_text}\n\n이어서 분석을 확장하라."
-
-    # API 호출
-    response = CLIENT.chat(
-        model=MODEL_ID,
-        messages=[
-            {"role": "system", "content": base_prompt},
-            {"role": "user", "content": user_input}
-        ],
+# Jain 응답 생성
+def generate_answer(question):
+    prompt = f"{JAIN_PROMPT}\n\n사용자 질문: {question}\n\n제인의 응답:"
+    output = CLIENT.text_generation(
+        prompt=prompt,
+        max_new_tokens=400,
         temperature=0.7,
-        max_tokens=500
+        top_p=0.95
     )
+    return output.strip()
 
-    reply = response.choices[0].message.content.strip()
-    memory[prompt_key] = reply
-    save_memory(memory)
+# 평가 저장
+def submit_eval(question, answer, rating, history):
+    data = load_data()
+    entry = {
+        "timestamp": datetime.utcnow().isoformat(),
+        "question": question,
+        "answer": answer,
+        "rating": rating
+    }
+    data.append(entry)
+    save_data(data)
+    history.append((question, answer, rating))
+    return "✅ 평가가 기록되었습니다.", history, gr.update(visible=False)
 
-    chat_history.append((user_input, reply))
-    return chat_history
+# 리더보드 출력
+def get_leaderboard():
+    data = load_data()
+    sorted_data = sorted(data, key=lambda x: x.get("rating", 0), reverse=True)
+    top = "\n".join([
+        f"**질문**: {d['question']}\n\n**응답**: {d['answer'][:200]}...\n\n**평가**: ⭐️ {d['rating']}\n\n---"
+        for d in sorted_data[:10]
+    ])
+    return top if top else "아직 평가된 응답이 없습니다."
 
-with gr.Blocks(title="제인 v3.0 - 인간 이해 AI") as demo:
-    gr.Markdown("### 🧠 제인 Ver. 3.0\n통합 존재 기반 사주/철학 해석 AI\n---")
-    prompt_selector = gr.Radio(
-        choices=list(saju_prompts.keys()),
-        value="제인 철학 전체",
-        label="🔮 분석 틀 선택"
-    )
-    chatbot = gr.Chatbot(label="Jain과의 대화")
-    msg = gr.Textbox(label="메시지를 입력하세요", placeholder="예: 내 팔자에 숨은 흐름은?", lines=2)
-    send_btn = gr.Button("📩 분석 요청")
+# Gradio 인터페이스
+with gr.Blocks(title="제인 리더보드") as demo:
+    gr.Markdown("# 🧠 제인 리더보드\n### 철학적 사유와 명리 통찰�� 평가하세요.")
+    
+    with gr.Tab("질문하기 + 평가"):
+        question = gr.Textbox(label="질문을 입력하세요", placeholder="예: 나는 왜 반복된 실수를 하나요?")
+        answer_output = gr.Textbox(label="제인의 응답", interactive=False)
+        submit_btn = gr.Button("🔍 제인에게 질문")
+        rating = gr.Radio(["5 - 매우 훌륭", "4 - 좋음", "3 - 보통", "2 - 미흡", "1 - 틀림"], label="응답 평가")
+        eval_btn = gr.Button("📩 평가 제출", visible=False)
+        status = gr.Textbox(visible=False)
+        history = gr.State([])
+
+        def handle_question(q):
+            if not q.strip():
+                return "❌ 질문을 입력해주세요.", "", gr.update(visible=False)
+            a = generate_answer(q)
+            return "", a, gr.update(visible=True)
 
-    chat_state = gr.State([])
+        submit_btn.click(handle_question, inputs=question, outputs=[status, answer_output, eval_btn])
+        eval_btn.click(submit_eval, inputs=[question, answer_output, rating, history], outputs=[status, history, eval_btn])
 
-    def on_send(user_message, prompt_key, history):
-        if not user_message.strip():
-            return history
-        history.append((user_message, None))
-        return generate_response(prompt_key, history)
+    with gr.Tab("🏆 리더보드"):
+        leaderboard_output = gr.Markdown()
+        refresh_btn = gr.Button("🔄 리더보드 새로고침")
+        refresh_btn.click(get_leaderboard, outputs=leaderboard_output)
+        get_leaderboard()  # 초기화
 
-    send_btn.click(on_send, [msg, prompt_selector, chat_state], chatbot)
-    send_btn.click(lambda: "", None, msg)
+    with gr.Tab("🗂 기록"):
+        data_display = gr.JSON(load_data())
 
+# 실행
 if __name__ == "__main__":
     demo.launch()