Update app.py

app.py

@@ -2,6 +2,14 @@
 """
 Turbo Air Viewer - Equipment Specification Database Viewer
 Enhanced version with product image extraction and display
+
+Required dependencies (add to requirements.txt):
+- streamlit
+- pandas
+- requests
+- Pillow
+- PyMuPDF
+- reportlab
 """
 
 import streamlit as st
@@ -20,6 +28,12 @@ from urllib.parse import quote
 from PIL import Image
 import fitz  # PyMuPDF
 import tempfile
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import letter, landscape
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib.units import inch
+from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph, Spacer, Image as RLImage, PageBreak
+from reportlab.lib.enums import TA_CENTER
 
 # Streamlit page config MUST be first
 st.set_page_config(
@@ -306,7 +320,7 @@ def extract_pdf_thumbnail(pdf_url, model_name, max_width=300, max_height=400):
         return st.session_state.product_images[cache_key]
     
     try:
-        # Download PDF to temporary file
+        # Download PDF to temporary file (silently)
         response = requests.get(pdf_url, timeout=30)
         if response.status_code == 200:
             with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as tmp_file:
@@ -314,13 +328,12 @@ def extract_pdf_thumbnail(pdf_url, model_name, max_width=300, max_height=400):
                 tmp_path = tmp_file.name
             
             # Open PDF and extract first page
-            pdf_document = fitz.Document(tmp_path)
+            pdf_document = fitz.open(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)
+            pix = first_page.get_pixmap(matrix=mat)
             
             # Convert to PIL Image
             img_data = pix.tobytes("png")
@@ -358,7 +371,7 @@ def extract_pdf_thumbnail(pdf_url, model_name, max_width=300, max_height=400):
             return img_base64
             
     except Exception as e:
-        st.warning(f"Could not extract image for {model_name}: {str(e)}")
+        # Silently fail - don't show warnings
         return None
 
 # Cache functions
@@ -575,6 +588,141 @@ def export_bookmarked_models():
     df = pd.DataFrame(export_data)
     return df.to_csv(index=False)
 
+def export_bookmarked_models_pdf():
+    """Export bookmarked models to PDF with images and specifications"""
+    if not st.session_state.bookmarked_models:
+        return None
+    
+    # Create PDF in memory
+    buffer = io.BytesIO()
+    doc = SimpleDocTemplate(buffer, pagesize=landscape(letter), 
+                           topMargin=0.5*inch, bottomMargin=0.5*inch,
+                           leftMargin=0.5*inch, rightMargin=0.5*inch)
+    
+    # Container for the 'Flowable' objects
+    elements = []
+    
+    # Styles
+    styles = getSampleStyleSheet()
+    title_style = ParagraphStyle(
+        'CustomTitle',
+        parent=styles['Heading1'],
+        fontSize=24,
+        textColor=colors.HexColor('#4CAF50'),
+        spaceAfter=30,
+        alignment=TA_CENTER
+    )
+    
+    model_title_style = ParagraphStyle(
+        'ModelTitle',
+        parent=styles['Heading2'],
+        fontSize=16,
+        textColor=colors.HexColor('#333333'),
+        spaceAfter=12
+    )
+    
+    # Title
+    elements.append(Paragraph("Turbo Air Equipment Selection Report", title_style))
+    elements.append(Paragraph(f"Generated: {datetime.now().strftime('%B %d, %Y at %I:%M %p')}", 
+                             styles['Normal']))
+    elements.append(Spacer(1, 0.5*inch))
+    
+    # Process each bookmarked model
+    for idx, model in enumerate(st.session_state.bookmarked_models):
+        if idx > 0:
+            elements.append(PageBreak())
+        
+        model_data = get_model_data(model)
+        if not model_data:
+            continue
+        
+        # Model header
+        elements.append(Paragraph(f"{model} - {get_product_type(model)}", model_title_style))
+        
+        # Try to get product image
+        if 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}"
+            
+            # Get cached image or extract it
+            cache_key = f"thumb_{model}"
+            img_base64 = st.session_state.product_images.get(cache_key)
+            
+            if not img_base64:
+                img_base64 = extract_pdf_thumbnail(pdf_url, model, max_width=200, max_height=250)
+            
+            if img_base64:
+                # Convert base64 to image for PDF
+                img_data = base64.b64decode(img_base64)
+                img = RLImage(io.BytesIO(img_data), width=2*inch, height=2.5*inch)
+                elements.append(img)
+                elements.append(Spacer(1, 0.2*inch))
+        
+        # Specifications table
+        specs = model_data['data'].get('specs', {})
+        specs = clean_spec_data(specs)
+        
+        # Create specifications data for table
+        spec_data = [['Specification', 'Value']]
+        
+        if specs.get('voltage') and specs.get('voltage') != 'N/A':
+            spec_data.append(['Voltage', specs['voltage']])
+        if specs.get('amperage') and specs.get('amperage') != 'N/A':
+            spec_data.append(['Amperage', specs['amperage']])
+        if specs.get('phase') and specs.get('phase') != 'N/A':
+            spec_data.append(['Phase', specs['phase']])
+        if specs.get('frequency') and specs.get('frequency') != 'N/A':
+            spec_data.append(['Frequency', specs['frequency']])
+        if specs.get('dimensions') and specs.get('dimensions') != 'N/A':
+            spec_data.append(['Dimensions', specs['dimensions']])
+        if specs.get('weight') and specs.get('weight') != 'N/A':
+            spec_data.append(['Weight', specs['weight']])
+        if specs.get('capacity') and specs.get('capacity') != 'N/A':
+            spec_data.append(['Capacity', specs['capacity']])
+        if specs.get('refrigerant') and specs.get('refrigerant') != 'N/A':
+            spec_data.append(['Refrigerant', specs['refrigerant']])
+        if specs.get('temperature_range') and specs.get('temperature_range') != 'N/A':
+            spec_data.append(['Temperature Range', specs['temperature_range']])
+        if specs.get('compressor') and specs.get('compressor') != 'N/A':
+            spec_data.append(['Compressor', specs['compressor']])
+        if specs.get('btu') and specs.get('btu') != 'N/A':
+            spec_data.append(['BTU', specs['btu']])
+        
+        if len(spec_data) > 1:
+            # Create table
+            spec_table = Table(spec_data, colWidths=[2.5*inch, 4*inch])
+            spec_table.setStyle(TableStyle([
+                ('BACKGROUND', (0, 0), (-1, 0), colors.HexColor('#4CAF50')),
+                ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
+                ('ALIGN', (0, 0), (-1, -1), 'LEFT'),
+                ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
+                ('FONTSIZE', (0, 0), (-1, 0), 12),
+                ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
+                ('BACKGROUND', (0, 1), (-1, -1), colors.beige),
+                ('GRID', (0, 0), (-1, -1), 1, colors.black),
+                ('FONTNAME', (0, 1), (-1, -1), 'Helvetica'),
+                ('FONTSIZE', (0, 1), (-1, -1), 10),
+                ('ROWBACKGROUNDS', (0, 1), (-1, -1), [colors.white, colors.HexColor('#f0f0f0')]),
+            ]))
+            elements.append(spec_table)
+            elements.append(Spacer(1, 0.3*inch))
+        
+        # Features
+        features = model_data['data'].get('features', [])
+        if features:
+            elements.append(Paragraph("<b>Features:</b>", styles['Heading3']))
+            for feature in features:
+                elements.append(Paragraph(f"• {feature}", styles['Normal']))
+            elements.append(Spacer(1, 0.2*inch))
+        
+        # Source info
+        elements.append(Paragraph(f"<i>Source: {model_data['filename']}</i>", styles['Normal']))
+    
+    # Build PDF
+    doc.build(elements)
+    buffer.seek(0)
+    return buffer.getvalue()
+
 def display_pdf_preview(file_path, model_name):
     """Display PDF inline in Streamlit app - optimized for HuggingFace Spaces"""
     
