Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import streamlit as st
 from llama_parse import LlamaParse
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -8,13 +9,9 @@ from langchain.prompts import PromptTemplate
 from langchain.chains import RetrievalQA
 from langchain_groq import ChatGroq
 import joblib
-import os
 import tempfile
 
-# Streamlit specific setup
-st.title("Uber Quarterly Report QA")
-
-# Access API keys from secrets
+# API keys
 llama_cloud_api_key = "llx-rVenNfvEyWTTZ2bOJIY7zymr6oyyucfdBusq407A6RzZhMKb"
 groq_api_key = "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"
 
@@ -37,6 +34,9 @@ uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")
 if uploaded_file is not None:
     llama_parse_documents = load_or_parse_data(uploaded_file)
 
+    # Create data directory if it doesn't exist
+    os.makedirs("data", exist_ok=True)
+
     # Further processing of the parsed data...
     # Further processing of the parsed data
     with open('data/output.md', 'a') as f: