Update app.py

app.py

@@ -2,12 +2,12 @@ import streamlit as st
 import pandas as pd
 from transformers import pipeline
 
-# Load the anomalies data
+# Load the anomalies data and convert all cells to strings
 df = pd.read_csv('anomalies.csv', sep=',', decimal='.')
+df = df.applymap(str)
+print(df.head())
 
-# Function to generate a response
 def response(user_question):
-    # Verifica se a pergunta é uma string
     if not isinstance(user_question, str):
         raise TypeError(f"Esperado uma string para a pergunta, mas recebeu {type(user_question)}")
     
@@ -18,7 +18,7 @@ def response(user_question):
     resposta = tqa(table=df, query=user_question)
     
     # Verifica se alguma célula foi retornada
-    if len(resposta['cells']) == 0:
+    if 'cells' not in resposta or len(resposta['cells']) == 0:
         raise IndexError("Nenhuma célula foi retornada pelo modelo.")
     
     # Obtém a primeira célula da resposta
@@ -26,7 +26,6 @@ def response(user_question):
     
     return final_response
 
-
 # Streamlit interface
 st.markdown("""
 <div style='display: flex; align-items: center;'>