@@ -697,19 +845,12 @@ def get_high_accuracy_models(limit=8):
     
     return high_accuracy[:limit]
 
-# MAIN UI
-st.title("❄️ Turbo Air Equipment Viewer")
-st.caption("Professional Equipment Specification Database")
-
-# Get all models
-all_models = get_all_models()
-
-if not all_models:
-    st.error("⚠️ No data found in database. Please ensure turbo_air_db.sqlite is available.")
-    st.stop()
-
-# Bookmarks section
-if st.session_state.bookmarked_models:
+def display_bookmarked_models():
+    """Display bookmarked models section with optimized toggle"""
+    if not st.session_state.bookmarked_models:
+        st.info("📌 No models bookmarked yet. Select models to create your custom list!")
+        return
+    
     st.markdown("### 📌 Bookmarked Models")
     
     view_col1, view_col2, view_col3 = st.columns([2, 1, 1])
@@ -717,14 +858,11 @@ if st.session_state.bookmarked_models:
         st.markdown(f"**{len(st.session_state.bookmarked_models)} models selected**")
     
     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
-        ):
+        # Toggle for text-only view - optimized to avoid loading
+        toggle_label = "📷 Show Images" if st.session_state.text_only_view else "📝 Text Only"
+        if st.button(toggle_label, key="toggle_view", use_container_width=True):
             st.session_state.text_only_view = not st.session_state.text_only_view
