6. PDF 내용으로 답하는 챗봇-2024
나만의 챗GPT 만들기4
목표
랭체인을 사용하여 PDF를 읽자.
그 내용을 바탕으로 질문에 답하는 챗봇을 만들어보자.
순서
PDF 다운로드
챗봇 구현
크로마 - 오픈 소스 임베딩 데이터 베이스 사용
크라마 DB 설치
랭체인 사용
1
크라마 DB 설치
pip install chromadb
2
chatbot.py
import os
os.environ["OPENAI_API_KEY"] = "sk-proj-gmP2swflnrO"
from openai import OpenAI
client = OpenAI()
python chatbot.py
D:\1work>python chatbot.py
질문을 입력하세요
핵심 평가 지표는?
핵심 평가 지표는 UN도시평가(LOSI*) 순위, 스마트서비스혁신성, 4차 산업혁명기술적용 서비스비율, 스마트도시 관련투자유치규모 등이 있습니다.
소스 코드 내용
import os
os.environ["OPENAI_API_KEY"] = "sk-proj-swflnrO"
from openai import OpenAI
client = OpenAI()
from langchain_community.document_loaders import PyPDFLoader
from langchain_openai import OpenAIEmbeddings
from langchain.indexes.vectorstore import VectorstoreIndexCreator
loader = PyPDFLoader("★ 서울특별시 스마트도시 및 정보화 기본계획(홈페이지 게시용).pdf")
index = VectorstoreIndexCreator(embedding=OpenAIEmbeddings()).from_loaders([loader])
print("질문을 입력하세요")
answer = index.query(input())
print(answer)
다음
https://brunch.co.kr/@topasvga/3869
