Upload app.py with huggingface_hub

app.py

@@ -25,251 +25,6 @@ TITLE = """
 <p><em>Powered by Dolphin 2.6 Mistral 7B DPO - Venice Edition Uncensored</em></p>
 </div>
 <hr>
-
-<script>
-// Venice Edition Import Functionality
-function fillVeniceEditionFields(data) {
-    console.log('🎯 Filling Venice Edition fields with data:', data);
-    
-    if (!data || !data.data) {
-        console.error('❌ Invalid data format');
-        return false;
-    }
-    
-    const joyCaptionData = data.data;
-    let fieldsFound = 0;
-    let fieldsFilled = 0;
-    
-    // Find all textareas and inputs in Venice Edition
-    const textFields = document.querySelectorAll('textarea, input[type="text"]');
-    fieldsFound = textFields.length;
-    
-    console.log(`📝 Found ${fieldsFound} text fields in Venice Edition`);
-    
-    // Function to fill a field by matching placeholder or context
-    function fillFieldByContext(searchTerms, value, fieldType = 'general') {
-        for (const field of textFields) {
-            const placeholder = (field.placeholder || '').toLowerCase();
-            const label = getFieldLabel(field).toLowerCase();
-            const context = (placeholder + ' ' + label).toLowerCase();
-            
-            // Check if any search term matches
-            const matches = searchTerms.some(term => context.includes(term.toLowerCase()));
-            
-            if (matches && !field.value.trim()) { // Only fill if empty
-                field.value = value;
-                field.dispatchEvent(new Event('input', { bubbles: true }));
-                field.dispatchEvent(new Event('change', { bubbles: true }));
-                fieldsFilled++;
-                console.log(`✅ Filled ${fieldType}: ${value.substring(0, 50)}...`);
-                return true;
-            }
-        }
-        return false;
-    }
-    
-    // Fill fields based on JoyCaption data
-    if (joyCaptionData.caption_engaging) {
-        fillFieldByContext(['original', 'description', 'paste your image'], joyCaptionData.caption_engaging, 'Engaging Caption');
-    }
-    
-    if (joyCaptionData.caption_casual_friend && !fieldsFilled) {
-        fillFieldByContext(['original', 'description', 'paste your image'], joyCaptionData.caption_casual_friend, 'Casual Caption');
-    }
-    
-    if (joyCaptionData.caption_keywords && !fieldsFilled) {
-        fillFieldByContext(['original', 'description', 'paste your image'], joyCaptionData.caption_keywords, 'Keywords Caption');
-    }
-    
-    if (joyCaptionData.keywords) {
-        fillFieldByContext(['keyword', 'seo', 'required'], joyCaptionData.keywords, 'Keywords');
-    }
-    
-    if (joyCaptionData.custom_instructions) {
-        fillFieldByContext(['correction', 'manual', 'guidance'], joyCaptionData.custom_instructions, 'Custom Instructions');
-    }
-    
-    // If no specific matches, fill the first few large text areas with captions
-    if (fieldsFilled === 0 && (joyCaptionData.caption_engaging || joyCaptionData.caption_casual_friend || joyCaptionData.caption_keywords)) {
-        const largeTextareas = Array.from(textFields).filter(field => 
-            field.tagName === 'TEXTAREA' && !field.value.trim()
-        );
-        
-        let index = 0;
-        const captions = [joyCaptionData.caption_engaging, joyCaptionData.caption_casual_friend, joyCaptionData.caption_keywords].filter(Boolean);
-        
-        if (captions.length > 0 && largeTextareas[index]) {
-            largeTextareas[index].value = captions[0];
-            largeTextareas[index].dispatchEvent(new Event('input', { bubbles: true }));
-            fieldsFilled++;
-            console.log(`✅ Filled textarea ${index} with caption`);
-        }
-    }
-    
-    console.log(`📊 Filled ${fieldsFilled} out of ${fieldsFound} fields`);
-    return fieldsFilled > 0;
-}
-
-function getFieldLabel(field) {
-    // Try to find associated label
-    if (field.id) {
-        const label = document.querySelector(`label[for="${field.id}"]`);
-        if (label) return label.textContent || '';
-    }
-    
-    // Look in parent elements for text context
-    let parent = field.parentElement;
-    let levels = 0;
-    while (parent && levels < 3) {
-        const labels = parent.querySelectorAll('label, span, div');
-        for (const label of labels) {
-            const text = label.textContent || '';
-            if (text.length > 0 && text.length < 100) {
-                return text;
-            }
-        }
-        parent = parent.parentElement;
-        levels++;
-    }
-    
-    return field.placeholder || '';
-}
-
-function loadFromFile(event) {
-    const file = event.target.files[0];
-    if (!file) return;
-    
-    const reader = new FileReader();
-    reader.onload = function(e) {
-        try {
-            const data = JSON.parse(e.target.result);
-            const success = fillVeniceEditionFields(data);
-            
-            if (success) {
-                alert(`✅ Successfully imported and filled Venice Edition fields!`);
-            } else {
-                alert('⚠️ Data imported but no matching fields found. Check field labels.');
-            }
-        } catch (error) {
-            console.error('❌ Import error:', error);
-            alert('❌ Failed to import file: ' + error.message);
-        }
-    };
-    reader.readAsText(file);
-}
-
-function loadFromClipboard() {
-    navigator.clipboard.readText().then(text => {
-        try {
-            const data = JSON.parse(text);
-            const success = fillVeniceEditionFields(data);
-            
-            if (success) {
-                alert(`✅ Successfully imported from clipboard and filled Venice Edition fields!`);
-            } else {
-                alert('⚠️ Data imported but no matching fields found. Check field labels.');
-            }
-        } catch (error) {
-            console.error('❌ Clipboard import error:', error);
-            alert('❌ Failed to import from clipboard: Invalid JSON format');
-        }
-    }).catch(err => {
-        console.error('❌ Clipboard read error:', err);
-        alert('❌ Failed to read clipboard: ' + err.message);
-    });
-}
-
-function createImportButtons() {
-    // Remove existing buttons first
-    const existingContainer = document.getElementById('venice-import-container');
-    if (existingContainer) existingContainer.remove();
-    
-    // Create import button container
-    const container = document.createElement('div');
-    container.id = 'venice-import-container';
-    container.style.cssText = `
-        position: fixed;
-        top: 20px;
-        right: 20px;
-        z-index: 9999;
-        display: flex;
-        flex-direction: column;
-        gap: 10px;
-    `;
-    
-    // Create file input (hidden)
-    const fileInput = document.createElement('input');
-    fileInput.type = 'file';
-    fileInput.accept = '.json';
-    fileInput.style.display = 'none';
-    fileInput.addEventListener('change', loadFromFile);
-    
-    // Create import from file button
-    const importFileBtn = document.createElement('button');
-    importFileBtn.innerHTML = '📁 Import from File';
-    importFileBtn.style.cssText = `
-        background: linear-gradient(135deg, #2e7d32, #388e3c);
-        color: white;
-        border: none;
-        padding: 12px 20px;
-        border-radius: 25px;
-        font-weight: 600;
-        cursor: pointer;
-        box-shadow: 0 4px 12px rgba(46, 125, 50, 0.3);
-        transition: all 0.3s ease;
-    `;
-    
-    importFileBtn.addEventListener('mouseover', () => {
-        importFileBtn.style.transform = 'translateY(-2px)';
-        importFileBtn.style.boxShadow = '0 6px 16px rgba(46, 125, 50, 0.4)';
-    });
-    
-    importFileBtn.addEventListener('mouseout', () => {
-        importFileBtn.style.transform = 'translateY(0)';
-        importFileBtn.style.boxShadow = '0 4px 12px rgba(46, 125, 50, 0.3)';
-    });
-    
-    importFileBtn.addEventListener('click', () => fileInput.click());
-    
-    // Create import from clipboard button
-    const importClipBtn = document.createElement('button');
-    importClipBtn.innerHTML = '📋 Import from Clipboard';
-    importClipBtn.style.cssText = `
-        background: linear-gradient(135deg, #1976d2, #1565c0);
-        color: white;
-        border: none;
-        padding: 12px 20px;
-        border-radius: 25px;
-        font-weight: 600;
-        cursor: pointer;
-        box-shadow: 0 4px 12px rgba(25, 118, 210, 0.3);
-        transition: all 0.3s ease;
-    `;
-    
-    importClipBtn.addEventListener('mouseover', () => {
-        importClipBtn.style.transform = 'translateY(-2px)';
-        importClipBtn.style.boxShadow = '0 6px 16px rgba(25, 118, 210, 0.4)';
-    });
-    
-    importClipBtn.addEventListener('mouseout', () => {
-        importClipBtn.style.transform = 'translateY(0)';
-        importClipBtn.style.boxShadow = '0 4px 12px rgba(25, 118, 210, 0.3)';
-    });
-    
-    importClipBtn.addEventListener('click', loadFromClipboard);
-    
-    // Add elements to container
-    container.appendChild(fileInput);
-    container.appendChild(importFileBtn);
-    container.appendChild(importClipBtn);
-    
-    document.body.appendChild(container);
-    console.log('✅ Import buttons created');
-}
-
-// Auto-create buttons when page loads
-setTimeout(createImportButtons, 3000); // Wait for Gradio to load
-</script>
 """
 
 print("🚀 Loading Venice Edition NSFW Enhancer... v2.1")
