Update app.py

app.py

@@ -13,14 +13,17 @@ data = {
     'Cidade': ['Nova York', 'Los Angeles', 'Chicago'],
     'Feedback': [None, None, None]
 }
-df = pd.DataFrame(data)
+#df = pd.DataFrame(data)
+
+df = pd.read_csv('anomalies.csv')
+df['Feedback'] = None
 
 # Função para responder perguntas com GPT-2
 def answer_question_with_gpt(question):
     if tokenizer.pad_token is None:
         tokenizer.pad_token = tokenizer.eos_token
 
-    prompt = f"Considerando os dados: {df.to_string(index=False)}. Pergunta: {question} Resposta:"
+    prompt = f"Considerando os dados, em que 'ds' é datetime, 'real' é valor monetário e 'Group' é o grupo ao qual pretence o valor: {df.to_string(index=False)}. Pergunta: {question} Resposta:"
     inputs = tokenizer(prompt, return_tensors='pt', padding='max_length', truncation=True, max_length=512)
     attention_mask = inputs['attention_mask']
     input_ids = inputs['input_ids']
@@ -44,11 +47,11 @@ def answer_question_with_gpt(question):
 # Função para adicionar feedback
 def add_feedback(nome, feedback):
     global df
-    if nome in df['Nome'].values:
-        df.loc[df['Nome'] == nome, 'Feedback'] = feedback
+    if grupo in df['Group'].values:
+        df.loc[df['Group'] == grupo, 'Feedback'] = feedback
         return "Feedback adicionado com sucesso."
     else:
-        return "Nome não encontrado no DataFrame."
+        return "Dado não encontrado no DataFrame."
 
 with gr.Blocks() as demo:
     gr.Markdown("# Sistema de Consulta e Feedback de Dados")