> 백엔드 개발 > 파이썬 튜토리얼 > OpenAI와 Streamlit을 활용한 문서 검색 및 Q&A 시스템 구축

OpenAI와 Streamlit을 활용한 문서 검색 및 Q&A 시스템 구축

Susan Sarandon
풀어 주다: 2024-11-07 15:50:03
원래의
559명이 탐색했습니다.

안녕하세요, 개발자 커뮤니티 여러분! ?

오늘은 제 프로젝트인 EzioDevIo RAG(Retrieval-Augmented Generation)를 소개하게 되어 기쁩니다. 이 시스템을 통해 사용자는 PDF 문서를 업로드하고, 내용에 따라 질문하고, OpenAI의 GPT-3.5 Turbo 모델에서 생성된 실시간 답변을 받을 수 있습니다. 이는 대용량 문서를 탐색하거나 관련 정보를 빠르게 추출하는 데 특히 유용합니다. ??

Building a Document Retrieval & Q&A System with OpenAI and Streamlit

내 GitHub: EzioDevIo RAG 프로젝트에서 전체 코드를 찾을 수 있습니다. 프로젝트를 자세히 살펴보고 각 단계를 자세히 살펴보겠습니다!

? EzioDevIo RAG 프로젝트 GitHub 저장소에서 전체 코드베이스 및 설정 지침을 살펴보세요!

사업개요

학습 내용

  1. OpenAI의 언어 모델을 PDF 문서 검색과 통합하는 방법
  2. Streamlit을 사용하여 사용자 친화적인 인터페이스를 만드는 방법
  3. 손쉬운 배포를 위해 Docker로 애플리케이션을 컨테이너화하는 방법. 프로젝트 특징
  • PDF를 업로드하고 정보를 얻으세요.
  • 업로드된 PDF의 내용을 바탕으로 질문하세요.
  • OpenAI의 gpt-3.5-turbo 모델이 생성한 실시간 응답
  • 확장성을 위해 Docker를 사용하여 쉽게 배포합니다.

*프로젝트 디렉토리의 최종 구조는 다음과 같습니다. *

RAG-project/
├── .env                       # Environment variables (API key)
├── app.py                     # Streamlit app for the RAG system
├── document_loader.py         # Code for loading and processing PDF documents
├── retriever.py               # Code for indexing and retrieving documents
├── main.py                    # Main script for RAG pipeline
├── requirements.txt           # List of required libraries
├── Dockerfile                 # Dockerfile for containerizing the app
├── .gitignore                 # Ignore sensitive and unnecessary files
├── data/
│   └── uploaded_pdfs/         # Folder to store uploaded PDFs
└── images/
    └── openai_api_setup.png   # Example image for OpenAI API setup instructions

로그인 후 복사
로그인 후 복사
로그인 후 복사
로그인 후 복사

1단계: 프로젝트 설정

전제조건

다음 사항을 확인하세요.

  • Python 3.8: 애플리케이션을 로컬에서 실행합니다.
  • OpenAI API 키: OpenAI 모델에 액세스하려면 이 키가 필요합니다. OpenAI API에 가입하여 키를 받으세요.
  • Docker: 선택 사항이지만 배포용 애플리케이션을 컨테이너화하는 데 권장됩니다.

2단계: 저장소 복제 및 가상 환경 설정

2.1. 저장소 복제
GitHub에서 프로젝트 저장소를 복제하고 프로젝트 디렉토리로 이동하여 시작하세요.

git clone https://github.com/EzioDEVio/RAG-project.git
cd RAG-project

로그인 후 복사
로그인 후 복사
로그인 후 복사

2.2. 가상 환경 설정
프로젝트 종속성을 격리하려면 가상 환경을 만들고 활성화하세요. 이는 다른 프로젝트 패키지와의 충돌을 방지하는 데 도움이 됩니다.

python -m venv venv
source venv/bin/activate  # On Windows, use `venv\Scripts\activate`

로그인 후 복사
로그인 후 복사
로그인 후 복사

2.3. 종속성 설치
요구 사항.txt에 나열된 필수 Python 라이브러리를 설치합니다. 여기에는 언어 모델용 OpenAI, UI용 Streamlit, PDF 처리용 PyMuPDF, 효율적인 유사성 검색용 FAISS가 포함됩니다.