@@ -753,8 +508,266 @@ def enhance_text(original_text, seo_keywords="", style="Professional", male_supp
         error_time = time.time() - start_time
         return f"❌ Error after {error_time:.1f}s: {str(e)[:300]}..."
 
+# JavaScript for import functionality
+IMPORT_JS = """
+<script>
+// Venice Edition Import Functionality
+function fillVeniceEditionFields(data) {
+    console.log('🎯 Filling Venice Edition fields with data:', data);
+    
+    if (!data || !data.data) {
+        console.error('❌ Invalid data format');
+        return false;
+    }
+    
+    const joyCaptionData = data.data;
+    let fieldsFound = 0;
+    let fieldsFilled = 0;
+    
+    // Find all textareas and inputs in Venice Edition
+    const textFields = document.querySelectorAll('textarea, input[type="text"]');
+    fieldsFound = textFields.length;
+    
+    console.log(`📝 Found ${fieldsFound} text fields in Venice Edition`);
+    
+    // Function to fill a field by matching placeholder or context
+    function fillFieldByContext(searchTerms, value, fieldType = 'general') {
+        for (const field of textFields) {
+            const placeholder = (field.placeholder || '').toLowerCase();
+            const label = getFieldLabel(field).toLowerCase();
+            const context = (placeholder + ' ' + label).toLowerCase();
+            
+            // Check if any search term matches
+            const matches = searchTerms.some(term => context.includes(term.toLowerCase()));
+            
+            if (matches && !field.value.trim()) { // Only fill if empty
+                field.value = value;
+                field.dispatchEvent(new Event('input', { bubbles: true }));
+                field.dispatchEvent(new Event('change', { bubbles: true }));
+                fieldsFilled++;
+                console.log(`✅ Filled ${fieldType}: ${value.substring(0, 50)}...`);
+                return true;
+            }
+        }
+        return false;
+    }
+    
+    // Fill fields based on JoyCaption data
+    if (joyCaptionData.caption_engaging) {
+        fillFieldByContext(['original', 'description', 'paste your image'], joyCaptionData.caption_engaging, 'Engaging Caption');
+    }
+    
+    if (joyCaptionData.caption_casual_friend && !fieldsFilled) {
+        fillFieldByContext(['original', 'description', 'paste your image'], joyCaptionData.caption_casual_friend, 'Casual Caption');
+    }
+    
+    if (joyCaptionData.caption_keywords && !fieldsFilled) {
+        fillFieldByContext(['original', 'description', 'paste your image'], joyCaptionData.caption_keywords, 'Keywords Caption');
+    }
+    
+    if (joyCaptionData.keywords) {
+        fillFieldByContext(['keyword', 'seo', 'required'], joyCaptionData.keywords, 'Keywords');
+    }
+    
+    if (joyCaptionData.custom_instructions) {
+        fillFieldByContext(['correction', 'manual', 'guidance'], joyCaptionData.custom_instructions, 'Custom Instructions');
+    }
+    
+    // If no specific matches, fill the first few large text areas with captions
+    if (fieldsFilled === 0 && (joyCaptionData.caption_engaging || joyCaptionData.caption_casual_friend || joyCaptionData.caption_keywords)) {
+        const largeTextareas = Array.from(textFields).filter(field => 
+            field.tagName === 'TEXTAREA' && !field.value.trim()
+        );
+        
+        let index = 0;
+        const captions = [joyCaptionData.caption_engaging, joyCaptionData.caption_casual_friend, joyCaptionData.caption_keywords].filter(Boolean);
+        
+        if (captions.length > 0 && largeTextareas[index]) {
+            largeTextareas[index].value = captions[0];
+            largeTextareas[index].dispatchEvent(new Event('input', { bubbles: true }));
+            fieldsFilled++;
+            console.log(`✅ Filled textarea ${index} with caption`);
+        }
+    }
+    
+    console.log(`📊 Filled ${fieldsFilled} out of ${fieldsFound} fields`);
+    return fieldsFilled > 0;
+}
+
+function getFieldLabel(field) {
+    // Try to find associated label
+    if (field.id) {
+        const label = document.querySelector(`label[for="${field.id}"]`);
+        if (label) return label.textContent || '';
+    }
+    
+    // Look in parent elements for text context
+    let parent = field.parentElement;
+    let levels = 0;
+    while (parent && levels < 3) {
+        const labels = parent.querySelectorAll('label, span, div');
+        for (const label of labels) {
+            const text = label.textContent || '';
+            if (text.length > 0 && text.length < 100) {
+                return text;
+            }
+        }
+        parent = parent.parentElement;
+        levels++;
+    }
+    
+    return field.placeholder || '';
+}
+
+function loadFromFile(event) {
+    const file = event.target.files[0];
+    if (!file) return;
+    
+    const reader = new FileReader();
+    reader.onload = function(e) {
+        try {
+            const data = JSON.parse(e.target.result);
+            const success = fillVeniceEditionFields(data);
+            
+            if (success) {
+                alert(`✅ Successfully imported and filled Venice Edition fields!`);
+            } else {
+                alert('⚠️ Data imported but no matching fields found. Check field labels.');
+            }
+        } catch (error) {
+            console.error('❌ Import error:', error);
+            alert('❌ Failed to import file: ' + error.message);
+        }
+    };
+    reader.readAsText(file);
+}
+
+function loadFromClipboard() {
+    navigator.clipboard.readText().then(text => {
+        try {
+            const data = JSON.parse(text);
+            const success = fillVeniceEditionFields(data);
+            
+            if (success) {
+                alert(`✅ Successfully imported from clipboard and filled Venice Edition fields!`);
+            } else {
+                alert('⚠️ Data imported but no matching fields found. Check field labels.');
+            }
+        } catch (error) {
+            console.error('❌ Clipboard import error:', error);
+            alert('❌ Failed to import from clipboard: Invalid JSON format');
+        }
+    }).catch(err => {
+        console.error('❌ Clipboard read error:', err);
+        alert('❌ Failed to read clipboard: ' + err.message);
+    });
+}
+
+function createImportButtons() {
+    // Remove existing buttons first
+    const existingContainer = document.getElementById('venice-import-container');
+    if (existingContainer) existingContainer.remove();
+    
+    // Create import button container
+    const container = document.createElement('div');
+    container.id = 'venice-import-container';
+    container.style.cssText = `
+        position: fixed;
+        top: 20px;
+        right: 20px;
+        z-index: 9999;
+        display: flex;
+        flex-direction: column;
+        gap: 10px;
+    `;
+    
+    // Create file input (hidden)
+    const fileInput = document.createElement('input');
+    fileInput.type = 'file';
+    fileInput.accept = '.json';
+    fileInput.style.display = 'none';
+    fileInput.addEventListener('change', loadFromFile);
+    
+    // Create import from file button
+    const importFileBtn = document.createElement('button');
+    importFileBtn.innerHTML = '📁 Import from File';
+    importFileBtn.style.cssText = `
+        background: linear-gradient(135deg, #2e7d32, #388e3c);
+        color: white;
+        border: none;
+        padding: 12px 20px;
+        border-radius: 25px;
+        font-weight: 600;
+        cursor: pointer;
+        box-shadow: 0 4px 12px rgba(46, 125, 50, 0.3);
+        transition: all 0.3s ease;
+    `;
+    
+    importFileBtn.addEventListener('mouseover', () => {
+        importFileBtn.style.transform = 'translateY(-2px)';
+        importFileBtn.style.boxShadow = '0 6px 16px rgba(46, 125, 50, 0.4)';
+    });
+    
+    importFileBtn.addEventListener('mouseout', () => {
+        importFileBtn.style.transform = 'translateY(0)';
+        importFileBtn.style.boxShadow = '0 4px 12px rgba(46, 125, 50, 0.3)';
+    });
+    
+    importFileBtn.addEventListener('click', () => fileInput.click());
+    
+    // Create import from clipboard button
+    const importClipBtn = document.createElement('button');
+    importClipBtn.innerHTML = '📋 Import from Clipboard';
+    importClipBtn.style.cssText = `
+        background: linear-gradient(135deg, #1976d2, #1565c0);
+        color: white;
+        border: none;
+        padding: 12px 20px;
+        border-radius: 25px;
+        font-weight: 600;
+        cursor: pointer;
+        box-shadow: 0 4px 12px rgba(25, 118, 210, 0.3);
+        transition: all 0.3s ease;
+    `;
+    
+    importClipBtn.addEventListener('mouseover', () => {
+        importClipBtn.style.transform = 'translateY(-2px)';
+        importClipBtn.style.boxShadow = '0 6px 16px rgba(25, 118, 210, 0.4)';
+    });
+    
+    importClipBtn.addEventListener('mouseout', () => {
+        importClipBtn.style.transform = 'translateY(0)';
+        importClipBtn.style.boxShadow = '0 4px 12px rgba(25, 118, 210, 0.3)';
+    });
+    
+    importClipBtn.addEventListener('click', loadFromClipboard);
+    
+    // Add elements to container
+    container.appendChild(fileInput);
+    container.appendChild(importFileBtn);
+    container.appendChild(importClipBtn);
+    
+    document.body.appendChild(container);
+    console.log('✅ Import buttons created and attached to body');
+}
+
+// Multiple attempts to create buttons after Gradio loads
+setTimeout(createImportButtons, 1000);
+setTimeout(createImportButtons, 3000);
+setTimeout(createImportButtons, 5000);
+
+// Also try when DOM changes (Gradio dynamic loading)
+const observer = new MutationObserver(() => {
+    if (!document.getElementById('venice-import-container')) {
+        createImportButtons();
+    }
+});
+observer.observe(document.body, { childList: true, subtree: true });
+</script>
+"""
+
 # Gradio Interface - TWO COLUMNS ONLY
-with gr.Blocks(title="Venice Edition NSFW Enhancer", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Venice Edition NSFW Enhancer", theme=gr.themes.Soft(), head=IMPORT_JS) as demo:
     gr.HTML(TITLE)
     
     with gr.Row():