Update app.py

app.py

@@ -3,87 +3,495 @@ import requests
 import random
 
 def get_random_dataset(min_likes=None, max_likes=None, task=None, language=None):
-    """Get a single random dataset from HuggingFace with optional filters"""
-    
     try:
-        # Build API request
         url = "https://huggingface.co/api/datasets"
         params = {"limit": 100, "full": "true"}
         
-        # Add task filter if provided
-        if task:
-            params["filter"] = task
+        if task and task.strip():
+            params["filter"] = task.strip()
         
-        # Fetch datasets
         response = requests.get(url, params=params, timeout=30)
         response.raise_for_status()
         datasets = response.json()
         
-        # Apply filters
         filtered = []
         for ds in datasets:
             likes = ds.get("likes", 0)
             
-            # Check likes range
-            if min_likes and likes < min_likes:
+            if min_likes is not None and likes < min_likes:
                 continue
-            if max_likes and likes > max_likes:
+            if max_likes is not None and likes > max_likes:
                 continue
             
-            # Check language
-            if language:
+            if language and language.strip():
                 langs = ds.get("cardData", {}).get("language", [])
                 if isinstance(langs, str):
                     langs = [langs]
-                if language not in langs:
+                if language.strip() not in langs:
                     continue
             
             filtered.append(ds)
         
         if not filtered:
-            return "No datasets found matching your filters. Try adjusting them."
+            return """
+            <div class='result-container empty-state'>
+                <div class='empty-icon'>
+                    <svg width="64" height="64" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5">
+                        <circle cx="12" cy="12" r="10"/>
+                        <path d="M12 6v6l4 2"/>
+                    </svg>
+                </div>
+                <h3>No Results Found</h3>
+                <p>Try adjusting your filters to discover datasets</p>
+            </div>
+            <script>
+                new Audio('https://assets.mixkit.co/active_storage/sfx/2572/2572-preview.mp3').play().catch(() => {});
+            </script>
+            """
         
-        # Pick random dataset
         dataset = random.choice(filtered)
+        dataset_id = dataset.get('id', 'Unknown')
+        likes = dataset.get('likes', 0)
+        downloads = dataset.get('downloads', 0)
+        author = dataset.get('author', 'Unknown')
+        description = dataset.get('cardData', {}).get('description', 'No description available')
+        tags = dataset.get('tags', [])[:6]
+        
+        tags_html = ''.join([f"<span class='tag'>{tag}</span>" for tag in tags])
+        
+        result = f"""
+        <div class='result-container success-state'>
+            <div class='result-header'>
+                <h2 class='dataset-title'>{dataset_id}</h2>
+                <p class='dataset-author'>by {author}</p>
+            </div>
+            
+            <div class='stats-grid'>
+                <div class='stat-card stat-primary'>
+                    <div class='stat-label'>Likes</div>
+                    <div class='stat-value'>{likes:,}</div>
+                </div>
+                <div class='stat-card stat-secondary'>
+                    <div class='stat-label'>Downloads</div>
+                    <div class='stat-value'>{downloads:,}</div>
+                </div>
+            </div>
+            
+            <div class='description-section'>
+                <h3>Description</h3>
+                <p>{description[:400]}{'...' if len(description) > 400 else ''}</p>
+            </div>
+            
+            <div class='tags-section'>
+                <h3>Tags</h3>
+                <div class='tags-container'>{tags_html if tags_html else '<span class="no-tags">No tags available</span>'}</div>
+            </div>
+            
+            <a href='https://huggingface.co/datasets/{dataset_id}' target='_blank' class='view-button'>
+                View on HuggingFace
+                <svg width="16" height="16" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+                    <path d="M18 13v6a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2V8a2 2 0 0 1 2-2h6"/>
+                    <polyline points="15 3 21 3 21 9"/>
+                    <line x1="10" y1="14" x2="21" y2="3"/>
+                </svg>
+            </a>
+        </div>
+        <script>
+            new Audio('https://assets.mixkit.co/active_storage/sfx/2570/2570-preview.mp3').play().catch(() => {});
+        </script>
+        """
+        return result
         
