app.py

import streamlit as st
from transformers import pipeline
from PIL import Image
import pandas as pd
from huggingface_hub import login

MIN_ACEPTABLE_SCORE = 0.1
MAX_N_LABELS = 3

import os

HF_TOKEN = os.environ.get("HF_TOKEN")
if HF_TOKEN:
    login(token=HF_TOKEN)

TRANSLATIONS = {
    "en-uk": {
        "title": "Image Classification - Compare All Models",
        "description": "🧪 Run **all** Hugging Face image classification models on a single image and compare results in an interactive table.",
        "upload": "Upload Image",
        "run_button": "Run All Models",
        "results": "Results",
        "download": "Download CSV",
        "columns": {
            "Model": "Model", "Category": "Category", "Label 1": "Label 1", "Score 1": "Score 1",
            "Label 2": "Label 2", "Score 2": "Score 2", "Label 3": "Label 3", "Score 3": "Score 3",
        },
        "lang_name": "English"
    },
    "pt-pt": {
        "title": "Classificação de Imagens - Compara Todos os Modelos",
        "description": "🧪 Executa **todos os modelos** de classificação de imagens do Hugging Face numa só imagem e compara os resultados numa tabela interativa.",
        "upload": "Carregar Imagem",
        "run_button": "Executar Todos os Modelos",
        "results": "Resultados",
        "download": "Descarregar CSV",
        "columns": {
            "Model": "Modelo", "Category": "Categoria", "Label 1": "Label 1", "Score 1": "Score 1",
            "Label 2": "Label 2", "Score 2": "Score 2", "Label 3": "Label 3", "Score 3": "Score 3",
        },
        "lang_name": "Português (PT)"
    },
    "pt-br": {
        "title": "Classificação de Imagens - Compare Todos os Modelos",
        "description": "🧪 Execute **todos os modelos** de classificação de imagens do Hugging Face em uma única imagem e compare os resultados em uma tabela interativa.",
        "upload": "Carregar Imagem",
        "run_button": "Executar Todos os Modelos",
        "results": "Resultados",
        "download": "Baixar CSV",
        "columns": {
            "Model": "Modelo", "Category": "Categoria", "Label 1": "Label 1", "Score 1": "Score 1",
            "Label 2": "Label 2", "Score 2": "Score 2", "Label 3": "Label 3", "Score 3": "Score 3",
        },
        "lang_name": "Português (BR)"
    },
    "zh-cn": {
        "title": "图像分类 - 比较所有模型",
        "description": "🧪 在单张图像上运行**所有**Hugging Face图像分类模型，并在交互式表格中比较结果。",
        "upload": "上传图片",
        "run_button": "运行所有模型",
        "results": "结果",
        "download": "下载CSV",
        "columns": {
            "Model": "模型", "Category": "类别", "Label 1": "标签1", "Score 1": "分数1",
            "Label 2": "标签2", "Score 2": "分数2", "Label 3": "标签3", "Score 3": "分数3",
        },
        "lang_name": "中文"
    },
    "hi": {
        "title": "छवि वर्गीकरण - सभी मॉडल की तुलना करें",
        "description": "🧪 एक ही छवि पर **सभी** Hugging Face छवि वर्गीकरण मॉडल चलाएं और परिणामों की एक इंटरैक्टिव टेबल में तुलना करें।",
        "upload": "छवि अपलोड करें",
        "run_button": "सभी मॉडल चलाएं",
        "results": "परिणाम",
        "download": "CSV डाउनलोड करें",
        "columns": {
            "Model": "मॉडल", "Category": "श्रेणी", "Label 1": "लेबल 1", "Score 1": "स्कोर 1",
            "Label 2": "लेबल 2", "Score 2": "स्कोर 2", "Label 3": "लेबल 3", "Score 3": "स्कोर 3",
        },
        "lang_name": "हिन्दी"
    },
    "es": {
        "title": "Clasificación de Imágenes - Compara Todos los Modelos",
        "description": "🧪 Ejecuta **todos** los modelos de clasificación de imágenes de Hugging Face en una sola imagen y compara los resultados en una tabla interactiva.",
        "upload": "Subir Imagen",
        "run_button": "Ejecutar Todos los Modelos",
        "results": "Resultados",
        "download": "Descargar CSV",
        "columns": {
            "Model": "Modelo", "Category": "Categoría", "Label 1": "Etiqueta 1", "Score 1": "Puntuación 1",
            "Label 2": "Etiqueta 2", "Score 2": "Puntuación 2", "Label 3": "Etiqueta 3", "Score 3": "Puntuación 3",
        },
        "lang_name": "Español"
    },
    "ar": {
        "title": "تصنيف الصور - قارن جميع النماذج",
        "description": "🧪 شغّل **جميع** نماذج تصنيف صور Hugging Face على صورة واحدة وقارن النتائج في جدول تفاعلي.",
        "upload": "تحميل صورة",
        "run_button": "تشغيل جميع النماذج",
        "results": "النتائج",
        "download": "تحميل CSV",
        "columns": {
            "Model": "النموذج", "Category": "الفئة", "Label 1": "التسمية 1", "Score 1": "النتيجة 1",
            "Label 2": "التسمية 2", "Score 2": "النتيجة 2", "Label 3": "التسمية 3", "Score 3": "النتيجة 3",
        },
        "lang_name": "العربية"
    },
    "fr": {
        "title": "Classification d'Images - Comparer Tous les Modèles",
        "description": "🧪 Exécutez **tous** les modèles de classification d'images Hugging Face sur une seule image et comparez les résultats dans un tableau interactif.",
        "upload": "Télécharger une Image",
        "run_button": "Exécuter Tous les Modèles",
        "results": "Résultats",
        "download": "Télécharger CSV",
        "columns": {
            "Model": "Modèle", "Category": "Catégorie", "Label 1": "Étiquette 1", "Score 1": "Score 1",
            "Label 2": "Étiquette 2", "Score 2": "Score 2", "Label 3": "Étiquette 3", "Score 3": "Score 3",
        },
        "lang_name": "Français"
    },
    "bn": {
        "title": "ছবি শ্রেণীবিভাগ - সব মডেল তুলনা করুন",
        "description": "🧪 একটি একক ছবিতে **সমস্ত** Hugging Face ছবি শ্রেণীবিভাগ মডেল চালান এবং একটি ইন্টারেক্টিভ টেবিলে ফলাফল তুলনা করুন।",
        "upload": "ছবি আপলোড করুন",
        "run_button": "সব মডেল চালান",
        "results": "ফলাফল",
        "download": "CSV ডাউনলোড করুন",
        "columns": {
            "Model": "মডেল", "Category": "বিভাগ", "Label 1": "লেবেল 1", "Score 1": "স্কোর 1",
            "Label 2": "লেবেল 2", "Score 2": "স্কোর 2", "Label 3": "লেবেল 3", "Score 3": "স্কোর 3",
        },
        "lang_name": "বাংলা"
    },
    "ru": {
        "title": "Классификация изображений - Сравните все модели",
        "description": "🧪 Запустите **все** модели классификации изображений Hugging Face на одном изображении и сравните результаты в интерактивной таблице.",
        "upload": "Загрузить изображение",
        "run_button": "Запустить все модели",
        "results": "Результаты",
        "download": "Скачать CSV",
        "columns": {
            "Model": "Модель", "Category": "Категория", "Label 1": "Метка 1", "Score 1": "Оценка 1",
            "Label 2": "Метка 2", "Score 2": "Оценка 2", "Label 3": "Метка 3", "Score 3": "Оценка 3",
        },
        "lang_name": "Русский"
    },
    "id": {
        "title": "Klasifikasi Gambar - Bandingkan Semua Model",
        "description": "🧪 Jalankan **semua** model klasifikasi gambar Hugging Face pada satu gambar dan bandingkan hasilnya dalam tabel interaktif.",
        "upload": "Unggah Gambar",
        "run_button": "Jalankan Semua Model",
        "results": "Hasil",
        "download": "Unduh CSV",
        "columns": {
            "Model": "Model", "Category": "Kategori", "Label 1": "Label 1", "Score 1": "Skor 1",
            "Label 2": "Label 2", "Score 2": "Skor 2", "Label 3": "Label 3", "Score 3": "Skor 3",
        },
        "lang_name": "Bahasa Indonesia"
    },
    "de": {
        "title": "Bildklassifizierung - Alle Modelle vergleichen",
        "description": "🧪 Führen Sie **alle** Hugging Face Bildklassifizierungsmodelle auf einem einzigen Bild aus und vergleichen Sie die Ergebnisse in einer interaktiven Tabelle.",
        "upload": "Bild hochladen",
        "run_button": "Alle Modelle ausführen",
        "results": "Ergebnisse",
        "download": "CSV herunterladen",
        "columns": {
            "Model": "Modell", "Category": "Kategorie", "Label 1": "Label 1", "Score 1": "Bewertung 1",
            "Label 2": "Label 2", "Score 2": "Bewertung 2", "Label 3": "Label 3", "Score 3": "Bewertung 3",
        },
        "lang_name": "Deutsch"
    },
    "nl": {
        "title": "Afbeeldingsclassificatie - Vergelijk Alle Modellen",
        "description": "🧪 Voer **alle** Hugging Face-afbeeldingsclassificatiemodellen uit op één enkele afbeelding en vergelijk de resultaten in een interactieve tabel.",
        "upload": "Afbeelding uploaden",
        "run_button": "Alle modellen uitvoeren",
        "results": "Resultaten",
        "download": "CSV downloaden",
        "columns": {
            "Model": "Model", "Category": "Categorie", "Label 1": "Label 1", "Score 1": "Score 1",
            "Label 2": "Label 2", "Score 2": "Score 2", "Label 3": "Label 3", "Score 3": "Score 3",
        },
        "lang_name": "Nederlands"
    },
    "it": {
        "title": "Classificazione Immagini - Confronta Tutti i Modelli",
        "description": "🧪 Esegui **tutti** i modelli di classificazione immagini Hugging Face su una singola immagine e confronta i risultati in una tabella interattiva.",
        "upload": "Carica Immagine",
        "run_button": "Esegui Tutti i Modelli",
        "results": "Risultati",
        "download": "Scarica CSV",
        "columns": {
            "Model": "Modello", "Category": "Categoria", "Label 1": "Etichetta 1", "Score 1": "Punteggio 1",
            "Label 2": "Etichetta 2", "Score 2": "Punteggio 2", "Label 3": "Etichetta 3", "Score 3": "Punteggio 3",
        },
        "lang_name": "Italiano"
    },
    "pl": {
        "title": "Klasyfikacja Obrazów - Porównaj Wszystkie Modele",
        "description": "🧪 Uruchom **wszystkie** modele klasyfikacji obrazów Hugging Face na jednym obrazie i porównaj wyniki w tabeli interaktywnej.",
        "upload": "Prześlij obraz",
        "run_button": "Uruchom wszystkie modele",
        "results": "Wyniki",
        "download": "Pobierz CSV",
        "columns": {
            "Model": "Model", "Category": "Kategoria", "Label 1": "Etykieta 1", "Score 1": "Wynik 1",
            "Label 2": "Etykieta 2", "Score 2": "Wynik 2", "Label 3": "Etykieta 3", "Score 3": "Wynik 3",
        },
        "lang_name": "Polski"
    },
    "ro": {
        "title": "Clasificarea Imaginilor - Compară Toate Modelele",
        "description": "🧪 Rulează **toate** modelele de clasificare a imaginilor Hugging Face pe o singură imagine și compară rezultatele într-un tabel interactiv.",
        "upload": "Încărcați imaginea",
        "run_button": "Rulează toate modelele",
        "results": "Rezultate",
        "download": "Descarcă CSV",
        "columns": {
            "Model": "Model", "Category": "Categorie", "Label 1": "Etichetă 1", "Score 1": "Scor 1",
            "Label 2": "Etichetă 2", "Score 2": "Scor 2", "Label 3": "Etichetă 3", "Score 3": "Scor 3",
        },
        "lang_name": "Română"
    },
    "hu": {
        "title": "Képbesorolás - Összes Modell Összehasonlítása",
        "description": "🧪 Futtassa **összes** Hugging Face képosztályozási modellt egyetlen képen, és hasonlítsa össze az eredményeket egy interaktív táblázatban.",
        "upload": "Kép feltöltése",
        "run_button": "Összes modell futtatása",
        "results": "Eredmények",
        "download": "CSV letöltése",
        "columns": {
            "Model": "Modell", "Category": "Kategoria", "Label 1": "Címke 1", "Score 1": "Pontszám 1",
            "Label 2": "Címke 2", "Score 2": "Pontszám 2", "Label 3": "Címke 3", "Score 3": "Pontszám 3",
        },
        "lang_name": "Magyar"
    },
    "cs": {
        "title": "Klasifikace obrázků - Porovnejte všechny modely",
        "description": "🧪 Spusťte **všechny** modely klasifikace obrázků Hugging Face na jednom obrázku a porovnejte výsledky v interaktivní tabulce.",
        "upload": "Nahrát obrázek",
        "run_button": "Spustit všechny modely",
        "results": "Výsledky",
        "download": "Stáhnout CSV",
        "columns": {
            "Model": "Model", "Category": "Kategorie", "Label 1": "Štítek 1", "Score 1": "Skóre 1",
            "Label 2": "Štítek 2", "Score 2": "Skóre 2", "Label 3": "Štítek 3", "Score 3": "Skóre 3",
        },
        "lang_name": "Čeština"
    },
    "sk": {
        "title": "Klasifikácia obrázkov - Porovnajte všetky modely",
        "description": "🧪 Spustite **všetky** modely klasifikácie obrázkov Hugging Face na jednom obrázku a porovnajte výsledky v interaktívnej tabuľke.",
        "upload": "Nahrať obrázok",
        "run_button": "Spustiť všetky modely",
        "results": "Výsledky",
        "download": "Stiahnuť CSV",
        "columns": {
            "Model": "Model", "Category": "Kategória", "Label 1": "Štítok 1", "Score 1": "Skóre 1",
            "Label 2": "Štítok 2", "Score 2": "Skóre 2", "Label 3": "Štítok 3", "Score 3": "Skóre 3",
        },
        "lang_name": "Slovenčina"
    },
    "sl": {
        "title": "Klasifikacija slik - Primerjajte vse modele",
        "description": "🧪 Zaženite **vse** modele klasifikacije slik Hugging Face na eni sliki in primerjajte rezultate v interaktivni tabeli.",
        "upload": "Naloži sliko",
        "run_button": "Zaženi vse modele",
        "results": "Rezultati",
        "download": "Prenesi CSV",
        "columns": {
            "Model": "Model", "Category": "Kategorije", "Label 1": "Oznaka 1", "Score 1": "Točke 1",
            "Label 2": "Oznaka 2", "Score 2": "Točke 2", "Label 3": "Oznaka 3", "Score 3": "Točke 3",
        },
        "lang_name": "Slovenščina"
    },
    "da": {
        "title": "Billedklassifikation - Sammenlign Alle Modeller",
        "description": "🧪 Kør **alle** Hugging Face billedklassifikationsmodeller på et enkelt billede og sammenlign resultaterne i en interaktiv tabel.",
        "upload": "Upload billede",
        "run_button": "Kør alle modeller",
        "results": "Resultater",
        "download": "Download CSV",
        "columns": {
            "Model": "Model", "Category": "Kategori", "Label 1": "Etiket 1", "Score 1": "Point 1",
            "Label 2": "Etiket 2", "Score 2": "Point 2", "Label 3": "Etiket 3", "Score 3": "Point 3",
        },
        "lang_name": "Dansk"
    },
    "sv": {
        "title": "Bildklassificering - Jämför Alla Modeller",
        "description": "🧪 Kör **alla** Hugging Face bildklassificeringsmodeller på en enda bild och jämför resultaten i en interaktiv tabell.",
        "upload": "Ladda upp bild",
        "run_button": "Kör alla modeller",
        "results": "Resultat",
        "download": "Ladda ner CSV",
        "columns": {
            "Model": "Modell", "Category": "Kategori", "Label 1": "Etikett 1", "Score 1": "Poäng 1",
            "Label 2": "Etikett 2", "Score 2": "Poäng 2", "Label 3": "Etikett 3", "Score 3": "Poäng 3",
        },
        "lang_name": "Svenska"
    },
    "fi": {
        "title": "Kuvaluokittelu - Vertaa Kaikkia Malleja",
        "description": "🧪 Suorita **kaikki** Hugging Face -kuvaluokittelumallit yhdellä kuvalla ja vertaa tuloksia vuorovaikutteisessa taulukossa.",
        "upload": "Lataa kuva",
        "run_button": "Suorita kaikki mallit",
        "results": "Tulokset",
        "download": "Lataa CSV",
        "columns": {
            "Model": "Malli", "Category": "Luokka", "Label 1": "Tunniste 1", "Score 1": "Pisteet 1",
            "Label 2": "Tunniste 2", "Score 2": "Pisteet 2", "Label 3": "Tunniste 3", "Score 3": "Pisteet 3",
        },
        "lang_name": "Suomi"
    },
    "el": {
        "title": "Ταξινόμηση Εικόνων - Σύγκριση Όλων των Μοντέλων",
        "description": "🧪 Εκτελέστε **όλα** τα μοντέλα ταξινόμησης εικόνων του Hugging Face σε μία εικόνα και συγκρίνετε τα αποτελέσματα σε έναν διαδραστικό πίνακα.",
        "upload": "Μεταφόρτωση εικόνας",
        "run_button": "Εκτέλεση όλων των μοντέλων",
        "results": "Αποτελέσματα",
        "download": "Λήψη CSV",
        "columns": {
            "Model": "Μοντέλο", "Category": "Κατηγορία", "Label 1": "Ετικέτα 1", "Score 1": "Βαθμολογία 1",
            "Label 2": "Ετικέτα 2", "Score 2": "Βαθμολογία 2", "Label 3": "Ετικέτα 3", "Score 3": "Βαθμολογία 3",
        },
        "lang_name": "Ελληνικά"
    },
    "bg": {
        "title": "Класификация на изображения - Сравнете всички модели",
        "description": "🧪 Стартирайте **всички** модели за класификация на изображения от Hugging Face на едно изображение и сравнете резултатите в интерактивна таблица.",
        "upload": "Качете изображение",
        "run_button": "Стартирайте всички модели",
        "results": "Резултати",
        "download": "Изтеглете CSV",
        "columns": {
            "Model": "Модел", "Category": "Категория", "Label 1": "Етикет 1", "Score 1": "Резултат 1",
            "Label 2": "Етикет 2", "Score 2": "Резултат 2", "Label 3": "Етикет 3", "Score 3": "Резултат 3",
        },
        "lang_name": "Български"
    },
    "lv": {
        "title": "Attēlu klasifikācija - Salīdziniet visus modeļus",
        "description": "🧪 Palaidiet **visus** Hugging Face attēlu klasifikācijas modeļus uz viena attēla un salīdziniet rezultātus interaktīvā tabulā.",
        "upload": "Augšupielādēt attēlu",
        "run_button": "Palaist visus modeļus",
        "results": "Rezultāti",
        "download": "Lejupielādēt CSV",
        "columns": {
            "Model": "Modelis", "Category": "Kategorijas", "Label 1": "Etiķete 1", "Score 1": "Rezultāts 1",
            "Label 2": "Etiķete 2", "Score 2": "Rezultāts 2", "Label 3": "Etiķete 3", "Score 3": "Rezultāts 3",
        },
        "lang_name": "Latviešu"
    },
    "lt": {
        "title": "Vaizdų klasifikavimas - Palyginkite visus modelius",
        "description": "🧪 Paleiskite **visus** Hugging Face vaizdų klasifikavimo modelius vienoje nuotraukoje ir palyginkite rezultatus interaktyvioje lentelėje.",
        "upload": "Įkelti vaizdą",
        "run_button": "Paleisti visus modelius",
        "results": "Rezultatai",
        "download": "Atsisiųsti CSV",
        "columns": {
            "Model": "Modelis", "Category": "Kategorijos", "Label 1": "Etiketė 1", "Score 1": "Rezultatas 1",
            "Label 2": "Etiketė 2", "Score 2": "Rezultatas 2", "Label 3": "Etiketė 3", "Score 3": "Rezultatas 3",
        },
        "lang_name": "Lietuvių"
    },
    "et": {
        "title": "Pildiklassifikatsioon - Võrdle kõiki mudeleid",
        "description": "🧪 Käivitage **kõik** Hugging Face pildiklassifikatsiooni mudelid ühel pildil ja võrrelge tulemusi interaktiivses tabelis.",
        "upload": "Laadi pilt üles",
        "run_button": "Käivita kõik mudelid",
        "results": "Tulemused",
        "download": "Laadi CSV alla",
        "columns": {
            "Model": "Mudel", "Category": "Kategooria", "Label 1": "Silt 1", "Score 1": "Skoor 1",
            "Label 2": "Silt 2", "Score 2": "Skoor 2", "Label 3": "Silt 3", "Score 3": "Skoor 3",
        },
        "lang_name": "Eesti"
    },
    "ga": {
        "title": "Rangú Imagacha - Déan Comparáid idir na Samhlaithe go léir",
        "description": "🧪 Rith **gach** samhail rangaithe imeagacha Hugging Face ar aon íomhá amháin agus déan comparáid ar thorthaí i dtábla idirghníomhach.",
        "upload": "Uaslódáil Íomhá",
        "run_button": "Rith gach Samhail",
        "results": "Torthaí",
        "download": "Íosluchtaigh CSV",
        "columns": {
            "Model": "Samhail", "Category": "Catagóir", "Label 1": "Lipéad 1", "Score 1": "Scór 1",
            "Label 2": "Lipéad 2", "Score 2": "Scór 2", "Label 3": "Lipéad 3", "Score 3": "Scór 3",
        },
        "lang_name": "Gaeilge"
    },
    "mt": {
        "title": "Klassifikazzjoni tal-Immaġnijiet - Qabbel il-Mudelli kollha",
        "description": "🧪 ŻRuns **il-mudelli kollha** ta' klassifikazzjoni tal-immaġnijiet ta' Hugging Face fuq immaġni waħda u qabbel ir-rizultati f'tabella interattiva.",
        "upload": "Upload Image",
        "run_button": "Run All Models",
        "results": "Risultati",
        "download": "Download CSV",
        "columns": {
            "Model": "Mudell", "Category": "Kategoriji", "Label 1": "Tikketta 1", "Score 1": "Punteġġ 1",
            "Label 2": "Tikketta 2", "Score 2": "Punteġġ 2", "Label 3": "Tikketta 3", "Score 3": "Punteġġ 3",
        },
        "lang_name": "Malti"
    },
    "hr": {
        "title": "Klasifikacija slika - Usporedite sve modele",
        "description": "🧪 Pokrenite **sve** modele klasifikacije slika Hugging Face na jednoj slici i usporedite rezultate u interaktivnoj tablici.",
        "upload": "Prenesi sliku",
        "run_button": "Pokreni sve modele",
        "results": "Rezultati",
        "download": "Preuzmi CSV",
        "columns": {
            "Model": "Model", "Category": "Kategorije", "Label 1": "Oznaka 1", "Score 1": "Rezultat 1",
            "Label 2": "Oznaka 2", "Score 2": "Rezultat 2", "Label 3": "Oznaka 3", "Score 3": "Rezultat 3",
        },
        "lang_name": "Hrvatski"
    },
}

