Update st.py

st.py

@@ -118,6 +118,20 @@ def main():
     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()
 
@@ -125,3 +139,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+