from llama_index import SimpleDirectoryReader, ServiceContext, VectorStoreIndex, set_global_service_context, \
    StorageContext, load_index_from_storage
from llama_index.llms import OpenAI
import gradio as gr

service_context = ServiceContext.from_defaults(
    llm=OpenAI(model="gpt-3.5-turbo-0613")
)

set_global_service_context(service_context)
def construct_index(directory_path):

    docs = SimpleDirectoryReader(directory_path).load_data()
    index = VectorStoreIndex.from_documents(docs, service_context=service_context)
    index.set_index_id("vector_index")
    index.storage_context.persist("./storage")
    return index

def chatbot(input_text):
    storage_context = StorageContext.from_defaults(persist_dir="storage")
    # load index
    index = load_index_from_storage(storage_context, index_id="vector_index")
    query_engine = index.as_query_engine()
    response = query_engine.query(input_text)
    return response.response
iface = gr.Interface(fn=chatbot,
                     inputs=gr.components.Textbox(lines=7, label="Ask a question!"),
                     outputs="text",
                     title="Bosch - Home Appliances ",
                     description = "Effortless living with Bosch Home appliances chatbot - troubleshoot, maintain, and get answers with ease.",
                     css=".main {background-image: url('https://media3.bosch-home.com/Images/3200x/21473378_Washing_Machine_Product.webp'); background-size: cover;} .header {background-image: url('https://upload.wikimedia.org/wikipedia/commons/c/c3/Bosch_logo.png'); background-repeat: no-repeat; background-position: center; height: 200px;}",
)
# index = construct_index("docs")
iface.launch(share=False)