DrishtiSharma commited on
Commit
97424be
Β·
verified Β·
1 Parent(s): 2fe7d8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -133,12 +133,15 @@ if not st.session_state.pdf_loaded and "pdf_path" in st.session_state:
133
  metadata = extract_metadata_llm(st.session_state.pdf_path)
134
 
135
  # Display extracted-metadata
136
- st.subheader("πŸ“„ Extracted Document Metadata")
137
- st.subheader("πŸ“„ Extracted Document Metadata")
138
- st.write(f"**Title:** {metadata.get('Title', 'Unknown')}")
139
- st.write(f"**Author:** {metadata.get('Author', 'Unknown')}")
140
- st.write(f"**Emails:** {metadata.get('Emails', 'No emails found')}")
141
- st.write(f"**Affiliations:** {metadata.get('Affiliations', 'No affiliations found')}")
 
 
 
142
 
143
  # Embedding Model
144
  model_name = "nomic-ai/modernbert-embed-base"
 
133
  metadata = extract_metadata_llm(st.session_state.pdf_path)
134
 
135
  # Display extracted-metadata
136
+ if isinstance(metadata, dict): # Ensure metadata is a dictionary
137
+ st.subheader("πŸ“„ Extracted Document Metadata")
138
+ st.write(f"**Title:** {metadata.get('Title', 'Unknown')}")
139
+ st.write(f"**Author:** {metadata.get('Author', 'Unknown')}")
140
+ st.write(f"**Emails:** {metadata.get('Emails', 'No emails found')}")
141
+ st.write(f"**Affiliations:** {metadata.get('Affiliations', 'No affiliations found')}")
142
+ else:
143
+ st.error("Metadata extraction failed. Check the LLM response format.")
144
+
145
 
146
  # Embedding Model
147
  model_name = "nomic-ai/modernbert-embed-base"