Lucasmacedo/AI-ifsp-chatbot-editais
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Feat: Final changes to prompt

Browse Source
This commit is contained in:
DNXBrasil
2025-10-22 16:44:01 -03:00
parent f2df55f0f9
commit 42f9d22c77
32 changed files with 1122 additions and 1607 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
front/app/front.py
View File

@@ -20,6 +20,9 @@ if "chat_answer_history" not in st.session_state:
if "chat_history" not in st.session_state:
st.session_state["chat_history"] = []
prompt=st.chat_input(placeholder="Digite uma mensagem...",key="prompt")
for generated_response, user_query in zip(st.session_state["chat_answer_history"],st.session_state["user_prompt_history"]):
st.chat_message("user").write(user_query)
st.chat_message("assistant").write(generated_response)
def create_sources_string(source_urls: Set[str])->str:
if not source_urls:
return ""
@@ -30,22 +33,25 @@ def create_sources_string(source_urls: Set[str])->str:
sources_string+=f"{i+1}, {source}\n"
return sources_string
if prompt:
st.chat_message("user").write(prompt)
with st.spinner("Generating response.."):
payload=[{"role":"user","content":prompt}]
content={"message":payload,"chat_history":st.session_state["chat_history"]}
content={"message":payload,"chat_history":st.session_state["chat_history"],"username":st.session_state["username"]}
headers={"Content-type":"application/json","x-api-key":json.loads(st_auth.get_secret())['api-gateway-api-key']}
generated_response=json.loads(requests.post(url,json=content,headers=headers).text)
if 'json' in generated_response:
if 'chat_history' in generated_response:
if st.session_state["chat_history"] == []:
st.session_state["chat_history"] = generated_response['chat_history']
if 'json' in generated_response:
generated_response=generated_response['json']
if 'message' in generated_response and generated_response['message']=="Endpoint request timed out":
generated_response="Falta de dados: Por favor encaminhe a conversa para a equipe desenvolvedora"
#generated_response=[{"role":"user","content":prompt}]
# sources= set([doc.metadata["source"] for doc in generated_response['context']])
#formatted_response=f"{generated_response['answer']} \n\n {create_sources_string(sources)}"
formatted_response=generated_response
st.chat_message("assistant").write(formatted_response)
st.session_state["user_prompt_history"].append(prompt)
st.session_state["chat_answer_history"].append(formatted_response)
st.session_state["chat_history"]=st.session_state["chat_history"]+[{"role":"user","content":prompt}]+[{"role":"assistant","content":formatted_response}]
st.session_state.user_input=""
if st.session_state["chat_answer_history"]:
for generated_response, user_query in zip(st.session_state["chat_answer_history"],st.session_state["user_prompt_history"]):
st.chat_message("user").write(user_query)
st.chat_message("assistant").write(generated_response)
st.session_state.user_input=""