-        # Format output
-        result = f"""## {dataset.get('id', 'Unknown')}
+    except Exception as e:
+        return f"""
+        <div class='result-container error-state'>
+            <div class='error-icon'>
+                <svg width="64" height="64" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5">
+                    <circle cx="12" cy="12" r="10"/>
+                    <line x1="15" y1="9" x2="9" y2="15"/>
+                    <line x1="9" y1="9" x2="15" y2="15"/>
+                </svg>
+            </div>
+            <h3>Error</h3>
+            <p>{str(e)}</p>
+        </div>
+        <script>
+            new Audio('https://assets.mixkit.co/active_storage/sfx/2573/2573-preview.mp3').play().catch(() => {});
+        </script>
+        """
 
-**Likes:** {dataset.get('likes', 0):,}  
-**Downloads:** {dataset.get('downloads', 0):,}  
-**Author:** {dataset.get('author', 'Unknown')}
+custom_css = """
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap');
 
-**Link:** https://huggingface.co/datasets/{dataset.get('id', '')}
+* {
+    font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
+}
 
-**Description:**  
-{dataset.get('cardData', {}).get('description', 'No description available')[:300]}...
+.gradio-container {
+    max-width: 920px !important;
+    margin: 0 auto !important;
+}
 
-**Tags:** {', '.join(dataset.get('tags', [])[:8])}
-"""
-        return result
-        
-    except Exception as e:
-        return f"Error: {str(e)}"
+body {
+    background: #FAFBFC;
+}
+
+.header-section {
+    text-align: center;
+    padding: 56px 24px 40px;
+    background: linear-gradient(135deg, #1E3A8A 0%, #1E40AF 100%);
+    border-radius: 20px;
+    margin-bottom: 40px;
+    position: relative;
+    overflow: hidden;
+}
+
+.header-section::before {
+    content: '';
+    position: absolute;
+    top: 0;
+    left: 0;
+    right: 0;
+    bottom: 0;
+    background: radial-gradient(circle at 30% 50%, rgba(59, 130, 246, 0.2) 0%, transparent 50%),
+                radial-gradient(circle at 70% 50%, rgba(147, 51, 234, 0.2) 0%, transparent 50%);
+    pointer-events: none;
+}
+
+.main-title {
+    color: white;
+    font-size: 42px;
+    font-weight: 700;
+    margin: 0 0 12px 0;
+    letter-spacing: -1px;
+    position: relative;
+    z-index: 1;
+}
+
+.main-subtitle {
+    color: rgba(255, 255, 255, 0.85);
+    font-size: 17px;
+    margin: 0;
+    font-weight: 400;
+    position: relative;
+    z-index: 1;
+}
+
+.filter-container {
+    background: white;
+    padding: 32px;
+    border-radius: 16px;
+    box-shadow: 0 1px 3px rgba(0, 0, 0, 0.06);
+    margin-bottom: 24px;
+    border: 1px solid #E5E7EB;
+}
+
+.filter-label {
+    color: #1F2937;
+    font-size: 16px;
+    font-weight: 600;
+    margin-bottom: 24px;
+    display: block;
+}
+
+label {
+    color: #374151 !important;
+    font-size: 14px !important;
+    font-weight: 500 !important;
+    margin-bottom: 8px !important;
+}
+
+input[type="number"], input[type="text"] {
+    border: 1.5px solid #D1D5DB !important;
+    border-radius: 10px !important;
+    padding: 12px 14px !important;
+    font-size: 15px !important;
+    transition: all 0.2s ease !important;
+    background: white !important;
+}
+
+input[type="number"]:focus, input[type="text"]:focus {
+    border-color: #3B82F6 !important;
+    box-shadow: 0 0 0 3px rgba(59, 130, 246, 0.1) !important;
+    outline: none !important;
+}
+
+button.primary {
+    background: linear-gradient(135deg, #1E3A8A 0%, #1E40AF 100%) !important;
+    border: none !important;
+    color: white !important;
+    font-size: 16px !important;
+    font-weight: 600 !important;
+    padding: 16px 40px !important;
+    border-radius: 12px !important;
+    cursor: pointer !important;
+    transition: all 0.25s cubic-bezier(0.4, 0, 0.2, 1) !important;
+    box-shadow: 0 4px 12px rgba(30, 58, 138, 0.3) !important;
+    width: 100% !important;
+    margin-top: 8px !important;
+}
+
+button.primary:hover {
+    transform: translateY(-2px) !important;
+    box-shadow: 0 8px 20px rgba(30, 58, 138, 0.4) !important;
+}
+
+button.primary:active {
+    transform: translateY(0px) !important;
+}
+
+.result-container {
+    background: white;
+    border-radius: 16px;
+    padding: 40px;
+    margin-top: 24px;
+    border: 1px solid #E5E7EB;
+    box-shadow: 0 1px 3px rgba(0, 0, 0, 0.06);
+}
+
+.empty-state, .error-state {
+    text-align: center;
+    padding: 60px 40px;
+}
+
+.empty-icon, .error-icon {
+    color: #9CA3AF;
+    margin-bottom: 20px;
+    display: flex;
+    justify-content: center;
+}
 
+.empty-state h3, .error-state h3 {
+    color: #1F2937;
+    font-size: 20px;
+    font-weight: 600;
+    margin: 0 0 8px 0;
+}
 
-with gr.Blocks(title="Random HF Dataset Picker") as app:
-    gr.Markdown("#Random HuggingFace Dataset Picker")
-    gr.Markdown("Pick a completely random dataset with optional filters")
+.empty-state p, .error-state p {
+    color: #6B7280;
+    font-size: 15px;
+    margin: 0;
+}
+
+.success-state {
+    animation: slideIn 0.4s cubic-bezier(0.4, 0, 0.2, 1);
+}
+
+@keyframes slideIn {
+    from {
+        opacity: 0;
+        transform: translateY(20px);
+    }
+    to {
+        opacity: 1;
+        transform: translateY(0);
+    }
+}
+
+.result-header {
+    border-bottom: 1px solid #F3F4F6;
+    padding-bottom: 24px;
+    margin-bottom: 28px;
+}
+
+.dataset-title {
+    color: #111827;
+    margin: 0 0 8px 0;
+    font-size: 26px;
+    font-weight: 700;
+    letter-spacing: -0.5px;
+}
+
+.dataset-author {
+    color: #6B7280;
+    margin: 0;
+    font-size: 15px;
+    font-weight: 500;
+}
+
+.stats-grid {
+    display: grid;
+    grid-template-columns: repeat(2, 1fr);
+    gap: 16px;
+    margin-bottom: 32px;
+}
+
+.stat-card {
+    padding: 20px;
+    border-radius: 12px;
+    text-align: center;
+    transition: transform 0.2s ease;
+}
+
+.stat-card:hover {
+    transform: translateY(-2px);
+}
+
+.stat-primary {
+    background: linear-gradient(135deg, #1E3A8A 0%, #1E40AF 100%);
+}
+
+.stat-secondary {
+    background: linear-gradient(135deg, #0891B2 0%, #06B6D4 100%);
+}
+
+.stat-label {
+    color: rgba(255, 255, 255, 0.85);
+    font-size: 13px;
+    margin-bottom: 6px;
+    font-weight: 500;
+    text-transform: uppercase;
+    letter-spacing: 0.5px;
+}
+
+.stat-value {
+    color: white;
+    font-size: 28px;
+    font-weight: 700;
+}
+
+.description-section, .tags-section {
+    margin-bottom: 28px;
+}
+
+.description-section h3, .tags-section h3 {
+    color: #111827;
+    font-size: 16px;
+    font-weight: 600;
+    margin: 0 0 12px 0;
+}
+
+.description-section p {
+    color: #4B5563;
+    line-height: 1.7;
+    font-size: 15px;
+    margin: 0;
+}
+
+.tags-container {
+    display: flex;
+    flex-wrap: wrap;
+    gap: 8px;
+}
+
+.tag {
+    display: inline-block;
+    background: #EFF6FF;
+    color: #1E40AF;
+    padding: 6px 14px;
+    border-radius: 20px;
+    font-size: 13px;
+    font-weight: 500;
+    border: 1px solid #DBEAFE;
+}
+
+.no-tags {
+    color: #9CA3AF;
+    font-size: 14px;
+}
+
+.view-button {
+    display: inline-flex;
+    align-items: center;
+    gap: 8px;
+    background: linear-gradient(135deg, #1E3A8A 0%, #1E40AF 100%);
+    color: white;
+    padding: 14px 28px;
+    border-radius: 10px;
+    text-decoration: none;
+    font-weight: 600;
+    font-size: 15px;
+    transition: all 0.25s cubic-bezier(0.4, 0, 0.2, 1);
+    box-shadow: 0 2px 8px rgba(30, 58, 138, 0.25);
+}
+
+.view-button:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 4px 12px rgba(30, 58, 138, 0.35);
+    text-decoration: none;
+}
+
+.footer-note {
+    text-align: center;
+    color: #6B7280;
+    font-size: 14px;
+    margin-top: 32px;
+    padding: 20px;
+    font-weight: 400;
+}
+"""
+
+with gr.Blocks(css=custom_css, title="Random Dataset Picker", theme=gr.themes.Soft()) as app:
+    gr.HTML("""
+        <div class='header-section'>
+            <h1 class='main-title'>Random Dataset Picker</h1>
+            <p class='main-subtitle'>Discover datasets from HuggingFace with customisable filters</p>
+        </div>
+    """)
     
