src/main.py

from datetime import date
from langchain_core.messages import HumanMessage, AIMessage, BaseMessage, SystemMessage
from typing import List, Sequence

from src.graph_builder import graph_app
from src.graph_state import GraphState
from src.database import conn_pool

def run_chatbot():
    try:
        if conn_pool is None:
            print("\nLỗi: Không thể kết nối Database. Chatbot không thể hoạt động.")
            return

        from src.tools import fetch_locations_tool
        initial_locations = fetch_locations_tool()
        if initial_locations is None:
            print("\nLỗi: Không thể truy vấn dữ liệu từ Database. Chatbot không thể hoạt động.")
            return
        print(f"\nChatbot đã sẵn sàng! Các địa điểm hỗ trợ: {', '.join(initial_locations[:10])}{'...' if len(initial_locations) > 10 else ''}")

    except Exception as e:
        print(f"\nLỗi khởi tạo: {e}")
        return

    conversation_history: List[BaseMessage] = []

    print("\n--- Bắt đầu trò chuyện (gõ 'quit' để thoát) ---")

    while True:
        try:
            user_input = input("Bạn: ")
            if user_input.lower() == 'quit':
                print("Chatbot: Tạm biệt!")
                break
            if not user_input.strip():
                continue

            conversation_history.append(HumanMessage(content=user_input))

            graph_input: GraphState = {
                "messages": conversation_history,
                "user_query": None, "current_date": None, "available_locations": None,
                "extracted_entities": None, "search_results": None,
                "final_response": None, "error": None,
                "routing_decision": None,
            }

            final_state = graph_app.invoke(graph_input)

            conversation_history = list(final_state.get("messages", conversation_history))
            response = final_state.get("final_response", "Xin lỗi, tôi không thể xử lý yêu cầu này.")

            print(f"Chatbot: {response}")

        except KeyboardInterrupt:
            print("\nChatbot: Tạm biệt!")
            break
        except Exception as e:
            print(f"Chatbot: Đã xảy ra lỗi: {e}. Vui lòng thử lại.")

if __name__ == "__main__":
    run_chatbot()