Update app.py

app.py

@@ -174,29 +174,65 @@ def download_pdf_from_url(url):
 # -----------------------------
 # Display PDF
 # -----------------------------
+# def display_pdf(pdf_url):
+#     if not pdf_url:
+#         return gr.update(value=None, visible=False)
+    
+#     try:
+#         logger.info(f"Displaying PDF from: {pdf_url}")
+#         pdf_bytes = download_pdf_from_url(pdf_url)
+#         doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+#         page = doc[0]
+        
+#         zoom = 5
+#         mat = fitz.Matrix(zoom, zoom)
+#         pix = page.get_pixmap(matrix=mat)
+        
+#         img_data = pix.tobytes("png")
+#         img = Image.open(io.BytesIO(img_data))
+#         doc.close()
+        
+#         logger.info("PDF displayed successfully")
+#         return gr.update(value=img, visible=True)
+#     except Exception as e:
+#         logger.error(f"Error displaying PDF: {str(e)}")
+#         return gr.update(value=None, visible=False)
+
 def display_pdf(pdf_url):
     if not pdf_url:
-        return gr.update(value=None, visible=False)
-    
+        return (
+            gr.update(value=None, visible=False),
+            gr.update(value="", visible=False)
+        )
+
     try:
-        logger.info(f"Displaying PDF from: {pdf_url}")
+        # Show loading text
+        yield (
+            gr.update(value=None, visible=False),
+            gr.update(value="📄 Loading PDF preview...", visible=True)
+        )
+
         pdf_bytes = download_pdf_from_url(pdf_url)
         doc = fitz.open(stream=pdf_bytes, filetype="pdf")
         page = doc[0]
-        
-        zoom = 5
-        mat = fitz.Matrix(zoom, zoom)
+
+        mat = fitz.Matrix(5, 5)
         pix = page.get_pixmap(matrix=mat)
-        
-        img_data = pix.tobytes("png")
-        img = Image.open(io.BytesIO(img_data))
+
+        img = Image.open(io.BytesIO(pix.tobytes("png")))
         doc.close()
-        
-        logger.info("PDF displayed successfully")
-        return gr.update(value=img, visible=True)
+
+        yield (
+            gr.update(value=img, visible=True),
+            gr.update(value="", visible=False)
+        )
+
     except Exception as e:
-        logger.error(f"Error displaying PDF: {str(e)}")
-        return gr.update(value=None, visible=False)
+        logger.error(f"PDF error: {e}")
+        yield (
+            gr.update(value=None, visible=False),
+            gr.update(value="❌ Failed to load PDF", visible=True)
+        )
 
 # -----------------------------
 # UI Wrapper for Blocks
@@ -322,6 +358,7 @@ with gr.Blocks(
         with gr.Column(scale=3):
 
             chatbot = gr.Chatbot(label="💬 Conversation", elem_classes="chatbot")
+            status_text = gr.Markdown("")
 
             question = gr.Textbox(
                 label="Ask Dox a question:",
@@ -348,7 +385,7 @@ with gr.Blocks(
             gr.Markdown("### 📄 Referenced PDF Document")
 
             link_state = gr.State()
-
+            pdf_status = gr.Markdown(visible=False)
             output_image = gr.Image(
                 label="Preview (Page 1)",
                 visible=False
@@ -361,6 +398,15 @@ with gr.Blocks(
         if chat_history is None:
             chat_history = []
     
+        # Show thinking immediately
+        yield (
+            chat_history,
+            None,
+            gr.update(value=None, visible=False),
+            "🤔 Thinking...",
+            ""
+        )
+    
         response_text, link = ask_agent(user_message)
     
         chat_history.append({
@@ -373,31 +419,32 @@ with gr.Blocks(
             "content": response_text
         })
     
-        return (
+        yield (
             chat_history,
             link,
-            gr.update(value=None, visible=False)
+            gr.update(value=None, visible=False),
+            "",  # clear thinking
+            ""
         )
 
     ask_btn.click(
         chat_ui,
         inputs=[question, chatbot],
-        outputs=[chatbot, link_state, output_image]
+        outputs=[chatbot, link_state, output_image, status_text, question]
     ).then(
         display_pdf,
         inputs=link_state,
-        outputs=output_image
+        outputs=[output_image, pdf_status]
     )
-
-    # Enter key submit
+    
     question.submit(
         chat_ui,
         inputs=[question, chatbot],
-        outputs=[chatbot, link_state, output_image]
+        outputs=[chatbot, link_state, output_image, status_text, question]
     ).then(
         display_pdf,
         inputs=link_state,
-        outputs=output_image
+        outputs=[output_image, pdf_status]
     )
 
 if __name__ == "__main__":