Update app.py

app.py

@@ -1,16 +1,16 @@
 import streamlit as st
 import pandas as pd
 import torch
-from transformers import pipeline
-from transformers import TapasTokenizer, TapexTokenizer, BartForConditionalGeneration
+from transformers import TapexTokenizer, BartForConditionalGeneration
 import datetime
 
-#df = pd.read_excel('discrepantes.xlsx', index_col='Unnamed: 0')
+# Load the data
 df = pd.read_excel('discrepantes.xlsx')
 df.fillna(0, inplace=True)
 table_data = df.astype(str)
 print(table_data.head())
 
+# Function to generate a response using the TAPEX model
 def response(user_question, table_data):
     a = datetime.datetime.now()
 
@@ -44,7 +44,6 @@ def response(user_question, table_data):
 
     return query_result
 
-
 # Streamlit interface
 st.markdown("""
 <div style='display: flex; align-items: center;'>
@@ -63,16 +62,16 @@ if 'history' not in st.session_state:
 user_question = st.text_input("Escreva sua questão aqui:", "")
 
 if user_question:
-    # Add person emoji when typing question
+    # Add human emoji when user asks a question
     st.session_state['history'].append(('👤', user_question))
-    st.markdown(f"🤖 {user_question}")
+    st.markdown(f"**👤 {user_question}**")
     
     # Generate the response
-    bot_response = response(user_question, table_data)
+    bot_response = response(user_question, table_data)["Resposta"]
     
     # Add robot emoji when generating response and align to the right
-    st.session_state['history'].append(('👤', bot_response))
-    st.markdown(f"<div style='text-align: right'>🤖 {bot_response}</div>", unsafe_allow_html=True)
+    st.session_state['history'].append(('🤖', bot_response))
+    st.markdown(f"<div style='text-align: right'>**🤖 {bot_response}**</div>", unsafe_allow_html=True)
 
 # Clear history button
 if st.button("Limpar"):
@@ -81,6 +80,6 @@ if st.button("Limpar"):
 # Display chat history
 for sender, message in st.session_state['history']:
     if sender == '👤':
-        st.markdown(f"🤖 {message}")
-    elif sender == '👤':
-        st.markdown(f"<div style='text-align: right'>🤖 {message}</div>", unsafe_allow_html=True)
+        st.markdown(f"**👤 {message}**")
+    elif sender == '🤖':
+        st.markdown(f"<div style='text-align: right'>**🤖 {message}**</div>", unsafe_allow_html=True)