Upload folder using huggingface_hub

hate_speech_demo.py

@@ -741,9 +741,20 @@ def create_gradio_app():
         # Add loading spinner
         loading_spinner = gr.HTML('<div id="loading-spinner"></div>')
         
-        # Create a file component to serve the PDF (hidden from UI)
+        # Add a hidden file component to serve the PDF
         pdf_file = gr.File("Hate Speech Policy.pdf", visible=False, label="Policy PDF")
         
+        # Add a "file preview" component that can be accessed
+        gr.HTML("""
+        <div id="pdf-container" style="display: none;">
+            <h3>Hate Speech Policy Document</h3>
+            <div class="file-preview">
+                <div class="file-name">Hate Speech Policy.pdf</div>
+                <a href="file=Hate Speech Policy.pdf" download="Hate Speech Policy.pdf" id="pdf-direct-link">Download</a>
+            </div>
+        </div>
+        """)
+        
         # Add policy popup HTML with robust PDF loading - ALWAYS LOADS FIRST TIME
         policy_popup_html = """
         <div id="policy-popup" class="policy-popup">
@@ -780,48 +791,47 @@ def create_gradio_app():
         </div>
         
         <script>
-        // PRELOAD THE PDF IMMEDIATELY WHEN PAGE LOADS
+        // DIRECTLY ACCESS THE PDF FILE COMPONENT
         document.addEventListener('DOMContentLoaded', function() {
-            // Find and cache all PDF links on page load
-            let pdfUrl = null;
-            const fileComponents = document.querySelectorAll("[data-testid='file']");
-            
-            // First try to get from the file component
-            for (const comp of fileComponents) {
-                const downloadBtn = comp.querySelector("a");
-                if (downloadBtn && downloadBtn.href && downloadBtn.href.includes("file=")) {
-                    pdfUrl = downloadBtn.href;
-                    break;
-                }
-            }
-            
-            // If no file component found, try to find any PDF link
-            if (!pdfUrl) {
-                const allLinks = document.querySelectorAll("a");
-                for (const link of allLinks) {
-                    if (link.href && link.href.includes("Hate%20Speech%20Policy.pdf")) {
-                        pdfUrl = link.href;
-                        break;
-                    }
-                }
-            }
-            
-            // If we found a PDF link, preload it in a hidden iframe
-            if (pdfUrl) {
-                // Create hidden preloader
-                const preloader = document.createElement('iframe');
-                preloader.style.display = 'none';
-                preloader.src = pdfUrl;
-                document.body.appendChild(preloader);
+            // Get references to all file components
+            setTimeout(function() {
+                const policyButtons = document.querySelectorAll('a[onclick*="openPolicyPopup"]');
                 
-                // Store the URL for later use
-                window.pdfPolicyUrl = pdfUrl;
+                // Look for file component
+                const fileComponents = document.querySelectorAll('.file-preview');
                 
-                console.log("Preloaded PDF:", pdfUrl);
-            }
+                if (fileComponents.length > 0) {
+                    // Found at least one file component, look for our PDF
+                    for (const comp of fileComponents) {
+                        const fileNameElement = comp.querySelector('.file-name');
+                        if (fileNameElement && fileNameElement.textContent.includes('Hate Speech Policy.pdf')) {
+                            // This is our PDF! Get the download link
+                            const downloadBtn = comp.querySelector('a[download]');
+                            if (downloadBtn && downloadBtn.href) {
+                                // Store the URL globally for immediate access
+                                window.pdfPolicyUrl = downloadBtn.href;
+                                console.log('Found PDF at', window.pdfPolicyUrl);
+                                
+                                // Update all the policy buttons to use this direct URL
+                                policyButtons.forEach(btn => {
+                                    btn.setAttribute('data-pdf-url', window.pdfPolicyUrl);
+                                });
+                                
+                                // Preload the PDF in a hidden iframe for faster loading
+                                const preloader = document.createElement('iframe');
+                                preloader.style.display = 'none';
+                                preloader.src = window.pdfPolicyUrl;
+                                document.body.appendChild(preloader);
+                                
+                                break;
+                            }
+                        }
+                    }
+                }
+            }, 1000); // Give the page a second to fully load
         });
         
-        // Function to handle opening the policy popup - GUARANTEED FIRST-CLICK LOADING
+        // Function to handle opening the policy popup - OPTIMIZED FOR SPEED
         function openPolicyPopup() {
             // Display the popup immediately
             const popup = document.getElementById('policy-popup');
@@ -838,31 +848,39 @@ def create_gradio_app():
             iframe.style.display = "block";
             fallback.style.display = "none";
             
-            // If we have a preloaded URL, use it
-            if (window.pdfPolicyUrl) {
-                loadPdf(window.pdfPolicyUrl);
-                return;
-            }
+            // DIRECT METHOD: Check if the link that was clicked has the PDF URL
+            const clickedButton = event ? event.currentTarget : null;
+            let pdfUrl = clickedButton ? clickedButton.getAttribute('data-pdf-url') : null;
             
-            // Otherwise search for the PDF link
-            let pdfUrl = null;
+            // If we don't have it from the button, try the cached version
+            if (!pdfUrl && window.pdfPolicyUrl) {
+                pdfUrl = window.pdfPolicyUrl;
+            }
             
-            // Try to find the file component first
-            const fileComponents = document.querySelectorAll("[data-testid='file']");
-            for (const comp of fileComponents) {
-                const downloadBtn = comp.querySelector("a");
-                if (downloadBtn && downloadBtn.href && downloadBtn.href.includes("file=")) {
-                    pdfUrl = downloadBtn.href;
-                    break;
+            // If we still don't have it, look for it directly
+            if (!pdfUrl) {
+                // Find all file components and look for our PDF
+                const fileComponents = document.querySelectorAll('.file-preview');
+                for (const comp of fileComponents) {
+                    const fileNameElement = comp.querySelector('.file-name');
+                    if (fileNameElement && fileNameElement.textContent.includes('Hate Speech Policy.pdf')) {
+                        const downloadBtn = comp.querySelector('a[download]');
+                        if (downloadBtn && downloadBtn.href) {
+                            pdfUrl = downloadBtn.href;
+                            window.pdfPolicyUrl = pdfUrl; // Cache for future use
+                            break;
+                        }
+                    }
                 }
             }
             
-            // If that didn't work, search for any PDF link
+            // If we still don't have it, look for any link with the PDF name
             if (!pdfUrl) {
                 const allLinks = document.querySelectorAll("a");
                 for (const link of allLinks) {
                     if (link.href && link.href.includes("Hate%20Speech%20Policy.pdf")) {
                         pdfUrl = link.href;
+                        window.pdfPolicyUrl = pdfUrl; // Cache for future use
                         break;
                     }
                 }
@@ -870,10 +888,9 @@ def create_gradio_app():
             
             // If we found a URL, load it
             if (pdfUrl) {
-                window.pdfPolicyUrl = pdfUrl; // Cache for future use
                 loadPdf(pdfUrl);
             } else {
-                // Show fallback if no PDF found
+                // No PDF found, show fallback
                 loadingIndicator.style.display = "none";
                 iframe.style.display = "none";
                 fallback.style.display = "block";
@@ -892,18 +909,14 @@ def create_gradio_app():
             // Set download link in case user wants to download
             downloadLink.href = pdfUrl;
             
-            // Load PDF into iframe
+            // Load PDF directly into iframe - no Google Viewer which adds extra delay
             iframe.src = pdfUrl;
             
             // Check if iframe loaded successfully
             let loadTimeout = setTimeout(function() {
-                // If still loading after 5 seconds, show fallback
+                // If still loading after 3 seconds, hide the loading indicator anyway
                 loadingIndicator.style.display = "none";
-                
-                // Try Google Viewer as a last resort
-                const googleViewerUrl = "https://docs.google.com/viewer?embedded=true&url=";
-                iframe.src = googleViewerUrl + encodeURIComponent(pdfUrl);
-            }, 5000);
+            }, 3000);
             
             // When iframe loads, hide loading indicator
             iframe.onload = function() {