DrishtiSharma commited on
Commit
006ee88
Β·
verified Β·
1 Parent(s): 15ed0e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,7 +21,7 @@ st.title("Blah-1")
21
  os.environ["GROQ_API_KEY"] = st.secrets.get("GROQ_API_KEY", "")
22
 
23
  # ----------------- ChromaDB Persistent Directory -----------------
24
- CHROMA_DB_DIR = "/mnt/data/chroma_db" # Hugging Face Spaces persistent storage
25
  os.makedirs(CHROMA_DB_DIR, exist_ok=True)
26
 
27
  # ----------------- Initialize Session State -----------------
@@ -101,6 +101,7 @@ if not st.session_state.pdf_loaded and "pdf_path" in st.session_state:
101
  with st.spinner("πŸ”„ Processing document... Please wait."):
102
  loader = PDFPlumberLoader(st.session_state.pdf_path)
103
  docs = loader.load()
 
104
 
105
  # Extract metadata
106
  title, author, email_str, affiliation_str = extract_metadata(st.session_state.pdf_path)
 
21
  os.environ["GROQ_API_KEY"] = st.secrets.get("GROQ_API_KEY", "")
22
 
23
  # ----------------- ChromaDB Persistent Directory -----------------
24
+ CHROMA_DB_DIR = "/mnt/data/chroma_db"
25
  os.makedirs(CHROMA_DB_DIR, exist_ok=True)
26
 
27
  # ----------------- Initialize Session State -----------------
 
101
  with st.spinner("πŸ”„ Processing document... Please wait."):
102
  loader = PDFPlumberLoader(st.session_state.pdf_path)
103
  docs = loader.load()
104
+ st.json(docs[0].metadata)
105
 
106
  # Extract metadata
107
  title, author, email_str, affiliation_str = extract_metadata(st.session_state.pdf_path)