Update app.py

app.py

@@ -495,17 +495,31 @@ def display_pdf_preview(file_path, model_name):
     """Display PDF preview inline"""
     import os
     
-    # Extract just the filename from the Windows path
-    pdf_filename = os.path.basename(file_path)
+    # Extract just the filename from the Windows path, ignoring subdirectories
+    # Convert backslashes to forward slashes first
+    normalized_path = file_path.replace('\\', '/')
+    pdf_filename = os.path.basename(normalized_path)
+    
+    # Clean up the filename (remove any extra spaces or special characters if needed)
+    pdf_filename = pdf_filename.strip()
+    
+    # Debug info
+    st.write(f"🔍 Looking for PDF: {pdf_filename}")
     
     # Try multiple possible locations
     possible_paths = [
         os.path.join("pdfs", pdf_filename),  # /pdfs/filename.pdf
         os.path.join(".", "pdfs", pdf_filename),  # ./pdfs/filename.pdf
-        pdf_filename,  # Just filename in root
-        file_path  # Original path (unlikely to work)
+        f"pdfs/{pdf_filename}",  # Direct path string
     ]
     
+    # Also try without spaces in filename (in case of naming issues)
+    pdf_filename_no_spaces = pdf_filename.replace(' ', '_')
+    possible_paths.extend([
+        os.path.join("pdfs", pdf_filename_no_spaces),
+        f"pdfs/{pdf_filename_no_spaces}"
+    ])
+    
     pdf_found = False
     actual_path = None
     
@@ -514,13 +528,13 @@ def display_pdf_preview(file_path, model_name):
         if os.path.exists(test_path):
             pdf_found = True
             actual_path = test_path
+            st.success(f"✅ Found PDF at: {test_path}")
             break
     
-    if pdf_found:
+    if pdf_found and actual_path is not None:
         try:
-            if actual_path:
-                with open(actual_path, "rb") as f:
-                    pdf_data = f.read()
+            with open(actual_path, "rb") as f:
+                pdf_data = f.read()
             
             file_size_mb = len(pdf_data) / (1024 * 1024)
             
@@ -531,7 +545,7 @@ def display_pdf_preview(file_path, model_name):
                 st.download_button(
                     "📥 Download PDF",
                     data=pdf_data,
-                    file_name=f"{model_name}_spec_sheet.pdf",
+                    file_name=pdf_filename,
                     mime="application/pdf",
                     use_container_width=True,
                     type="primary"
@@ -573,16 +587,33 @@ def display_pdf_preview(file_path, model_name):
             st.info(f"Attempted to read from: {actual_path}")
     else:
         st.error("❌ PDF file not found.")
-        st.info(f"Looking for: {pdf_filename}")
-        st.info("The PDF may not have been uploaded yet or the filename may be different.")
+        st.info(f"Expected filename: {pdf_filename}")
+        
+        # List files in pdfs directory to help debug
+        try:
+            pdfs_dir = "pdfs"
+            if os.path.exists(pdfs_dir):
+                files_in_pdfs = os.listdir(pdfs_dir)
+                # Find similar filenames
+                similar_files = [f for f in files_in_pdfs if pdf_filename.lower()[:10] in f.lower()]
+                
+                if similar_files:
+                    st.warning("🔍 Similar files found in /pdfs folder:")
+                    for f in similar_files[:5]:  # Show max 5 similar files
+                        st.write(f"- {f}")
+                else:
+                    st.info("No similar files found. The PDF might not have been uploaded.")
+            else:
+                st.error("The /pdfs directory doesn't exist!")
+                
+        except Exception as e:
+            st.error(f"Error listing PDF directory: {e}")
         
         # Debug information
-        with st.expander("🔍 Debug Information"):
-            st.write(f"**Original path in database:** {file_path}")
+        with st.expander("🔧 Debug Information"):
+            st.write(f"**Original path:** {file_path}")
             st.write(f"**Extracted filename:** {pdf_filename}")
-            st.write(f"**Searched locations:**")
-            for path in possible_paths:
-                st.write(f"- {path}")
+            st.write(f"**Model:** {model_name}")
 
 def get_high_accuracy_models(limit=8):
     """Get models with highest extraction quality"""