Update app.py

app.py

@@ -377,16 +377,40 @@ agent=initialize_agent(tools,llm,agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT
 # ──────────────────────────────
 def chat(message, history):
     try:
-        result = agent.invoke(message)
-        # dict 대응 처리
-        if isinstance(result, dict):
-            response = result.get("output") or result.get("answer") or str(result)
-        else:
-            response = str(result)
+        response = agent.run(message)
+
+        # JSON 형태로 출력될 가능성이 있는 경우 처리
+        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.append((message, response))
+
+    history = history + [(message, response)]
     return history, history
+
 with gr.Blocks(theme=gr.themes.Soft(), title="PIXAL Assistant (LangChain + GitHub LLM)") as demo:
     gr.Markdown("""
     ## 🤖 PIXAL Assistant