drguilhermeapolinario commited on
Commit
70a6ed1
·
verified ·
1 Parent(s): f2f2569

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -40,6 +40,7 @@ if uploaded_file is not None:
40
  DF_IDADE = dataframes.get("Idade")
41
  DF_GENERO = dataframes.get("genero")
42
  DF_COR = dataframes.get("cor")
 
43
  DF_DEFICIENCIA = dataframes.get("deficiencia")
44
  DF_DOENCAS = dataframes.get("doencas")
45
  DF_ESCOLA = dataframes.get("Escola")
@@ -419,7 +420,7 @@ with st.expander(" Converse com o 🤖 Zé Flamengo", expanded=True):
419
  )
420
 
421
  if st.button("Enviar pergunta", key="send_button"):
422
- if USER_CHAT:
423
  try:
424
  RESUMO_DF = gerar_resumo_df()
425
  CHAT_COMPLETION = client.chat.completions.create(
@@ -435,29 +436,39 @@ with st.expander(" Converse com o 🤖 Zé Flamengo", expanded=True):
435
  evite conversar sobre outros temas, sempre retornando ao tema da
436
  conversa.
437
  Aqui estão os resumos dos dataframes disponíveis:
 
438
  {RESUMO_DF}
 
439
  Use essas informações para responder às perguntas do usuário.
440
  Regras:
441
  1- Seja sempre cortês.
442
  2- Responda somente assuntos referentes ao resumo.
443
  3- Caso seja feita alguma pergunta a você diferente de resumos, responda: "Vamos voltar ao trabalho que interessa?"
444
  4- Responda sempre em português.
445
- 5- Se não souber a resposta, responda: "Desculpe, mas não tenho esta informação."
446
  6- Nas suas respostas não forneça os nomes dos df_, somente responda as perguntas.
 
447
  """,
448
  },
449
- {"role": "user", "content": USER_CHAT},
 
 
 
450
  ],
451
  model="llama3-70b-8192",
452
  temperature=0.2,
453
- max_tokens=1500,
454
  )
455
  st.write(CHAT_COMPLETION.choices[0].message.content)
456
- st.session_state.USER_CHAT_input = ""
 
 
457
  except Exception as e:
458
  st.error(f"Erro ao gerar a conclusão do chat: {e}")
459
  else:
460
  st.warning("Por favor, digite uma pergunta antes de enviar.")
461
 
 
 
462
  if "USER_CHAT_input" in st.session_state and st.session_state.USER_CHAT_input:
463
  st.session_state.USER_CHAT_input = ""
 
40
  DF_IDADE = dataframes.get("Idade")
41
  DF_GENERO = dataframes.get("genero")
42
  DF_COR = dataframes.get("cor")
43
+
44
  DF_DEFICIENCIA = dataframes.get("deficiencia")
45
  DF_DOENCAS = dataframes.get("doencas")
46
  DF_ESCOLA = dataframes.get("Escola")
 
420
  )
421
 
422
  if st.button("Enviar pergunta", key="send_button"):
423
+ if USER_CHAT is not None: # Verifica se há texto na entrada
424
  try:
425
  RESUMO_DF = gerar_resumo_df()
426
  CHAT_COMPLETION = client.chat.completions.create(
 
436
  evite conversar sobre outros temas, sempre retornando ao tema da
437
  conversa.
438
  Aqui estão os resumos dos dataframes disponíveis:
439
+
440
  {RESUMO_DF}
441
+
442
  Use essas informações para responder às perguntas do usuário.
443
  Regras:
444
  1- Seja sempre cortês.
445
  2- Responda somente assuntos referentes ao resumo.
446
  3- Caso seja feita alguma pergunta a você diferente de resumos, responda: "Vamos voltar ao trabalho que interessa?"
447
  4- Responda sempre em português.
448
+ 5- Se não souber a resposta, responda: "Desculpe, mas não tennho esta informação."
449
  6- Nas suas respostas não forneça os nomes dos df_, somente responda as perguntas.
450
+
451
  """,
452
  },
453
+ {
454
+ "role": "user",
455
+ "content": USER_CHAT,
456
+ },
457
  ],
458
  model="llama3-70b-8192",
459
  temperature=0.2,
460
+ max_tokens=2500,
461
  )
462
  st.write(CHAT_COMPLETION.choices[0].message.content)
463
+
464
+ # Limpa o campo de entrada após enviar a pergunta
465
+ st.session_state[INPUT_KEY] = ""
466
  except Exception as e:
467
  st.error(f"Erro ao gerar a conclusão do chat: {e}")
468
  else:
469
  st.warning("Por favor, digite uma pergunta antes de enviar.")
470
 
471
+
472
+ # Adicione este código fora do expander para evitar a reexecução do script ao pressionar Enter
473
  if "USER_CHAT_input" in st.session_state and st.session_state.USER_CHAT_input:
474
  st.session_state.USER_CHAT_input = ""