Update app.py

app.py

@@ -380,12 +380,12 @@ import json
 # ✅ 대화 요약 함수
 # ──────────────────────────────
 def summarize_title(history):
-    """대화 요약 제목"""
+    """대화 제목 요약"""
     if not history: return "새 대화"
     text = "\n".join(f"User:{h[0]} AI:{h[1]}" for h in history[-3:])
     try:
-        title = llm._call(f"다음 대화의 제목을 한 줄로 요약해줘:\n{text}")
-        return title.strip().replace("\n", " ")[:50]
+        title = llm._call(f"다음 대화의 주제를 한 줄로 요약해줘:\n{text}")
+        return title.strip().replace("\n", " ")[:60]
     except Exception:
         return "요약 실패"
 
@@ -409,9 +409,77 @@ def list_conversations(username):
 
 def load_conversation(file):
     with open(os.path.join("user_logs", file), encoding="utf-8") as jf:
-        data = json.load(jf)
-        return data["history"]
+        return json.load(jf)["history"]
 
+# ──────────────────────────────
+# ✅ 로그인 후 사용자 정보 가져오기
+# ──────────────────────────────
+def get_hf_user(token):
+    """HF OAuth 토큰으로 사용자 정보 조회"""
+    try:
+        r = requests.get("https://huggingface.co/api/whoami-v2", headers={"Authorization": f"Bearer {token}"})
+        if r.status_code == 200:
+            data = r.json()
+            return data.get("name") or data.get("email") or "unknown_user"
+    except Exception:
+        pass
+    return "guest"
+def chat(message, history, hf_token):
+    username = get_hf_user(hf_token) if hf_token else "guest"
+    try:
+        response = agent.run(message)
+    except Exception as e:
+        response = f"⚠️ 오류: {e}"
+    history = history + [(message, response)]
+    if username:
+        save_conversation(username, history)
+    return history, history, ""  # 입력 초기화
+
+def refresh_list(hf_token):
+    username = get_hf_user(hf_token)
+    items = list_conversations(username)
+    return gr.Dropdown.update(choices=[x[1] for x in items])
+
+def load_selected(file):
+    return load_conversation(file)
+
+
+# ──────────────────────────────
+# ✅ Gradio UI with HF Auth
+# ──────────────────────────────
+with gr.Blocks(theme=gr.themes.Soft(), title="PIXAL Assistant (HF Auth)") as demo:
+    gr.Markdown("## 🤖 PIXAL Assistant — Hugging Face 계정 기반 대화 저장")
+
+    hf_login = gr.LoginButton("huggingface", label="🔑 Hugging Face로 로그인")
+    hf_token = gr.State()
+
+    @hf_login.click(inputs=None, outputs=hf_token)
+    def login(token):  # 로그인 후 token 반환
+        return token
+
+    with gr.Row():
+        with gr.Column(scale=2):
+            chatbot = gr.Chatbot(label="PIXAL 대화", height=600, render_markdown=True)
+            msg = gr.Textbox(label="메시지", placeholder="입력 후 Enter 또는 전송 클릭")
+            send = gr.Button("전송")
+            clear = gr.Button("초기화")
+
+            msg.submit(chat, [msg, chatbot, hf_token], [chatbot, chatbot, msg])
+            send.click(chat, [msg, chatbot, hf_token], [chatbot, chatbot, msg])
+            clear.click(lambda: None, None, chatbot, queue=False)
+
+        with gr.Column(scale=1):
+            gr.Markdown("### 💾 저장된 대화 기록")
+            convo_files = gr.Dropdown(label="대화 선택", choices=[])
+            refresh_btn = gr.Button("🔄 목록 새로고침")
+            load_btn = gr.Button("불러오기")
+
+            refresh_btn.click(refresh_list, [hf_token], convo_files)
+            load_btn.click(load_selected, [convo_files], chatbot)
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)
+'''
 def chat(message, history):
     try:
         response = agent.run(message)
@@ -488,4 +556,4 @@ with gr.Blocks(theme=gr.themes.Soft(), title="PIXAL Assistant") as demo:
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)
 
-
+'''