-    with gr.Row():
-        min_likes = gr.Number(label="Min Likes (optional)", precision=0)
-        max_likes = gr.Number(label="Max Likes (optional)", precision=0)
+    with gr.Group():
+        gr.HTML("<div class='filter-container'><span class='filter-label'>Filter Options</span></div>")
+        
+        with gr.Row():
+            min_likes = gr.Number(
+                label="Minimum Likes",
+                precision=0,
+                container=True,
+                scale=1
+            )
+            max_likes = gr.Number(
+                label="Maximum Likes",
+                precision=0,
+                container=True,
+                scale=1
+            )
+        
+        with gr.Row():
+            task = gr.Textbox(
+                label="Task Category",
+                placeholder="e.g., text-classification, image-classification",
+                container=True,
+                scale=1
+            )
+            language = gr.Textbox(
+                label="Language Code",
+                placeholder="e.g., en, fr, de, zh",
+                container=True,
+                scale=1
+            )
     
-    task = gr.Textbox(label="Task (optional)", placeholder="e.g., text-classification")
-    language = gr.Textbox(label="Language Code (optional)", placeholder="e.g., en")
+    btn = gr.Button("Generate Random Dataset", variant="primary", size="lg", elem_classes=["generate-btn"])
     
-    btn = gr.Button("Pick Random Dataset", variant="primary")
-    output = gr.Markdown()
+    output = gr.HTML(
+        value="""<div class='result-container empty-state'>
+            <div class='empty-icon'>
+                <svg width="64" height="64" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5">
+                    <path d="M21 16V8a2 2 0 0 0-1-1.73l-7-4a2 2 0 0 0-2 0l-7 4A2 2 0 0 0 3 8v8a2 2 0 0 0 1 1.73l7 4a2 2 0 0 0 2 0l7-4A2 2 0 0 0 21 16z"/>
+                    <polyline points="3.27 6.96 12 12.01 20.73 6.96"/>
+                    <line x1="12" y1="22.08" x2="12" y2="12"/>
+                </svg>
+            </div>
+            <h3>Ready to Explore</h3>
+            <p>Click the button above to discover a random dataset</p>
+        </div>"""
+    )
     
     btn.click(
         fn=get_random_dataset,
         inputs=[min_likes, max_likes, task, language],
         outputs=output
     )
+    
+    gr.HTML("<div class='footer-note'>All filters are optional — Leave empty for completely random selection</div>")
+    
+    gr.HTML("""
+        <script>
+            document.querySelector('.generate-btn').addEventListener('click', function() {
+                const audio = new Audio('https://assets.mixkit.co/active_storage/sfx/2568/2568-preview.mp3');
+                audio.play().catch(() => {});
+            });
+        </script>
+    """)
 
 app.launch()