Update app.py

app.py

@@ -8,7 +8,7 @@ from qdrant_client import models
 from utils import setup_openai_embeddings, setup_qdrant_client, delete_collection, is_document_embedded
 from embed import embed_documents_into_qdrant
 from preprocess import split_documents, update_metadata, load_documents_OCR
-from retrieve import retrieve_documents, retrieve_documents_from_collection
+from retrieve import  retrieve_documents_from_collection
 from summarize import summarize_documents
 
 # Load environment variables
@@ -41,7 +41,7 @@ def main():
 
     pages = {
         "Lex Document Summarization": page_summarization,
-        "Chat with Data Bank": page_qna,
+        "Chat with RSCA": page_qna,
         "Chat with Uploaded Docs": page_chat_with_uploaded_docs,
         "Chat with VOO": page_chat_with_voo
     }
@@ -90,8 +90,8 @@ def page_summarization(uploaded_files):
 
 def page_qna(uploaded_files):
     """Page for Q&A functionality."""
-    st.title("Chat with Data Bank")
-    user_query = st.text_area("Enter your question here:", height=300)
+    st.title("Chat with RSCA")
+    user_query = st.text_area("Enter your question here:", height=200)
     if st.button('Get Answer'):
         if user_query:
             answer = handle_query(user_query)
@@ -106,7 +106,7 @@ def page_qna(uploaded_files):
 def page_chat_with_uploaded_docs(uploaded_files):
     """Page for chatting with uploaded documents."""
     st.title("Chat with Uploaded Documents")
-    user_query = st.text_area("Enter your question here:", height=300)
+    user_query = st.text_area("Enter your question here:", height=200)
     if st.button('Get Answer'):
         if user_query:
             answer = handle_uploaded_docs_query(user_query, st.session_state['uploaded_collection_name'])
@@ -128,7 +128,7 @@ def page_chat_with_uploaded_docs(uploaded_files):
 def page_chat_with_voo(uploaded_files):
     """Page for chatting with VOO documents."""
     st.title("Chat with VOO")
-    user_query = st.text_area("Enter your question here:", height=300)
+    user_query = st.text_area("Enter your question here:", height=200)
     if st.button('Get Answer'):
         if user_query:
             answer = handle_voo_query(user_query)
@@ -191,7 +191,7 @@ def add_docs_to_current_chat(files_info):
 def handle_query(query):
     """Retrieve answers based on the query."""
     try:
-        answer = retrieve_documents(query, os.getenv('OPENAI_API_KEY'), os.getenv('QDRANT_URL'), os.getenv('QDRANT_API_KEY'))
+        answer = retrieve_documents_from_collection(query, os.getenv('OPENAI_API_KEY'), os.getenv('QDRANT_URL'), os.getenv('QDRANT_API_KEY'),'Lex-v1')
         return answer or "No relevant answer found."
     except Exception as e:
         return f"Error processing the query: {str(e)}"