pip install -r requirements.txt
로그인 후 복사
로그인 후 복사

2.4. OpenAI API 키 구성
프로젝트 루트 디렉터리에 .env 파일을 만듭니다. 이 파일은 OpenAI API 키를 안전하게 저장합니다. 파일에 다음 줄을 추가하고 your_openai_api_key_here를 실제 API 키로 바꿉니다.

RAG-project/
├── .env                       # Environment variables (API key)
├── app.py                     # Streamlit app for the RAG system
├── document_loader.py         # Code for loading and processing PDF documents
├── retriever.py               # Code for indexing and retrieving documents
├── main.py                    # Main script for RAG pipeline
├── requirements.txt           # List of required libraries
├── Dockerfile                 # Dockerfile for containerizing the app
├── .gitignore                 # Ignore sensitive and unnecessary files
├── data/
│   └── uploaded_pdfs/         # Folder to store uploaded PDFs
└── images/
    └── openai_api_setup.png   # Example image for OpenAI API setup instructions

로그인 후 복사
로그인 후 복사
로그인 후 복사
로그인 후 복사

? 팁: 프로젝트를 공개 저장소에 푸시하는 경우 API 키가 노출되지 않도록 .gitignore 파일에 .env가 추가되었는지 확인하세요.

3단계: 프로젝트 구조 이해
다음은 코드 탐색에 도움이 되는 디렉토리 구조에 대한 간략한 개요입니다.
다음은 코드 탐색에 도움이 되는 디렉토리 구조에 대한 간략한 개요입니다.

git clone https://github.com/EzioDEVio/RAG-project.git
cd RAG-project

로그인 후 복사
로그인 후 복사
로그인 후 복사

각 파일에는 특정 역할이 있습니다.

  • app.py: Streamlit 인터페이스를 관리하여 사용자가 파일을 업로드하고 질문할 수 있도록 합니다.
  • document_loader.py: PyMuPDF를 사용하여 PDF 로드 및 처리를 처리합니다.
  • retriever.py: FAISS를 사용하여 문서 텍스트를 색인화하고 사용자 쿼리를 기반으로 관련 섹션을 검색합니다.
  • main.py: OpenAI의 API를 호출하여 응답을 생성하는 것을 포함하여 모든 것을 하나로 묶습니다.

4단계: 핵심 코드 구축
이제 프로젝트의 주요 구성요소를 살펴보겠습니다.

4.1. 문서 로딩(document_loader.py)
document_loader.py 파일은 PDF에서 텍스트를 추출하는 역할을 합니다. 여기서는 PyMuPDF 라이브러리를 사용하여 PDF의 각 페이지를 처리하고 텍스트를 저장합니다.

python -m venv venv
source venv/bin/activate  # On Windows, use `venv\Scripts\activate`

로그인 후 복사
로그인 후 복사
로그인 후 복사

설명: 이 기능은 지정된 폴더에 있는 모든 PDF 파일을 읽고 각 페이지에서 텍스트를 추출한 다음 해당 텍스트를 사전 목록에 추가합니다. 각 사전은 텍스트와 파일 이름으로 문서를 나타냅니다.

4.2. 문서 색인화 및 검색(retriever.py)
FAISS(Facebook AI 유사성 검색)는 유사성 검색을 수행하는 데 도움이 됩니다. 이를 사용하여 사용자가 질문할 때 관련 섹션을 검색할 수 있는 문서 임베딩 색인을 생성합니다.

pip install -r requirements.txt
로그인 후 복사
로그인 후 복사

설명:

create_index: OpenAIEmbeddings를 사용하여 문서 텍스트를 임베딩으로 변환하고 FAISS로 인덱스를 생성합니다.
검색_문서: 사용자 쿼리를 기반으로 관련 문서 섹션을 검색합니다.

4.3. 응답 생성(main.py)
이 모듈은 OpenAI의 언어 모델을 사용하여 사용자 쿼리를 처리하고 관련 문서를 검색하며 답변을 생성합니다.

OPENAI_API_KEY=your_openai_api_key_here
로그인 후 복사

