Update app.py
Browse files
app.py
CHANGED
|
@@ -14,6 +14,7 @@ load_dotenv()
|
|
| 14 |
os.getenv("GOOGLE_API_KEY")
|
| 15 |
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
| 16 |
|
|
|
|
| 17 |
# read all pdf files and return text
|
| 18 |
|
| 19 |
|
|
@@ -98,6 +99,10 @@ def main():
|
|
| 98 |
"Upload your PDF Files and Click on the Submit & Process Button", accept_multiple_files=True)
|
| 99 |
if st.button("Submit & Process"):
|
| 100 |
with st.spinner("Processing..."):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 101 |
raw_text = get_pdf_text(pdf_docs)
|
| 102 |
text_chunks = get_text_chunks(raw_text)
|
| 103 |
get_vector_store(text_chunks)
|
|
|
|
| 14 |
os.getenv("GOOGLE_API_KEY")
|
| 15 |
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
| 16 |
|
| 17 |
+
DEFAULT_PDF_FILES = ["2024_25_Annex_Budget.pdf", "2024_25_Budget_Speech.pdf"] # Define default PDFs
|
| 18 |
# read all pdf files and return text
|
| 19 |
|
| 20 |
|
|
|
|
| 99 |
"Upload your PDF Files and Click on the Submit & Process Button", accept_multiple_files=True)
|
| 100 |
if st.button("Submit & Process"):
|
| 101 |
with st.spinner("Processing..."):
|
| 102 |
+
for file_name in pdf_docs:
|
| 103 |
+
if not os.path.exists(file_name):
|
| 104 |
+
st.error(f"Default file '{file_name}' not found!")
|
| 105 |
+
return # Exit if a default file is missing
|
| 106 |
raw_text = get_pdf_text(pdf_docs)
|
| 107 |
text_chunks = get_text_chunks(raw_text)
|
| 108 |
get_vector_store(text_chunks)
|