Update app.py

app.py

@@ -357,7 +357,7 @@ MODEL_OPTIONS = [
 ]
 
 with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
-    # We no longer need vdb_state – we keep only session_state
+    # We only need a single State to track whether extraction has happened
     session_state = gr.State({})
 
     # ─── Welcome Screen ─────────────────────────────────────────────
@@ -407,9 +407,17 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                     label="Vision-Language Model"
                 )
                 extract_btn = gr.Button("Extract")
-                preview_text = gr.Textbox(lines=10, label="Sample Text", interactive=False)
-                preview_img  = gr.Gallery(label="Sample Images", rows=2, value=[])
-                preview_html = gr.HTML()  # for the “Done!” message
+                preview_text = gr.Textbox(
+                    lines=10,
+                    label="Sample Text",
+                    interactive=False
+                )
+                preview_img = gr.Gallery(
+                    label="Sample Images",
+                    rows=2,
+                    value=[]
+                )
+                preview_html = gr.HTML()
 
                 extract_btn.click(
                     fn=extract_data_from_pdfs,
@@ -422,10 +430,10 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                         vlm_dd
                     ],
                     outputs=[
-                        session_state,  # session “processed” flag
-                        preview_text,   # preview text
-                        preview_img,    # preview images
-                        preview_html    # done HTML
+                        session_state,   # updates processed flag
+                        preview_text,    # shows sample text
+                        preview_img,     # shows sample images
+                        preview_html     # shows “Done!” message
                     ]
                 )
 
@@ -434,7 +442,7 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                 with gr.Row():
                     with gr.Column(scale=3):
                         chat = gr.Chatbot(type="messages", label="Chat")
-                        msg  = gr.Textbox(
+                        msg = gr.Textbox(
                             placeholder="Ask about your PDF...",
                             label="Your question"
                         )
@@ -447,13 +455,13 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                         )
                         num_ctx = gr.Slider(1, 20, value=3, label="Text Contexts")
                         img_ctx = gr.Slider(1, 10, value=2, label="Image Contexts")
-                        temp    = gr.Slider(0.1, 1.0, step=0.1, value=0.4, label="Temperature")
+                        temp = gr.Slider(0.1, 1.0, step=0.1, value=0.4, label="Temperature")
                         max_tok = gr.Slider(10, 1000, step=10, value=200, label="Max Tokens")
 
                 send.click(
                     fn=conversation,
                     inputs=[
-                        session_state,  # now drives conversation
+                        session_state,  # drives conversation
                         msg,
                         num_ctx,
                         img_ctx,
@@ -464,7 +472,7 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                     ],
                     outputs=[
                         chat,
-                        gr.Dataframe(),   # returned docs
+                        gr.Dataframe(),  # returns the retrieved docs
                         gr.Gallery(label="Relevant Images", rows=2, value=[])
                     ]
                 )