app.py

# coding=utf8
from gpt_index import SimpleDirectoryReader, GPTListIndex, GPTSimpleVectorIndex, LLMPredictor, PromptHelper
from langchain import OpenAI
import gradio as gr
import sys
import os

os.environ["OPENAI_API_KEY"] = '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'

def construct_index(directory_path):
    max_input_size = 4096
    num_outputs = 512
    max_chunk_overlap = 20
    chunk_size_limit = 600

    prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)

    llm_predictor = LLMPredictor(llm=OpenAI(temperature=0.7, model_name="text-davinci-003", max_tokens=num_outputs))

    documents = SimpleDirectoryReader(directory_path).load_data()

    index = GPTSimpleVectorIndex.from_documents(documents)
    index.save_to_disk('index.json')

    return index


def chatbot(input_text):
    index = GPTSimpleVectorIndex.load_from_disk('index.json')
    response = index.query(input_text)
    return response.response

description = """

<center>Olá eu sou a Pharma, seu farmaceutico virtual. Estou aqui para tirar suas duvidas sobre procedimentos, agendamentos, preços, cuidados, bulas etc.
<img src="https://t3.ftcdn.net/jpg/02/01/38/00/360_F_201380017_cNP9520EXzGG3xPUsGkSZvqxcFQsB5Ae.jpg" width=200px></center>
"""

iface = gr.Interface(fn=chatbot,
                     inputs=gr.inputs.Textbox(lines=3, label='O quê gostaria de saber?') ,
                     outputs=gr.inputs.Textbox(lines=7, label="Veja o que encontrei"),
                     description=description,
                     css=".gradio-container-3-23-0 {background-color: #e3e3e3} .gradio-container-3-23-0 .prose * {color: #000000de}",
                     title="Farmaceutico Virtual")

index = construct_index("docs")
iface.launch()