설명:

generate_response: 검색된 문서와 사용자 쿼리의 컨텍스트가 포함된 프롬프트를 생성한 다음 이를 OpenAI의 API로 보냅니다. 그러면 응답이 답변으로 반환됩니다.

5단계: 간소화된 인터페이스 만들기(app.py)
Streamlit은 대화형 프런트 엔드를 제공하므로 사용자가 쉽게 파일을 업로드하고 질문할 수 있습니다.

RAG-project/
├── .env                       # Environment variables (API key)
├── app.py                     # Streamlit app for the RAG system
├── document_loader.py         # Code for loading and processing PDF documents
├── retriever.py               # Code for indexing and retrieving documents
├── main.py                    # Main script for RAG pipeline
├── requirements.txt           # List of required libraries
├── Dockerfile                 # Dockerfile for containerizing the app
├── .gitignore                 # Ignore sensitive and unnecessary files
├── data/
│   └── uploaded_pdfs/         # Folder to store uploaded PDFs
└── images/
    └── openai_api_setup.png   # Example image for OpenAI API setup instructions

로그인 후 복사
로그인 후 복사
로그인 후 복사
로그인 후 복사

설명:

  • 이 코드는 Streamlit을 사용하여 간단한 UI를 생성하여 사용자가 PDF를 업로드하고 질문을 입력할 수 있도록 합니다.
  • 사용자가 '답변 받기'를 클릭하면 앱이 관련 문서를 검색하고 답변을 생성합니다.

6단계: 애플리케이션 Docker화
Docker를 사용하면 앱을 컨테이너에 패키징하여 쉽게 배포할 수 있습니다.

도커파일

RAG-project/
├── .env                       # Environment variables (API key)
├── app.py                     # Streamlit app for the RAG system
├── document_loader.py         # Code for loading and processing PDF documents
├── retriever.py               # Code for indexing and retrieving documents
├── main.py                    # Main script for RAG pipeline
├── requirements.txt           # List of required libraries
├── Dockerfile                 # Dockerfile for containerizing the app
├── .gitignore                 # Ignore sensitive and unnecessary files
├── data/
│   └── uploaded_pdfs/         # Folder to store uploaded PDFs
└── images/
    └── openai_api_setup.png   # Example image for OpenAI API setup instructions

로그인 후 복사
로그인 후 복사
로그인 후 복사
로그인 후 복사

설명:

최종 이미지를 간결하게 유지하기 위해 다단계 빌드를 사용합니다.
보안을 위해 애플리케이션은 루트가 아닌 사용자로 실행됩니다.

Docker 컨테이너 실행

  1. Docker 이미지 빌드:
git clone https://github.com/EzioDEVio/RAG-project.git
cd RAG-project

로그인 후 복사
로그인 후 복사
로그인 후 복사
  1. 컨테이너 실행:
python -m venv venv
source venv/bin/activate  # On Windows, use `venv\Scripts\activate`

로그인 후 복사
로그인 후 복사
로그인 후 복사

7단계: GitHub Actions로 CI/CD 설정
프로덕션 준비를 위해 CI/CD 파이프라인을 추가하여 Docker 이미지를 빌드, 테스트 및 스캔합니다. 이 설정을 위한 저장소에서 .github/workflows 파일을 찾을 수 있습니다.

최종 생각
이 프로젝트는 OpenAI의 언어 모델 기능과 문서 검색을 결합하여 기능적이고 대화형 도구를 만듭니다. 이 프로젝트가 마음에 드셨다면 GitHub 저장소에 별표를 표시하고 개발자 커뮤니티에서 저를 팔로우해 주세요. 더 놀라운 프로젝트를 함께 만들어 봅시다! ?

? GitHub 저장소를 보시겠습니까? EzioDevIo RAG 프로젝트 GitHub 저장소!

위 내용은 OpenAI와 Streamlit을 활용한 문서 검색 및 Q&A 시스템 구축의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

원천:dev.to
본 웹사이트의 성명
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.
저자별 최신 기사
인기 튜토리얼
더>
최신 다운로드
더>
웹 효과
웹사이트 소스 코드
웹사이트 자료
프론트엔드 템플릿