43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
import gradio as gr
|
|
from multimodal import multimodal_answer, search_with_images, build_multimodal_index
|
|
from image_analyzer import process_documnets, build_rag_chain
|
|
|
|
vectorstore = None
|
|
rag_chain = None
|
|
|
|
def upload_and_process(files):
|
|
global vectorstore, rag_chain
|
|
|
|
if not files:
|
|
return "파일을 업로드해주세요."
|
|
|
|
paths = [f.name for f in files]
|
|
vectorstore = process_documnets(files)
|
|
rag_chain = build_rag_chain(vectorstore)
|
|
|
|
return f"{len(paths)}개 문서 처리 완료!! 질문하세요"
|
|
|
|
def answer_question(question):
|
|
if rag_chain is None:
|
|
return "먼저 문서를 업로드 해 주세요."
|
|
|
|
return rag_chain.invoke(question)
|
|
|
|
with gr.Blocks(title="이미지 기반 문서 분석 시스템") as app:
|
|
gr.Markdown("# 이미지 기반 문서 분석 시스템")
|
|
gr.Markdown("스캔 문서, 영수증, 계약서 이미지를 업로드하면 질문할 수 있습니다.")
|
|
|
|
with gr.Row():
|
|
file_input = gr.File(label="문서 이미지 업로드", file_types=[".jpg", ".jpeg", ".png", ".webp"], file_count="multiple",)
|
|
status_bar = gr.Textbox(label="처리 상태", lines=3)
|
|
|
|
upload_btn = gr.Button("문서 분석 시작", variant="primary")
|
|
upload_btn.click(fn=upload_and_process, inputs=[file_input], outputs=[status_bar])
|
|
|
|
gr.Markdown("---")
|
|
question = gr.Textbox(label="질문 입력", placeholder="문서에서 찾고 싶은 내용을 입력하세요.")
|
|
answer_box = gr.Textbox(label="답변", lines=8)
|
|
ask_btn = gr.Button("질문하기")
|
|
ask_btn.click(fn=answer_question, inputs=[question], outputs=[answer_box])
|
|
|
|
app.launch() |