Update app.py

app.py

@@ -516,8 +516,8 @@ with gr.Blocks(css=CSS, title="Document Analysis System") as demo:
     def hide_spinner():
         return gr.update(visible=False)
 
+    # In update_display()
     def update_display(results):
-        """Update all display components with processed results"""
         if not results:
             return [
                 gr.update(value='<div class="no-preview-message">No file to display</div>'),
@@ -525,24 +525,19 @@ with gr.Blocks(css=CSS, title="Document Analysis System") as demo:
                 gr.update(value="No content"),
                 gr.update(value=[])
             ]
-        
-        # PDF preview
-        pdf_preview = results.get("pdf_preview", '<div class="no-preview-message">Preview generation failed</div>')
-        
-        # Image gallery
+    
         images = results.get("pdf_images", [])
-        show_gallery = bool(images) and results.get("file_type") == "pdf"
-        
-        # Markdown content
+        show_gallery = bool(images)
         display_content = results["markdown_contents"][0] if results.get("markdown_contents") else "No content"
-        
+    
         return [
-            gr.update(value=pdf_preview),
+            gr.update(value='<div class="no-preview-message">Preview rendered as images</div>'),
             gr.update(visible=show_gallery),
             gr.update(value=display_content),
             gr.update(value=images if show_gallery else [])
         ]
 
+
     process_btn.click(
         toggle_spinner,
         outputs=[loading_spinner]