dnj0 commited on
Commit
a8b1b7a
·
verified ·
1 Parent(s): 4f6cdd1

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +9 -0
src/app.py CHANGED
@@ -135,6 +135,15 @@ def main():
135
  # Main area
136
  st.header("📤 Upload PDF Document")
137
  col1, col2 = st.columns([2, 1])
 
 
 
 
 
 
 
 
 
138
 
139
  with col1:
140
  uploaded_file = st.file_uploader(
 
135
  # Main area
136
  st.header("📤 Upload PDF Document")
137
  col1, col2 = st.columns([2, 1])
138
+
139
+ if st.button("⚙️ Summarize & Store Each Component"):
140
+ results = enhanced_rag.process_and_store_document(
141
+ text=st.session_state.current_text,
142
+ images=st.session_state.current_images,
143
+ tables=st.session_state.current_tables,
144
+ vector_store=vector_store,
145
+ doc_id=st.session_state.current_document
146
+ )
147
 
148
  with col1:
149
  uploaded_file = st.file_uploader(