import gradio as gr
import numpy as np

# ----------------------------------------------------
# Бұл кодтық үлгі - Қосымшаның ЖҰМЫС ІСТЕЙТІНІН тексеруге арналған
# Модельді қосу үшін 4-қадамды қараңыз
# ----------------------------------------------------

# Кездейсоқ (random) болжам функциясы
def classify_trash_bin(image):
    if image is None:
        return {"Empty": 0.5, "Full": 0.5}

    # Кездейсоқ сандар генерациялау (0.01 мен 0.99 арасында)
    rand_val = np.random.rand()

    # Нәтижені Gradio-ға арналған сөздікке айналдыру
    results = {
        "Empty": float(rand_val),
        "Full": float(1 - rand_val)
    }

    return results

# Gradio интерфейсі
iface = gr.Interface(
    fn=classify_trash_bin,
    inputs=gr.Image(type="numpy", label="Қоқыс жәшігінің суретін жүктеңіз"),
    outputs=gr.Label(num_top_classes=2, label="Болжам"),
    title="Қоқыс жәшігінің толуын анықтау (Уақытша тексеру)",
    description="Қосымшаның жұмыс істеп тұрғанын тексеру үшін кездейсоқ нәтиже береді."
)

iface.launch()