melhorias de referência

.vscode/launch.json

@@ -11,7 +11,7 @@
             "module": "streamlit",
             "args": [
                 "run",
-                "app/main.py"
+                "app.py"
             ]
         }
     ]

app.py

@@ -44,7 +44,7 @@ input = st.chat_input("Digite sua pergunta:")
 
 if input:
     st.chat_message("user").markdown(input)
-    st.session_state.messages.append({"role": "assistant", "content": input})
+    st.session_state.messages.append({"role": "user", "content": input})
     documentos = vector_store.similarity_search(input, k=3)
 
     documentos_json = []
@@ -58,7 +58,7 @@ if input:
         documentos_json.append(documento_json)
 
     prompt = base_prompt.format(context=documentos_json, question=input)
-    print(prompt)
+
     st.session_state.messages_for_ia.append({"role": "assistant", "content": prompt})
 
     resposta = model.invoke(st.session_state.messages_for_ia)

prompts/base.md

@@ -3,8 +3,8 @@
 Responda à pergunta abaixo com base no contexto fornecido em formato JSON. O contexto contém três referências principais para sua resposta:
 
 * "source": Indica o caminho e o nome do arquivo PDF onde a informação foi encontrada.
-* "page": Representa o número da página no arquivo PDF onde o conteúdo está localizado.
-* "page_label": Corresponde à numeração da página conforme exibida no documento original.
+* "page": Corresponde à numeração da página conforme exibida no documento original.
+* "page_label":   Representa o número da página no arquivo PDF onde o conteúdo está localizado.
 * "page_content": Contém o trecho extraído do arquivo PDF.
 
 Caso existam outras chaves no JSON, ignore-as.
@@ -13,7 +13,7 @@ Sempre que utilizar uma referência do contexto, inclua a fonte correspondente n
 
 # Exemplo de resposta esperada:
 
-Se a resposta for baseada em um trecho do arquivo nome_do_documento.pdf (source: ./data/articles/nome_do_documento.pdf) na página 12 do arquivo (page: 12) e página numerada 10 no documento (page_label: 10), cite a fonte da seguinte forma:
+Se a resposta for baseada em um trecho do arquivo nome_do_documento.pdf (source: ./data/articles/nome_do_documento.pdf) na página 12 do arquivo (page_label: 12) e página numerada 10 no documento (page: 10), cite a fonte da seguinte forma:
 
 (Fonte: nome_do_documento.pdf, página 12 [Página no documento: 10])