import os
import gradio as gr
from dotenv import load_dotenv
from llama_index import GPTVectorStoreIndex, download_loader
import openai

load_dotenv()

openai.api_key = os.getenv("OPENAI_API_KEY")

UnstructuredReader = download_loader('UnstructuredReader', refresh_cache=True)
loader = UnstructuredReader()
# data = loader.load_data(pdf.name, split_documents=False)
# index = GPTVectorStoreIndex.from_documents(data)
# query_engine = index.as_query_engine()

data = loader.load_data("A Developer's Guide to Bitcoin Ordinals.pdf", split_documents=False)
index = GPTVectorStoreIndex.from_documents(data)
query_engine = index.as_query_engine()

def same_auth(username, password):
    return username == password

def answer_question(question):
    response = query_engine.query(question)
    return response

iface = gr.Interface(fn=answer_question, 
                     inputs=[gr.File(label="Upload a PDF file"), 
                             gr.Textbox(label="Enter your question")], 
                     outputs="text")

iface.launch()