app updated

app.py

@@ -1,3 +1,4 @@
+import math
 import os
 
 import openai
@@ -7,8 +8,8 @@ from openai import OpenAI
 
 from helper.utils import *
 
-st.set_page_config(layout="centered", page_title="Document Search🤖📖")
-st.header("Document Search🤖📖")
+st.set_page_config(layout="wide", page_title="Document Search using QIM🤖📖")
+st.header("Document Search using Quantized Influence Measure (QIM)🤖📖")
 st.write("---")
 
 
@@ -19,7 +20,7 @@ with st.sidebar:
         # Display the instruction manual for the Document Data Chatbot in a formatted markdown
         st.markdown(
             """
-            # Document Data Chatbot User Manual 🤖💊
+            # Document Data Chatbot User Manual 🤖📖
             
             Welcome to the Document Data Chatbot, your interactive assistant for information on the textual "Document Data". This chatbot offers quick and accurate responses to your queries. Follow these steps to interact with the chatbot:
 
@@ -56,6 +57,9 @@ with st.sidebar:
         "Upload documents", accept_multiple_files=True, type=["txt", "pdf"]
     )
 
+    # Input filter
+    top_n = st.number_input("Insert a number (top n rows to be selected):")
+
     # Clear button
     clear_button = st.sidebar.button("Clear Conversation", key="clear")
 
@@ -87,7 +91,7 @@ if uploaded_files is None:
 
 elif uploaded_files:
     # Inform the user how many documents have been loaded
-    st.sidebar.write(f"{len(uploaded_files)} document(s) loaded..")
+    st.sidebar.success(f"{len(uploaded_files)} document(s) loaded...")
 
     # Process the uploaded files to extract text and source information
     textify_output = read_and_textify(uploaded_files)
@@ -102,6 +106,7 @@ elif uploaded_files:
     refs_tab = query_search(
         "pful for understanding federal income", documents, query_database, sources
     )
+    refs_tab.head(math.ceil(top_n))
 
     # React to user input
     if prompt := st.chat_input("What is up?"):