Update app.py

app.py

@@ -117,8 +117,23 @@ def main():
             pdf_text = extract_text_from_pdf(uploaded_file)
             st.success("Text extracted successfully!")
             st.text_area("Extracted Text:", pdf_text, height=300)
+            add_pdf_text_to_db(collection, pdf_text)
+            st.success("PDF text has been added to the database. You can now query it!")
+            if query:
+                try:
+                    answer, metadata = query_pdf_data(collection, query, retriever_model)
+                    st.subheader("Answer:")
+                    st.write(answer[0]['generated_text'])
+                    st.subheader("Retrieved Context:")
+                    for meta in metadata[0]:
+                        st.write(meta)
+                except Exception as e:
+                    st.error(f"An error occurred: {str(e)}")
         except Exception as e:
             st.error(f"Error extracting text: {e}")
+
+
+            
     # if uploaded_file:
     #     st.write("Extracting text and populating the database...")
     #     pdf_text = extract_text_from_pdf(uploaded_file)