Update app.py

app.py

@@ -359,7 +359,7 @@ MODEL_OPTIONS = [
 ]
 
 with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
-    # State to track extraction
+    # State to track that extraction completed (and carry any metadata)
     session_state = gr.State({})
 
     # ─── Welcome Screen ─────────────────────────────────────────────
@@ -374,6 +374,9 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
     with gr.Column(visible=False) as app_col:
         gr.Markdown("## 📚 Multimodal Chat-PDF Playground")
 
+        # We need to capture the extract‐event so we can chain the “show chat tab” later
+        extract_event = None
+
         with gr.Tabs() as tabs:
             # ── Tab 1: Upload & Extract ───────────────────────────────
             with gr.TabItem("1. Upload & Extract"):
@@ -418,9 +421,8 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                 )
                 preview_html = gr.HTML()
 
-                # 1) Run extraction
-                # 2) When done, reveal the Chat tab
-                extract_btn.click(
+                # Kick off extraction and capture the event
+                extract_event = extract_btn.click(
                     fn=extract_data_from_pdfs,
                     inputs=[
                         docs,
@@ -431,23 +433,19 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                         vlm_dd
                     ],
                     outputs=[
-                        session_state,
-                        preview_text,
-                        preview_img,
-                        preview_html
+                        session_state,   # sets session["processed"]=True
+                        preview_text,    # shows first bits of text
+                        preview_img,     # shows first images
+                        preview_html     # shows “<h3>Done!</h3>”
                     ]
-                ).then(
-                    fn=lambda: gr.update(visible=True),
-                    inputs=[],
-                    outputs=[tabs.select("2. Chat")]
                 )
 
-            # ── Tab 2: Chat (starts hidden) ────────────────────────────
+            # ── Tab 2: Chat (initially hidden) ──────────────────────────
             with gr.TabItem("2. Chat", visible=False) as chat_tab:
                 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"
                         )
@@ -477,11 +475,18 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                     ],
                     outputs=[
                         chat,
-                        gr.Dataframe(),
+                        gr.Dataframe(),   # shows retrieved text chunks
                         gr.Gallery(label="Relevant Images", rows=2, value=[])
                     ]
                 )
 
+        # After both tabs are defined, chain the “unhide chat tab” event
+        extract_event.then(
+            fn=lambda: gr.update(visible=True),
+            inputs=[],
+            outputs=[chat_tab]
+        )
+
         gr.HTML("<center>Made with ❤️ by Zamal</center>")
 
     # ─── Wire the Start button ───────────────────────────────────────
@@ -493,3 +498,4 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
 
 if __name__ == "__main__":
     demo.launch()
+