Update app.py

app.py

@@ -384,6 +384,7 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
+# Streamlit interface
 # Streamlit interface
 st.title("📄 PDF Reading Assistant")
 st.markdown("### Extract tables, figures, summaries, and answers from your PDF files easily.")
@@ -391,6 +392,14 @@ st.markdown("### Extract tables, figures, summaries, and answers from your PDF f
 uploaded_file = st.file_uploader("Upload a PDF", type="pdf")
 if uploaded_file:
     file_path = save_uploaded_file(uploaded_file)
+
+    if 'figures' not in st.session_state:
+        st.session_state['figures'] = None
+    if 'tables' not in st.session_state:
+        st.session_state['tables'] = None
+    if 'summary' not in st.session_state:
+        st.session_state['summary'] = None
+
     with st.container():
         st.markdown("<div class='section-header'>Extract Tables and Figures</div>", unsafe_allow_html=True)
         with st.expander("Click to Extract Tables and Figures", expanded=True):
@@ -398,21 +407,21 @@ if uploaded_file:
                 extract_button = st.button("Extract")
                 if extract_button:
                     figures, tables = process_pdf(file_path)
+                    st.session_state['figures'] = figures
+                    st.session_state['tables'] = tables
+
+                if st.session_state['figures']:
                     col1, col2 = st.columns(2)
                     with col1:
                         st.write("### Figures")
-                        if figures:
-                            for figure in figures:
-                                st.image(figure, use_column_width=True)
-                        else:
-                            st.write("No figures found.")
+                        for figure in st.session_state['figures']:
+                            st.image(figure, use_column_width=True)
                     with col2:
                         st.write("### Tables")
-                        if tables:
-                            for table in tables:
-                                st.image(table, use_column_width=True)
-                        else:
-                            st.write("No tables found.")
+                        for table in st.session_state['tables']:
+                            st.image(table, use_column_width=True)
+                else:
+                    st.write("No figures or tables found.")
     
     with st.container():
         st.markdown("<div class='section-header'>Get Summary</div>", unsafe_allow_html=True)
@@ -421,7 +430,10 @@ if uploaded_file:
                 summary_button = st.button("Generate Summary")
                 if summary_button:
                     summary = summarize_pdf(file_path)
-                    st.markdown(summary, unsafe_allow_html=True)
+                    st.session_state['summary'] = summary
+
+                if st.session_state['summary']:
+                    st.markdown(st.session_state['summary'], unsafe_allow_html=True)
 
     with st.container():
         st.markdown("<div class='section-header'>Chat with your PDF</div>", unsafe_allow_html=True)