Update app.py

app.py

@@ -492,63 +492,110 @@ def export_bookmarked_models():
     return df.to_csv(index=False)
 
 def display_pdf_preview(file_path, model_name):
-    """Display PDF using Google Docs Viewer - most reliable method"""
+    """Display PDF - works on Hugging Face with Git LFS"""
     import os
+    import requests
+    import base64
     
     # Extract filename
     normalized_path = file_path.replace('\\', '/')
     pdf_filename = normalized_path.split('/')[-1]
     
-    # Create direct URL
-    pdf_url = f"https://huggingface.co/spaces/TurboAir/TurboAirViewer/resolve/main/pdfs/{pdf_filename}"
+    # Check if running locally or on Hugging Face
+    is_local = not os.environ.get('SPACE_ID')
+    
+    if is_local:
+        # Local file system access
+        pdf_path = os.path.join('pdfs', pdf_filename)
+        if os.path.exists(pdf_path):
+            with open(pdf_path, 'rb') as f:
+                pdf_data = f.read()
+                file_size_mb = len(pdf_data) / (1024 * 1024)
+        else:
+            st.error(f"PDF not found locally: {pdf_path}")
+            return
+    else:
+        # Hugging Face - need to download from URL
+        pdf_url = f"https://huggingface.co/spaces/TurboAir/TurboAirViewer/resolve/main/pdfs/{pdf_filename}"
+        
+        try:
+            with st.spinner("Loading PDF..."):
+                response = requests.get(pdf_url, timeout=30)
+                response.raise_for_status()
+                pdf_data = response.content
+                file_size_mb = len(pdf_data) / (1024 * 1024)
+                
+                # Verify we got actual PDF data, not Git LFS pointer
+                if len(pdf_data) < 1000:
+                    st.error("PDF file is too small - might be Git LFS pointer")
+                    # Use external viewer as fallback
+                    st.markdown(f"[📄 Open PDF in new tab]({pdf_url})")
+                    return
+                    
+        except Exception as e:
+            st.error(f"Error loading PDF: {str(e)}")
+            # Fallback to external link
+            st.markdown(f"[📄 Open PDF in new tab]({pdf_url})")
+            return
     
     # Control buttons
     col1, col2, col3 = st.columns([2, 1, 1])
     
     with col1:
-        st.markdown(f'<a href="{pdf_url}" download="{pdf_filename}"><button style="width: 100%; padding: 0.5rem; background-color: #FF4B4B; color: white; border: none; border-radius: 5px; cursor: pointer;">📥 Download PDF</button></a>', unsafe_allow_html=True)
+        st.download_button(
+            "📥 Download PDF",
+            data=pdf_data,
+            file_name=pdf_filename,
+            mime="application/pdf",
+            use_container_width=True,
+            type="primary"
+        )
     
     with col2:
-        st.success("✅ PDF ready")
+        st.success(f"✅ PDF ready ({file_size_mb:.1f} MB)")
     
     with col3:
         if st.button("❌ Close Preview", use_container_width=True):
             st.session_state[f'show_pdf_{model_name}'] = False
             st.rerun()
     
-    # Display PDF - NO GREEN BORDER
     st.markdown("---")
     
-    # Method 1: Direct embed (try this first)
-    pdf_viewer = f"""
-    <embed 
-        src="{pdf_url}#view=FitH&toolbar=0" 
-        width="100%" 
-        height="900px" 
-        type="application/pdf">
+    # Display PDF using base64 encoding (works everywhere)
+    base64_pdf = base64.b64encode(pdf_data).decode('utf-8')
+    
+    # Try multiple display methods
+    pdf_display = f"""
+    <style>
+        .pdf-viewer {{
+            width: 100%;
+            height: 900px;
+            border: none;
+        }}
+    </style>
+    
+    <!-- Method 1: Object with embed fallback -->
+    <object data="data:application/pdf;base64,{base64_pdf}" 
+            type="application/pdf" 
+            class="pdf-viewer">
+        <embed src="data:application/pdf;base64,{base64_pdf}" 
+               type="application/pdf" 
+               class="pdf-viewer" />
+        <p>PDF preview not available. Please use the download button above.</p>
+    </object>
     """
     
-    # Method 2: If embed doesn't work, use iframe
-    # pdf_viewer = f"""
-    # <iframe 
-    #     src="{pdf_url}#view=FitH&toolbar=0" 
-    #     width="100%" 
-    #     height="900px" 
-    #     frameborder="0">
-    # </iframe>
-    # """
-    
-    # Method 3: If neither works, use Google Docs
-    # pdf_viewer = f"""
-    # <iframe 
-    #     src="https://docs.google.com/viewer?url={pdf_url}&embedded=true" 
-    #     width="100%" 
-    #     height="900px" 
-    #     frameborder="0">
-    # </iframe>
-    # """
-    
-    st.markdown(pdf_viewer, unsafe_allow_html=True)
+    st.markdown(pdf_display, unsafe_allow_html=True)
+    
+    # If base64 doesn't work, show alternative
+    st.markdown("---")
+    st.info("💡 If PDF doesn't display above, you can:")
+    col1, col2 = st.columns(2)
+    with col1:
+        st.markdown("• Use the Download button above")
+    with col2:
+        if not is_local:
+            st.markdown(f"• [Open in new tab ↗]({pdf_url})")
 
 def get_high_accuracy_models(limit=8):
     """Get models with highest extraction quality"""