Update app.py

app.py

@@ -70,19 +70,21 @@ def main():
             add_pdf_text_to_db(collection, pdf_text)
             st.success("PDF text has been added to the database. You can now query it!")
             query = st.text_input("Enter your query about the PDF:")
-            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 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)}")
+
 
 if __name__ == "__main__":
     main()