Update app.py

app.py

@@ -426,7 +426,31 @@ def get_hf_user(token):
 def chat(message, history, hf_token):
     username = get_hf_user(hf_token) if hf_token else "guest"
     try:
-        response = agent.run(message)
+        response = agent.invoke(message)
+        if isinstance(response, dict):
+            if "action_input" in response:
+                response = response["action_input"]
+            elif "output" in response:
+                response = response["output"]
+            elif "text" in response:
+                response = response["text"]
+            else:
+                response = str(response)
+        elif isinstance(response, str):
+            # "Final Answer"가 포함된 문자열이면 그 부분만 추출
+            if '"action_input":' in response:
+                import re, json
+                match = re.search(r'["\']action_input["\']\s*:\s*["\'](.*?)["\']', response)
+                if match:
+                    response = match.group(1)
+            elif "Final Answer" in response:
+                # {"action": "Final Answer", "action_input": "..."} 형식일 때
+                try:
+                    data = json.loads(response)
+                    if isinstance(data, dict) and "action_input" in data:
+                        response = data["action_input"]
+                except Exception:
+                    response = response.replace("Final Answer", "").strip()
     except Exception as e:
         response = f"⚠️ 오류: {e}"
     history = history + [(message, response)]
@@ -434,11 +458,10 @@ def chat(message, history, hf_token):
         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 refresh_conversation_list():
+    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)
 def load_selected(file):
     return load_conversation(file)
 
@@ -473,7 +496,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_list, [hf_token], convo_files)
+            refresh_btn.click(refresh_conversation_list, [hf_token], convo_files)
             load_btn.click(load_selected, [convo_files], chatbot)
 
 if __name__ == "__main__":