Update app.py

app.py

@@ -8,6 +8,7 @@ from llama_index.llms.openai import OpenAI
 # Configuração inicial
 GIT_REPO = "https://github.com/luiz-ouroboros/payment_api.git"
 PROJECT_DIR = "payment_api"
+STORAGE_DIR = "storage"
 
 # Baixar o repositório do GitHub (se não existir)
 if not os.path.exists(PROJECT_DIR):
@@ -25,9 +26,17 @@ Settings.embed_model = OpenAIEmbedding()
 # Lendo os arquivos do projeto
 documents = SimpleDirectoryReader(PROJECT_DIR).load_data()
 
-# Criando um índice vetorial para buscas
-storage_context = StorageContext.from_defaults(persist_dir="storage")
-index = VectorStoreIndex.from_documents(documents, storage_context=storage_context)
+# Verificar se o diretório de persistência existe
+if not os.path.exists(STORAGE_DIR):
+    # Criar o índice e persistir os dados
+    print("Criando índice e persistindo dados...")
+    index = VectorStoreIndex.from_documents(documents)
+    index.storage_context.persist(persist_dir=STORAGE_DIR)
+else:
+    # Carregar o índice existente
+    print("Carregando índice existente...")
+    storage_context = StorageContext.from_defaults(persist_dir=STORAGE_DIR)
+    index = VectorStoreIndex.from_documents(documents, storage_context=storage_context)
 
 # Função para responder perguntas sobre o projeto
 def respond(message, history):
@@ -39,4 +48,4 @@ def respond(message, history):
 demo = gr.ChatInterface(respond, title="IA Especializada no Meu Projeto")
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()