Update app.py

app.py

@@ -423,6 +423,45 @@ def get_hf_user(token):
     except Exception:
         pass
     return "guest"
+import re, json
+
+def chat(message, history, system_prompt):
+    try:
+        # 시스템 메시지 포함 프롬프트 구성
+        prompt = f"{system_prompt}\n\n사용자: {message}" if system_prompt else message
+
+        raw_response = agent.invoke(prompt)
+
+        # 문자열 변환
+        text = str(raw_response)
+
+        # ✅ JSON 블록이 있을 경우 파싱
+        match = re.search(r"\{.*\}", text, re.DOTALL)
+        if match:
+            try:
+                obj = json.loads(match.group(0))
+                # "action_input" 또는 "Final Answer" 키 우선 출력
+                if "Final Answer" in obj:
+                    output = obj["Final Answer"]
+                else:
+                    # 혹시 "output"이나 "content" 같은 키가 있으면 그걸 사용
+                    for k in ("output", "content", "answer"):
+                        if k in obj:
+                            output = obj[k]
+                            break
+                    else:
+                        output = text
+            except Exception:
+                output = text
+        else:
+            output = text
+
+    except Exception as e:
+        output = f"⚠️ 오류: {e}"
+
+    history = history + [(message, output)]
+    return history, history, ""
+'''
 def chat(message, history, hf_token):
     username = get_hf_user(hf_token) if hf_token else "guest"
     try:
@@ -457,11 +496,15 @@ def chat(message, history, hf_token):
     if username:
         save_conversation(username, history)
     return history, history, ""  # 입력 초기화
-
-def refresh_conversation_list():
+'''
+# 예: hf_token (혹은 username) 을 입력으로 받도록 변경
+def refresh_conversation_list(_=None):
+    if not os.path.exists("user_logs"):
+        return gr.update(choices=[], value=None)
     files = sorted(os.listdir("user_logs"), reverse=True)
     titles = [f.replace(".json", "") for f in files]
-    return gr.update(choices=titles, value=titles[-1] if titles else None)
+    return gr.update(choices=titles, value=titles[0] if titles else None)
+
 def load_selected(file):
     return load_conversation(file)
 
@@ -496,7 +539,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="PIXAL Assistant (HF Auth)") as dem
             refresh_btn = gr.Button("🔄 목록 새로고침")
             load_btn = gr.Button("불러오기")
 
-            refresh_btn.click(refresh_conversation_list, [hf_token], convo_files)
+            refresh_btn.click(refresh_conversation_list, None, convo_files)
             load_btn.click(load_selected, [convo_files], chatbot)
 
 if __name__ == "__main__":