Update app.py

app.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 """
 Turbo Air Viewer - Equipment Specification Database Viewer
-Deployment-ready version with automatic database download and inline PDF display
+Enhanced version with product image extraction and display
 """
 
 import streamlit as st
@@ -17,6 +17,9 @@ import io
 import os
 import requests
 from urllib.parse import quote
+from PIL import Image
+import fitz  # PyMuPDF
+import tempfile
 
 # Streamlit page config MUST be first
 st.set_page_config(
@@ -27,7 +30,6 @@ st.set_page_config(
 
 # Configuration
 DB_FILENAME = "turbo_air_db.sqlite"
-# Use the exact URL you provided
 DB_URL = "https://huggingface.co/spaces/TurboAir/turbo-air-viewer/resolve/main/turbo_air_db.sqlite"
 
 def download_database():
@@ -153,7 +155,6 @@ if DB_PATH is None:
     st.error("❌ Unable to load database. Please refresh the page to try again.")
     st.stop()
 
-# Rest of your original code continues here...
 # Product type mappings
 PRODUCT_TYPES = {
     'TSR': 'Reach-In Refrigerators',
@@ -269,6 +270,20 @@ st.markdown("""
         background-color: #FF5722;
         color: white;
     }
+    
+    .bookmark-item {
+        border: 1px solid #444;
+        border-radius: 8px;
+        padding: 10px;
+        margin-bottom: 10px;
+        background-color: #2d2d2d;
+    }
+    
+    .bookmark-image {
+        border: 1px solid #555;
+        border-radius: 4px;
+        margin-bottom: 8px;
+    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -277,6 +292,74 @@ if 'selected_model' not in st.session_state:
     st.session_state.selected_model = None
 if 'bookmarked_models' not in st.session_state:
     st.session_state.bookmarked_models = []
+if 'text_only_view' not in st.session_state:
+    st.session_state.text_only_view = False
+if 'product_images' not in st.session_state:
+    st.session_state.product_images = {}
+
+def extract_pdf_thumbnail(pdf_url, model_name, max_width=300, max_height=400):
+    """Extract first page of PDF as thumbnail image"""
+    cache_key = f"thumb_{model_name}"
+    
+    # Check if already cached in session state
+    if cache_key in st.session_state.product_images:
+        return st.session_state.product_images[cache_key]
+    
+    try:
+        # Download PDF to temporary file
+        response = requests.get(pdf_url, timeout=30)
+        if response.status_code == 200:
+            with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as tmp_file:
+                tmp_file.write(response.content)
+                tmp_path = tmp_file.name
+            
+            # Open PDF and extract first page
+            pdf_document = fitz.Document(tmp_path)
+            first_page = pdf_document[0]
+            
+            # Render page as image (2x resolution for better quality)
+            mat = fitz.Matrix(2, 2)
+            display_list = first_page.get_displaylist()
+            pix = display_list.get_pixmap(matrix=mat)
+            
+            # Convert to PIL Image
+            img_data = pix.tobytes("png")
+            img = Image.open(io.BytesIO(img_data))
+            
+            # Calculate aspect ratio and resize
+            width, height = img.size
+            aspect_ratio = width / height
+            
+            if width > max_width:
+                new_width = max_width
+                new_height = int(new_width / aspect_ratio)
+            else:
+                new_width = width
+                new_height = height
+            
+            if new_height > max_height:
+                new_height = max_height
+                new_width = int(new_height * aspect_ratio)
+            
+            img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
+            
+            # Convert to base64 for caching
+            buffered = io.BytesIO()
+            img.save(buffered, format="PNG")
+            img_base64 = base64.b64encode(buffered.getvalue()).decode()
+            
+            # Cache in session state
+            st.session_state.product_images[cache_key] = img_base64
+            
+            # Cleanup
+            pdf_document.close()
+            os.unlink(tmp_path)
+            
+            return img_base64
+            
+    except Exception as e:
+        st.warning(f"Could not extract image for {model_name}: {str(e)}")
+        return None
 
 # Cache functions
 @st.cache_data
@@ -629,31 +712,88 @@ if not all_models:
 if st.session_state.bookmarked_models:
     st.markdown("### 📌 Bookmarked Models")
     
-    view_col1, view_col2 = st.columns([2, 1])
+    view_col1, view_col2, view_col3 = st.columns([2, 1, 1])
     with view_col2:
         st.markdown(f"**{len(st.session_state.bookmarked_models)} models selected**")
     
-    # List view
-    display_limit = 5
-    for idx, model in enumerate(st.session_state.bookmarked_models[:display_limit]):
-        col_model, col_remove = st.columns([5, 1])
-        with col_model:
-            st.text(f"• {model}")
-        with col_remove:
-            if st.button("❌", key=f"remove_bookmark_list_{idx}", help=f"Remove {model}"):
-                st.session_state.bookmarked_models.remove(model)
-                st.rerun()
+    with view_col3:
+        # Toggle for text-only view
+        if st.button(
+            "📷 Show Images" if st.session_state.text_only_view else "📝 Text Only",
+            key="toggle_view",
+            use_container_width=True
+        ):
+            st.session_state.text_only_view = not st.session_state.text_only_view
+            st.rerun()
     
-    if len(st.session_state.bookmarked_models) > display_limit:
-        with st.expander(f"Show all {len(st.session_state.bookmarked_models)} bookmarks"):
-            for idx, model in enumerate(st.session_state.bookmarked_models[display_limit:], display_limit):
-                col_model, col_remove = st.columns([5, 1])
-                with col_model:
-                    st.text(f"• {model}")
-                with col_remove:
-                    if st.button("❌", key=f"remove_bookmark_exp_{idx}", help=f"Remove {model}"):
-                        st.session_state.bookmarked_models.remove(model)
-                        st.rerun()
+    # Display bookmarked models with or without images
+    if st.session_state.text_only_view:
+        # Text-only view (original compact list)
+        display_limit = 5
+        for idx, model in enumerate(st.session_state.bookmarked_models[:display_limit]):
+            col_model, col_remove = st.columns([5, 1])
+            with col_model:
+                st.text(f"• {model}")
+            with col_remove:
+                if st.button("❌", key=f"remove_bookmark_list_{idx}", help=f"Remove {model}"):
+                    st.session_state.bookmarked_models.remove(model)
+                    st.rerun()
+        
+        if len(st.session_state.bookmarked_models) > display_limit:
+            with st.expander(f"Show all {len(st.session_state.bookmarked_models)} bookmarks"):
+                for idx, model in enumerate(st.session_state.bookmarked_models[display_limit:], display_limit):
+                    col_model, col_remove = st.columns([5, 1])
+                    with col_model:
+                        st.text(f"• {model}")
+                    with col_remove:
+                        if st.button("❌", key=f"remove_bookmark_exp_{idx}", help=f"Remove {model}"):
+                            st.session_state.bookmarked_models.remove(model)
+                            st.rerun()
+    else:
+        # Image view
+        # Display in grid layout
+        cols_per_row = 4
+        for i in range(0, len(st.session_state.bookmarked_models), cols_per_row):
+            cols = st.columns(cols_per_row)
+            
+            for j, col in enumerate(cols):
+                if i + j < len(st.session_state.bookmarked_models):
+                    model = st.session_state.bookmarked_models[i + j]
+                    model_data = get_model_data(model)
+                    
+                    with col:
+                        # Container for each bookmarked item
+                        st.markdown('<div class="bookmark-item">', unsafe_allow_html=True)
+                        
+                        # Try to get and display thumbnail
+                        if model_data and model_data.get('file_path'):
+                            pdf_filename = model_data['file_path'].replace('\\', '/').split('/')[-1]
+                            pdf_url = f"https://huggingface.co/spaces/TurboAir/TurboAirViewer/resolve/main/pdfs/{pdf_filename}"
+                            
+                            # Show loading spinner while extracting image
+                            with st.spinner(f"Loading image for {model}..."):
+                                img_base64 = extract_pdf_thumbnail(pdf_url, model)
+                            
+                            if img_base64:
+                                st.markdown(
+                                    f'<img src="data:image/png;base64,{img_base64}" '
+                                    f'style="width:100%; max-height:200px; object-fit:contain;" '
+                                    f'class="bookmark-image">',
+                                    unsafe_allow_html=True
+                                )
+                            else:
+                                st.info("📄 No preview available")
+                        
+                        # Model info
+                        st.markdown(f"**{model}**")
+                        st.caption(get_product_type(model))
+                        
+                        # Remove button
+                        if st.button("Remove", key=f"remove_img_{i}_{j}", use_container_width=True):
+                            st.session_state.bookmarked_models.remove(model)
+                            st.rerun()
+                        
+                        st.markdown('</div>', unsafe_allow_html=True)
     
     # Export section
     st.markdown("---")
@@ -673,6 +813,7 @@ if st.session_state.bookmarked_models:
     with export_col2:
         if st.button("🗑️ Clear All", use_container_width=True):
             st.session_state.bookmarked_models = []
+            st.session_state.product_images = {}  # Clear image cache too
             st.rerun()
 else:
     st.info("📌 No models bookmarked yet. Select models to create your custom list!")
@@ -684,6 +825,7 @@ with col1:
     st.markdown("### 💡 Quick Tips")
     st.write("• View PDF spec sheets")
     st.write("• Bookmark models for lists")
+    st.write("• Toggle image/text view")
     st.write("• Google search finds prices")
     st.write("• Export includes all specs")