Update app.py

app.py

@@ -18,7 +18,7 @@ from googleapiclient.discovery import build
 from googleapiclient.http import MediaIoBaseDownload
 from google.oauth2 import service_account
 import tempfile
-import os 
+import os
 
 
 # SQLite Database Functions (database.py)
@@ -205,11 +205,6 @@ def main():
         create_tables(conn)
     else:
         st.error("Error! Cannot create the database connection.")
-# ... (other imports and functions) ...
-
-# Streamlit App Interface (app.py)
-def main():
-    # ... (other code) ...
 
     # Dashboard Overview Tab
     st.sidebar.markdown("<h2 style='color: #1E3A8A;'>Dashboard Overview</h2>", unsafe_allow_html=True)
@@ -266,12 +261,56 @@ def main():
                             # Initialize QA System for Selected Documents
                             qa_system = initialize_qa_system(vector_store)
                             if qa_system:
-                                # ... (rest of your query processing code) ...
+                                 # Query Input
+                                user_query = st.text_input("Enter your query about the RFPs:", placeholder="e.g., What are the evaluation criteria?", label_visibility='visible')
+                                if user_query:
+                                    st.markdown("<p style='color: #1E3A8A;'>Retrieving answer...</p>", unsafe_allow_html=True)
+                                    try:
+                                        response, source_documents = qa_system.run(user_query, return_source_documents=True)
+                                        st.markdown("<h4 style='color: #1E3A8A;'>Answer:</h4>", unsafe_allow_html=True)
+                                        st.write(response)
+
+                                        # Store Query and Response in Database
+                                        with conn:
+                                            for doc in source_documents:
+                                                source_name = doc.metadata["source"]
+                                                document_id = conn.execute("SELECT id FROM documents WHERE name = ?", (source_name,)).fetchone()
+                                                if document_id:
+                                                    conn.execute("INSERT INTO queries (query, response, document_id) VALUES (?, ?, ?)", (user_query, response, document_id[0]))
+
+                                        # Display Source Information
+                                        st.markdown("<h4 style='color: #1E3A8A;'>Sources:</h4>", unsafe_allow_html=True)
+                                        for doc in source_documents:
+                                            source_name = doc.metadata["source"]
+                                            matched_text = doc.page_content
+                                            st.write(f"- Source Document: {source_name}")
+                                            # Display the matching text with highlighting
+                                            for idx, page_content in enumerate(document_pages[document_names.index(source_name)]):
+                                                if matched_text in page_content:
+                                                    highlighted_content = re.sub(re.escape(matched_text), f"<mark>{matched_text}</mark>", page_content)
+                                                    st.write(f"  - Page {idx + 1}: {highlighted_content}")
 
+                                    except Exception as e:
+                                        st.error(f"Error generating response: {e}")
         except Exception as e:
             st.error(f"Error retrieving documents from database: {e}")
 
-    # ... (rest of your code) ...
+    # Document Upload Section
+    st.markdown("<h2 style='color: #1E3A8A;'>Upload RFP Documents</h2>", unsafe_allow_html=True)
+    uploaded_documents = st.file_uploader("Upload PDF documents", type="pdf", accept_multiple_files=True)
+    if uploaded_documents:
+        st.write(f"Uploaded {len(uploaded_documents)} documents.")
+        all_texts, document_names, document_pages = upload_and_parse_documents(uploaded_documents)
+        if all_texts:
+            # Store Documents in Database
+            if conn is not None:
+                try:
+                    with conn:
+                        for doc, doc_name in zip(all_texts, document_names):
+                            conn.execute("INSERT INTO documents (name, content) VALUES (?, ?)", (doc_name, doc))
+                    st.success("Documents uploaded and parsed successfully.", icon="✅")
+                except Exception as e:
+                    st.error(f"Error saving documents to database: {e}")
 
     # URL Input Section
     st.markdown("<h2 style='color: #1E3A8A;'>Or Provide a URL</h2>", unsafe_allow_html=True)