-            st.rerun()
+            # Don't use st.rerun() to avoid loading screen
     
     # Display bookmarked models with or without images
     if st.session_state.text_only_view:
@@ -770,11 +908,12 @@ if st.session_state.bookmarked_models:
                             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)
+                            # Check if image is already cached
+                            cache_key = f"thumb_{model}"
+                            img_base64 = st.session_state.product_images.get(cache_key)
                             
                             if img_base64:
+                                # Use cached image
                                 st.markdown(
                                     f'<img src="data:image/png;base64,{img_base64}" '
                                     f'style="width:100%; max-height:200px; object-fit:contain;" '
@@ -782,7 +921,18 @@ if st.session_state.bookmarked_models:
                                     unsafe_allow_html=True
                                 )
                             else:
-                                st.info("📄 No preview available")
+                                # Extract image with minimal loading indication
+                                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}**")
@@ -797,7 +947,7 @@ if st.session_state.bookmarked_models:
     
     # Export section
     st.markdown("---")
-    export_col1, export_col2 = st.columns(2)
+    export_col1, export_col2, export_col3 = st.columns(3)
     
     with export_col1:
         csv_data = export_bookmarked_models()
@@ -811,12 +961,37 @@ if st.session_state.bookmarked_models:
             )
     
     with export_col2:
+        # PDF export button
+        pdf_data = export_bookmarked_models_pdf()
+        if pdf_data:
+            st.download_button(
+                "📄 Export PDF",
+                data=pdf_data,
+                file_name=f"turbo_air_report_{datetime.now().strftime('%Y%m%d_%H%M')}.pdf",
+                mime="application/pdf",
+                use_container_width=True,
+                type="primary"
+            )
+    
+    with export_col3:
         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!")
+
+# MAIN UI
+st.title("❄️ Turbo Air Equipment Viewer")
+st.caption("Professional Equipment Specification Database")
+
+# Get all models
+all_models = get_all_models()
+
+if not all_models:
+    st.error("⚠️ No data found in database. Please ensure turbo_air_db.sqlite is available.")
+    st.stop()
+
+# Bookmarks section
+display_bookmarked_models()
 
 # Main content area
 col1, col2 = st.columns([1, 3])
@@ -826,8 +1001,8 @@ with col1:
     st.write("• View PDF spec sheets")
     st.write("• Bookmark models for lists")
     st.write("• Toggle image/text view")
+    st.write("• Export to CSV or PDF")
     st.write("• Google search finds prices")
-    st.write("• Export includes all specs")
 
 with col2:
     st.markdown('### 🔍 Model Search')