Update app.py

app.py

@@ -492,59 +492,63 @@ def export_bookmarked_models():
     return df.to_csv(index=False)
 
 def display_pdf_preview(file_path, model_name):
-    """Display PDF inline without auto-downloading"""
+    """Display PDF using Google Docs Viewer - most reliable method"""
     import os
     
-    # Extract filename from the database path
+    # Extract filename
     normalized_path = file_path.replace('\\', '/')
     pdf_filename = normalized_path.split('/')[-1]
     
-    # Create direct Hugging Face URL for the PDF
+    # Create direct URL
     pdf_url = f"https://huggingface.co/spaces/TurboAir/TurboAirViewer/resolve/main/pdfs/{pdf_filename}"
     
-    # Create columns for controls
+    # Control buttons
     col1, col2, col3 = st.columns([2, 1, 1])
     
     with col1:
-        # Only download when button is clicked - NOT automatic
-        if st.button("📥 Download PDF", key=f"download_{model_name}", use_container_width=True, type="primary"):
-            import requests
-            try:
-                response = requests.get(pdf_url, timeout=30)
-                st.download_button(
-                    "💾 Click to Save",
-                    data=response.content,
-                    file_name=pdf_filename,
-                    mime="application/pdf",
-                    key=f"save_{model_name}"
-                )
-            except:
-                st.error("Could not download PDF")
+        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)
     
     with col2:
-        st.success("✅ PDF ready to view")
+        st.success("✅ PDF ready")
     
     with col3:
         if st.button("❌ Close Preview", use_container_width=True):
             st.session_state[f'show_pdf_{model_name}'] = False
             st.rerun()
     
-    # Display PDF inline WITHOUT downloading
+    # Display PDF - NO GREEN BORDER
     st.markdown("---")
     
-    # Method 1: Google Docs Viewer (most reliable, no download)
-    google_viewer_url = f"https://docs.google.com/viewer?url={pdf_url}&embedded=true"
-    
-    st.markdown(
-        f'<iframe src="{google_viewer_url}" width="100%" height="800px" frameborder="0" style="border: 2px solid #4CAF50; border-radius: 8px;"></iframe>',
-        unsafe_allow_html=True
-    )
-    
-    # Method 2: If Google Viewer doesn't work, try direct embed (comment out Method 1 and uncomment below)
-    # st.markdown(
-    #     f'<embed src="{pdf_url}#toolbar=0&navpanes=0&scrollbar=1" width="100%" height="800px" type="application/pdf" style="border: 2px solid #4CAF50; border-radius: 8px;">',
-    #     unsafe_allow_html=True
-    # )
+    # 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">
+    """
+    
+    # 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)
 
 def get_high_accuracy_models(limit=8):
     """Get models with highest extraction quality"""