lang = st.selectbox("Language / Idioma", list(TRANSLATIONS.keys()), format_func=lambda x: TRANSLATIONS[x]["lang_name"])
t = TRANSLATIONS[lang]

MODELS = [
            ("google/vit-base-patch16-224", "General Image Classification"),
            ("microsoft/resnet-50", "General Image Classification"),
            ("microsoft/resnet-18", "General Image Classification"),
            ("microsoft/resnet-34", "General Image Classification"),
            ("microsoft/resnet-101", "General Image Classification"),
            ("microsoft/swin-tiny-patch4-window7-224", "General Image Classification"),
            ("microsoft/swinv2-base-patch4-window16-256", "General Image Classification"),
            ("facebook/convnext-large-224", "General Image Classification"),
            ("facebook/convnext-base-224-22k-1k", "General Image Classification"),
            ("facebook/convnext-tiny-224", "General Image Classification"),
            ("timm/resnet50.a1_in1k", "General Image Classification"),
            ("timm/tf_efficientnetv2_s.in21k", "General Image Classification"),
            ("timm/convnext_tiny.fb_in22k", "General Image Classification"),
            ("google/vit-base-patch16-224-in21k", "General Image Classification"),
            ("facebook/deit-base-distilled-patch16-224", "General Image Classification"),
            ("WinKawaks/vit-tiny-patch16-224", "General Image Classification"),

            ("nateraw/vit-age-classifier", "Age Classification"),

            ("Falconsai/nsfw_image_detection", "NSFW Detection"),
            ("carbon225/vit-base-patch16-224-hentai", "NSFW Detection"),

            ("cafeai/cafe_aesthetic", "Aesthetic/Art Classification"),

            ("trpakov/vit-face-expression", "Face/Emotion Classification"),
            ("RickyIG/emotion_face_image_classification", "Face/Emotion Classification"),
            ("rizvandwiki/gender-classification", "Face/Emotion Classification"),

            ("Ateeqq/ai-vs-human-image-detector", "AI vs Human Detection"),
            ("umm-maybe/AI-image-detector", "AI vs Human Detection"),

            ("DunnBC22/vit-base-patch16-224-in21k_Human_Activity_Recognition", "Human Activity Recognition"),

            ("andupets/real-estate-image-classification", "Real Estate"),

            ("lamnt2008/car_brands_classification", "Car Classification"),

            ("docling-project/DocumentFigureClassifier-v2.5", "Document Classification"),

            ("timm/efficientnet_b0.ra_in1k", "EfficientNet"),
            ("timm/mobilenetv3_large_100.ra_in1k", "EfficientNet"),
            ("timm/mobilenetv3_small_100.lamb_in1k", "EfficientNet"),
        ]

