app.py

import gradio as gr
import PyPDF2
import io

# PDF 파일에서 텍스트를 추출하는 함수
def extract_text_from_pdf(pdf_file):
    reader = PyPDF2.PdfReader(io.BytesIO(pdf_file))
    text = ""
    for page in range(len(reader.pages)):
        text += reader.pages[page].extract_text()
    return text

# 추출된 텍스트를 기반으로 질문에 답변하는 함수
def answer_question(pdf_file, question):
    extracted_text = extract_text_from_pdf(pdf_file)
    if question in extracted_text:
        start = extracted_text.find(question)
        end = extracted_text.find('.', start) + 1
        return extracted_text[start:end]
    else:
        return "질문에 대한 답변을 찾을 수 없습니다."

# Gradio 인터페이스 정의
iface = gr.Interface(
    fn=answer_question,
    inputs=[gr.File(type="binary"), gr.Textbox(label="질문")],
    outputs=gr.Textbox()
)

# 인터페이스 실행
iface.launch()