Update app.py

app.py

@@ -9,6 +9,7 @@ import os
 OPENAI_API_KEY = os.getenv('token')
 
 
+
 def construct_index(directory_path):
     max_input_size = 3500
     num_outputs = 512
@@ -23,26 +24,13 @@ def construct_index(directory_path):
 
     index = GPTSimpleVectorIndex(documents, llm_predictor=llm_predictor, prompt_helper=prompt_helper)
 
-    # Criar um índice invertido
-    inverted_index = {}
-    for doc_id, doc in enumerate(documents):
-        words = doc.split()
-        for word in words:
-            if word not in inverted_index:
-                inverted_index[word] = []
-            inverted_index[word].append(doc_id)
-
-    # Salvar o índice invertido em um arquivo
-    with open('inverted_index.json', 'w') as f:
-        json.dump(inverted_index, f)
-
     index.save_to_disk('index.json')
 
-    return index, inverted_index
+    return index
 
 
 def chatbot(input_text):
-    index, inverted_index = construct_index("docs")
+    index = GPTSimpleVectorIndex.load_from_disk('index.json')
 
     # Ler e concatenar os documentos da pasta "docs" como o contexto relevante
     docs_directory_path = "docs"
@@ -56,15 +44,7 @@ def chatbot(input_text):
     texto_entrada = f"Dentro do assunto agilidade me responda: {input_text}{contexto} se não for agilidade não responda"
     print(texto_entrada)
       
-    # Usar o índice invertido para encontrar documentos relevantes
-    relevant_docs = set()
-    for word in input_text.split():
-        if word in inverted_index:
-            relevant_docs.update(inverted_index[word])
-    relevant_docs = list(relevant_docs)
-
-    # Consultar o índice GPT-3.5 com os documentos relevantes
-    response = index.query(texto_entrada, relevant_docs=relevant_docs, response_mode="compact")
+    response = index.query(texto_entrada, response_mode="compact")
     return response.response
 
 
@@ -76,6 +56,10 @@ iface = gr.Interface(fn=chatbot,
                      inputs=gr.components.Textbox(lines=7, label="Como podemos te ajudar?"),
                      outputs="text",
                      description=description,                     
-                     title="MentorIA Ágil GPT (Beta)")
+                     title="MentorIA Ágil GPT (Beta)",
+                     allow_flagging=True,
+                     flagging_dir='flagged')
+
 
-iface.launch(share=False)
+index = construct_index("docs")
+iface.launch(share=False)