def classify(image, model_name):
    classifier = pipeline("image-classification", model=model_name)
    result = classifier(image)
    return result

def format_results(results):
    formatted = []
    for r in results[:MAX_N_LABELS]:
        if r['score'] >= MIN_ACEPTABLE_SCORE:
            formatted.append({
                "label": r['label'],
                "score": f"{r['score']:.2%}"
            })
    while len(formatted) < 3:
        formatted.append({"label": "-", "score": "-"})
    return formatted

def main():
    st.title(t["title"])
    st.markdown(t["description"])
    
    bmc_link = "https://www.buymeacoffee.com/nuno.tome" 
    image_url = "https://i.giphy.com/RETzc1mj7HpZPuNf3e.webp" 
    image_link_markdown = f"[![Buy Me a Coffee]({image_url})]({bmc_link})"
    st.markdown(image_link_markdown, unsafe_allow_html=True)
  
    input_image = st.file_uploader(t["upload"])

    if input_image is not None:
        image_to_classify = Image.open(input_image)
        st.image(image_to_classify, caption="Uploaded Image", use_column_width=True)
        
        if st.button(t["run_button"], type="primary"):
            results_data = []
            progress_bar = st.progress(0)
            status_text = st.empty()
            
            for i, (model_name, category) in enumerate(MODELS):
                status_text.text(f"Running model {i+1}/{len(MODELS)}: {model_name}")
                try:
                    classification_result = classify(image_to_classify, model_name)
                    formatted = format_results(classification_result)
                    results_data.append({
                        "Model": model_name,
                        "Category": category,
                        "Label 1": formatted[0]["label"],
                        "Score 1": formatted[0]["score"],
                        "Label 2": formatted[1]["label"],
                        "Score 2": formatted[1]["score"],
                        "Label 3": formatted[2]["label"],
                        "Score 3": formatted[2]["score"],
                    })
                except Exception as e:
                    results_data.append({
                        "Model": model_name,
                        "Category": category,
                        "Label 1": f"Error: {str(e)[:50]}",
                        "Score 1": "-",
                        "Label 2": "-",
                        "Score 2": "-",
                        "Label 3": "-",
                        "Score 3": "-",
                    })
                progress_bar.progress((i + 1) / len(MODELS))
            
            status_text.text("Done!")
            
            if results_data:
                df = pd.DataFrame(results_data)
                df = df.rename(columns=t["columns"])
                st.subheader(f"{t['results']} ({len(results_data)} models)")
                
                st.markdown("""
                <style>
                .dataframe {font-size: 12px;}
                </style>
                """, unsafe_allow_html=True)
                
                st.dataframe(
                    df,
                    use_container_width=True,
                    hide_index=True,
                    column_config={
                        t["columns"]["Model"]: st.column_config.TextColumn(t["columns"]["Model"], width="medium"),
                        t["columns"]["Category"]: st.column_config.TextColumn(t["columns"]["Category"], width="small"),
                        t["columns"]["Label 1"]: st.column_config.TextColumn(t["columns"]["Label 1"], width="medium"),
                        t["columns"]["Score 1"]: st.column_config.TextColumn(t["columns"]["Score 1"], width="small"),
                        t["columns"]["Label 2"]: st.column_config.TextColumn(t["columns"]["Label 2"], width="medium"),
                        t["columns"]["Score 2"]: st.column_config.TextColumn(t["columns"]["Score 2"], width="small"),
                        t["columns"]["Label 3"]: st.column_config.TextColumn(t["columns"]["Label 3"], width="medium"),
                        t["columns"]["Score 3"]: st.column_config.TextColumn(t["columns"]["Score 3"], width="small"),
                    }
                )
                
                csv = df.to_csv(index=False).encode('utf-8')
                st.download_button(
                    t["download"],
                    csv,
                    "classification_results.csv",
                    "text/csv",
                    key='download-csv'
                )


if __name__ == "__main__":
    main()