Capstone 2024 - 컴퓨터공학과 캡스톤 프로젝트 (AI담당)

프로젝트 소개

본 프로젝트는 2024년도 2학기 컴퓨터공학과 캡스톤 디자인 과목에서 진행된 결과물로, RAG를 이용하여 검색 기반으로 LLM을 통해 답변을 생성하여 AI 기반 마트 상품 찾기 시스템을 구현한 프로젝트입니다.

주요 기능

1. 텍스트 기반 상품 검색

   • 사용자가 찾고자 하는 상품을 질문하면 AI가 상품 정보를 검색하고 결과를 반환합니다.

2. 이미지 기반 상품 검색

   • 사용자가 찾고자 하는 이미지를 업로드하면 AI가 상품 이미지를 분석하여 관련 상품 정보를 제공합니다.

주요 파일 구성

프로젝트 파일의 주요 구성은 다음과 같습니다.

• rag_text.py
  • 텍스트 기반 상품 검색을 위한 RAG 처리 로직
• rag_img.py
  • 이미지 기반 상품 검색을 위한 RAG처리 로직
• db_connector.py
  • 상품 데이터베이스와 연결 및 데이터 조회 로직
• app.py
  • Flask를 사용하여 배포하는 파일