Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 import pandas as pd
+from transformers import AutoTokenizer, AutoModelForTableQuestionAnswering
 from datetime import datetime
 import requests
-from transformers import AutoTokenizer, AutoModelForTableQuestionAnswering
 
 class InterestCalculatorApp:
     def __init__(self):
@@ -115,15 +115,29 @@ def main():
 
     app = InterestCalculatorApp()
 
+    app.download_boe_rates()  # Download base rates first
+
     file_path = st.file_uploader("Upload Invoices File", type=["xlsx"])
 
     if file_path is not None:
         app.load_invoices(file_path)
 
+    query = st.text_input("Enter your query:")
+    if query:
+        # Assuming you have a DataFrame named 'invoices_df' containing the invoice data
+        if not app.invoices_df.empty:
+            # Display the invoice data
+            st.write("Invoice Data:")
+            st.write(app.invoices_df)
+
+            # Call TAPAS model to answer user's query
+            answer = app.ask_tapas(query, app.invoices_df)
+            st.write("Answer:", answer)
+        else:
+            st.warning("Please upload the invoices file first.")
+
     if st.button("Calculate Interest"):
         app.calculate_interest()
 
-    app.download_boe_rates()
-
 if __name__ == "__main__":
     main()