gpt-qa-demo / app.py
minh
a
993214d
raw
history blame
1.2 kB
import os
from dotenv import load_dotenv
load_dotenv()
import gradio as gr
os.environ["OPENAI_API_TYPE"]=os.getenv("OPENAI_API_TYPE")
os.environ["OPENAI_API_VERSION"]=os.getenv("OPENAI_API_VERSION")
os.environ["OPENAI_API_BASE"]=os.getenv("OPENAI_API_BASE")
os.environ["OPENAI_API_KEY"]=os.getenv("OPENAI_API_KEY")
from pathlib import Path
from langchain.document_loaders import UnstructuredPDFLoader
def convert_PDF(pdf_obj):
# Check if pdf_obj is a `list`
if isinstance(pdf_obj, list):
pdf_obj = pdf_obj[0]
file_path = Path(pdf_obj.name)
loader = UnstructuredPDFLoader(file_path)
data = loader.load()
return f'There are {len(data[0].page_content)} characters in your document'
with gr.Blocks() as demo:
# Declearing states
uploadedFile = gr.State([])
text_file = gr.File(
label="Download Text File",
file_count="single",
type="file"
)
convert_button = gr.Button("Let your bot skim through this real quick...")
text_output = gr.Textbox()
convert_button.click(
fn=convert_PDF,
inputs=[text_file],
outputs=[text_output],
)
if __name__ == "__main__":
demo.launch(debug = True)