import chromadb
import logging
from chromadb.config import Settings

# 로깅 설정
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# ----------------- 수정할 내용 -----------------
# 1. ChromaDB 클라이언트 설정
chroma_client = chromadb.HttpClient(
    host="localhost",
    port=8001,
    settings=Settings(chroma_api_impl="rest")
)

# 2. 컬렉션 이름
COLLECTION_NAME = "segue_saved_qna13"

# 3. 삭제할 ID 목록 (search_entry.py 실행 결과에서 복사하여 붙여넣으세요)
IDS_TO_DELETE = [
    "saved_1756036338"
    # 필요한 만큼 ID를 추가하세요.
]
# -----------------------------------------------

# ID 목록이 비어있는지 확인
if not IDS_TO_DELETE or IDS_TO_DELETE == ["id_1_to_delete", "id_2_to_delete"]:
    logging.warning("삭제할 ID 목록이 비어있습니다. IDS_TO_DELETE에 삭제할 ID를 입력하세요.")
else:
    try:
        collection = chroma_client.get_collection(name=COLLECTION_NAME)
        logging.info(f"'{COLLECTION_NAME}' 컬렉션에 연결했습니다.")

        # ID를 사용하여 데이터 삭제
        collection.delete(ids=IDS_TO_DELETE)

        logging.info(f"요청하신 {len(IDS_TO_DELETE)}개 데이터의 삭제를 완료했습니다: {IDS_TO_DELETE}")

    except Exception as e:
        logging.error(f"오류 발생: {e}")