Reverting

app.py

@@ -1,34 +1,19 @@
-import base64
-from io import BytesIO
-from PIL import Image
-
-from fastapi import FastAPI
-from pydantic import BaseModel
-
+import gradio as gr
 import random
 import torch
-from transformers import T5Tokenizer, T5ForConditionalGeneration, pipeline
-
-# =========================
-# FASTAPI APP
-# =========================
-app = FastAPI()
-
-# =========================
-# API MODELS
-# =========================
-class APIImage(BaseModel):
-    image_base64: str
-
-
-# =========================
-# LOAD CLASSIFIER
-# =========================
+from transformers import (
+    AutoTokenizer, AutoModelForCausalLM,
+    T5Tokenizer,
+    T5ForConditionalGeneration,
+    pipeline
+)
+# ------------------ LOAD CLASSIFIER ------------------
 cls_model_name = "yangy50/garbage-classification"
 classifier = pipeline("image-classification", model=cls_model_name)
 
 def classify_image(image):
     preds = classifier(image)
+
     results = {}
     for item in preds[:3]:
         label = item["label"]
@@ -37,70 +22,205 @@ def classify_image(image):
     return results
 
 
-# ------------------ LOAD CHAT MODEL
-tiny_model = "google/flan-t5-small"
+# ------------------ LOAD CHAT MODEL ------------------
+tiny_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 
-# tokenizer + model
-tokenizer = T5Tokenizer.from_pretrained(tiny_model)
-chat_model = T5ForConditionalGeneration.from_pretrained(tiny_model)
+tokenizer = AutoTokenizer.from_pretrained(tiny_model)
+chat_model = AutoModelForCausalLM.from_pretrained(
+    tiny_model,
+    dtype=torch.bfloat16, 
+    device_map="auto",
+    low_cpu_mem_usage=True)
 
-# Use text2text-generation for T5-style models
 pipe = pipeline(
-    task="text2text-generation",
+    "text-generation",
     model=chat_model,
     tokenizer=tokenizer,
-    device=-1,              # -1 = CPU (safe)
+    device_map="auto",
     max_new_tokens=80
 )
 
+def clean_chat_output(full_text):
+
+    if "<|assistant|>" in full_text:
+        full_text = full_text.split("<|assistant|>")[-1]
+
+    lines = full_text.strip().split("\n")
+    if lines[0].lower().startswith("item:"):
+        lines = lines[1:]
+
+    return "\n".join(lines).strip()
+
 def explain_recycling(class_label):
-    # Construct a simple single-string prompt for T5
-    prompt = (
-        "You are an expert in waste sorting. "
-        "Always answer using exactly two bullet points:\n"
-        "• Recycling type: <Item category>\n"
-        "• Disposal: <clear, detailed correct sentence>\n"
-        f"Item: {class_label}\n"
-        "Return the two bullet points now."
+    system_msg = {
+        "role": "system",
+        "content": (
+            "You are an expert in waste sorting. "
+            "You ALWAYS answer using exactly two bullet points:\n"
+            "• Recycling type: <Item category>\n"
+            "• Disposal: <clear, detailed correct sentence>\n"
+            "No extra text, no introductions, no explanations."
+        )
+    }
+    user_msg = {
+        "role": "user",
+        "content": f"Item: {class_label}\nReturn the two bullet points now."
+    }
+    messages = [system_msg, user_msg]
+    
+    prompt = tokenizer.apply_chat_template(
+        messages, 
+        tokenize=False, 
+        add_generation_prompt=True
     )
 
-    outputs = pipe(prompt, max_new_tokens=80, do_sample=False)
-    # outputs is a list of dicts: [{"generated_text": "..."}]
-    text = outputs[0].get("generated_text", "").strip()
-    return text
-
+    outputs = pipe(
+        prompt,
+        max_new_tokens=80,
+        do_sample=True,
+        top_p = 0.9,
+        temperature=0.3
+    )
 
-# =========================
-# WASTE ANALYSIS
-# =========================
-waste_analyzation = {...}  # KEEP SAME CONTENTS
-waste_analyzation_v2 = {...}
-waste_analyzation_v3 = {...}
+    raw = outputs[0]["generated_text"]
+    return clean_chat_output(raw)
 
-analysis = [waste_analyzation_v3, waste_analyzation_v2, waste_analyzation]
 
+# ------------------ PIPELINE ------------------
 top_label = None
+waste_analyzation = {
+    "cardboard": {
+        "Recycling type": "Paper/Cardboard recycling",
+        "Disposal": "Flatten the cardboard boxes and remove any non-paper packing materials before placing them in the recycling bin.",
+        "Tips": "Keep cardboard dry and clean to ensure it can be recycled efficiently.",
+        "Extra": "Avoid using waxed or heavily coated cardboard, as it may not be recyclable."
+    },
+    "glass": {
+        "Recycling type": "Glass recycling",
+        "Disposal": "Rinse glass bottles and jars, remove lids, and place them in a dedicated glass recycling container.",
+        "Tips": "Avoid breaking the glass; broken pieces can be hazardous and may complicate processing.",
+        "Extra": "Colored glass is often recycled separately; check local recycling rules."
+    },
+    "metal": {
+        "Recycling type": "Metal recycling",
+        "Disposal": "Rinse aluminum cans and steel containers, remove labels if possible, and put them in the metal recycling bin.",
+        "Tips": "Crush cans to save space and facilitate transport.",
+        "Extra": "Avoid contaminating with food waste to improve recycling efficiency."
+    },
+    "paper": {
+        "Recycling type": "Paper recycling",
+        "Disposal": "Sort newspapers, office paper, and magazines into the paper recycling bin; remove plastic coatings and staples.",
+        "Tips": "Shred sensitive documents but keep shredded paper in a bag to avoid scattering.",
+        "Extra": "Do not recycle wet or food-stained paper."
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling",
+        "Disposal": "Rinse plastic bottles, containers, and packaging, and place them in the plastic recycling bin.",
+        "Tips": "Check recycling codes; some plastics (like #3 PVC) are not widely recyclable.",
+        "Extra": "Remove caps and labels when possible to improve sorting efficiency."
+    },
+    "trash": {
+        "Recycling type": "General waste / landfill",
+        "Disposal": "Place non-recyclable items in the regular trash bin; avoid mixing with recyclables.",
+        "Tips": "Separate hazardous materials like batteries, electronics, or chemicals for special disposal.",
+        "Extra": "Try to reduce overall waste by reusing and composting when possible."
+    }
+}
+waste_analyzation_v2 = {
+    "cardboard": {
+        "Recycling type": "Paper/Cardboard recycling",
+        "Disposal": "Fold or cut cardboard into smaller flat pieces so it fits properly in recycling bins.",
+        "Tips": "Keep cardboard free from oils or food stains to avoid contaminating the batch.",
+        "Extra": "Cardboard with shiny or laminated surfaces may need separate processing."
+    },
+    "glass": {
+        "Recycling type": "Glass recycling",
+        "Disposal": "Remove lids and caps, lightly rinse, and place glass containers into the glass-only bin.",
+        "Tips": "Do not place broken drinking glasses or ceramics with recyclable glass—they melt at different temperatures.",
+        "Extra": "Some regions require taking glass to drop-off centers instead of curbside bins."
+    },
+    "metal": {
+        "Recycling type": "Metal recycling",
+        "Disposal": "Empty metal containers and place them in the metal bin; labels may stay on.",
+        "Tips": "Keep metal separate from electronics or batteries, which require special facilities.",
+        "Extra": "Large metal objects like appliances may require bulk recycling pickup."
+    },
+    "paper": {
+        "Recycling type": "Paper recycling",
+        "Disposal": "Place clean, dry paper products such as envelopes and notebooks into paper bins.",
+        "Tips": "Do not recycle coated paper (e.g., laminated or plastic-lined pages).",
+        "Extra": "Paper with heavy ink coverage might be processed separately."
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling",
+        "Disposal": "Clean and drain plastic bottles or tubs, then place them in the plastics bin.",
+        "Tips": "Flatten plastic bottles to reduce volume unless your local guidelines say otherwise.",
+        "Extra": "Some plastics like polystyrene foam require special drop-off locations."
+    },
+    "trash": {
+        "Recycling type": "General waste / landfill",
+        "Disposal": "Dispose of unrecyclable materials in regular waste; secure loose items in bags.",
+        "Tips": "Separate hazardous or toxic waste like paint, solvents, or chemicals.",
+        "Extra": "Reduce landfill impact by choosing reusable products whenever possible."
+    }
+}
+waste_analyzation_v3 = {
+    "cardboard": {
+        "Recycling type": "Paper/Cardboard recycling",
+        "Disposal": "Remove packing tape when possible and place cardboard in dry storage until pickup day.",
+        "Tips": "Avoid leaving cardboard outdoors where rain could weaken fibers and ruin recyclability.",
+        "Extra": "Food-contaminated cardboard can often be composted instead of recycled."
+    },
+    "glass": {
+        "Recycling type": "Glass recycling",
+        "Disposal": "Sort by color only if required locally; rinse lightly and drop into the proper bin.",
+        "Tips": "Handle carefully to avoid breakage, as shattered glass is often not recyclable in curbside programs.",
+        "Extra": "Glass jars with metal clasps or rubber seals may need partial disassembly before recycling."
+    },
+    "metal": {
+        "Recycling type": "Metal recycling",
+        "Disposal": "Ensure metal food cans are clean and empty before placing in the recycling bin.",
+        "Tips": "Rinse cans briefly—no need for perfect cleaning—as this prevents pests and odors.",
+        "Extra": "Metal lids from jars should be recycled separately from the glass container."
+    },
+    "paper": {
+        "Recycling type": "Paper recycling",
+        "Disposal": "Place paper in the appropriate bin; keep shredded paper in a paper bag if accepted.",
+        "Tips": "Avoid mixing paper with wet waste like food scraps to maintain recyclability.",
+        "Extra": "Sticky notes and small scraps may or may not be accepted depending on the facility."
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling",
+        "Disposal": "Recycle only plastics accepted by your local program; rinse and drain them completely.",
+        "Tips": "Avoid recycling small plastic pieces under 2 inches—they may jam sorting machinery.",
+        "Extra": "Hard-to-recycle plastics may be collected through special community programs."
+    },
+    "trash": {
+        "Recycling type": "General waste / landfill",
+        "Disposal": "Place all non-recyclables in the trash bin; tie bags securely to avoid leakage.",
+        "Tips": "Do not place batteries, electronics, or sharp objects directly into household trash.",
+        "Extra": "Consider composting organic waste to reduce household trash volume."
+    }
+}
+analysis = [waste_analyzation_v3, waste_analyzation_v2, waste_analyzation]
 
-
-# =========================
-# CLASSIFICATION PIPELINES
-# =========================
 def classify_pipeline(image):
     global top_label
+
     predictions = classify_image(image)
-    top_label = max(predictions, key=predictions.get)
+    top_label = max(predictions, key=predictions.get)  # top-1
     return predictions
 
-
 def analyze_pipeline():
     global top_label
 
     if top_label is None:
         return "Please classify an image first."
-
+    
+    explanation = explain_recycling(top_label)
+    
     choice = random.randint(0, 2)
-    info = analysis[choice][top_label]
-
+    info = analysis[choice][top_label]  
     result = (
         f"• Recycling type: {info['Recycling type']}\n"
         f"• Disposal: {info['Disposal']}\n"
@@ -110,19 +230,120 @@ def analyze_pipeline():
     return result
 
 
+# ------------------ GRADIO UI ------------------
 # =========================
-# API ENDPOINTS
+#  CSS
 # =========================
-@app.post("/api/classify")
-def api_classify(data: APIImage):
-    img_bytes = base64.b64decode(data.image_base64)
-    image = Image.open(BytesIO(img_bytes)).convert("RGB")
+custom_css = """
+
+
+#main-title {
+    text-align: center;
+    color: #2E7D32;
+    font-size: 34px;
+    font-weight: 800;
+    margin-bottom: 22px;
+}
+
+.gradio-container {
+    background: linear-gradient(135deg, #E8F5E9 0%, #F1F8E9 100%);
+    font-family: 'Segoe UI', sans-serif;
+}
+
+#explainbox textarea {
+    background: #ffffff;
+    height: 120px; 
+    border: 2px solid #A5D6A7;
+    border-radius: 12px;
+    padding: 12px;
+    font-size: 15px;
+}
 
-    preds = classify_pipeline(image)
-    return {"predictions": preds}
+.gr-button.primary {
+    background: #43A047 !important;
+    color: white !important;
+    border-radius: 12px !important;
+    padding: 12px 20px !important;
+    font-size: 17px !important;
+    box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.15);
+    transition: 0.2s ease;
+}
+
+textarea, .gr-textbox textarea, #explainbox textarea {
+    color: #1B5E20 !important;
+}
+
+#tips-box li {
+    color: #2E7D32 !important;
+}
+
+.gr-button.primary:hover {
+    background: #2E7D32 !important;
+    transform: translateY(-2px);
+}
+
+
+"""
+
+# =========================
+# GRADIO UI
+# =========================
+with gr.Blocks() as demo:
+
+    # TITLE
+    gr.Markdown("<h1 id='main-title'>♻️ AI Waste Classifier</h1>")
+
+    # INPUT ROW
+    with gr.Row():
+        with gr.Column(scale=1):
+            img_input = gr.Image(
+                type="pil",
+                label="📸 Upload waste image",
+                elem_id="upload-area"
+            )
+
+        with gr.Column(scale=1):
+            gr.Markdown(
+                """
+                <div id="tips-box" style="border:2px solid #A5D6A7; padding:16px; border-radius:14px; background:white;">
+                    <h3 style="color:#2E7D32;">🌍 Quick recycling tips:</h3>
+                    <ul>
+                        <li>Organic waste → green bin</li>
+                        <li>Plastic, metal waste → recycle</li>
+                        <li>Wash your glass waste!</li>
+                        <li>Battery, eletrical devices → non-metal container</li>
+                    </ul>
+                </div>
+                """
+            )
+
+    # OUTPUTS
+    cls_output = gr.Label(
+        num_top_classes=3,
+        label="🔍 Classifier Prediction (Top 3)"
+    )
+    analyze_btn = gr.Button("Classify Waste", variant="primary")
+
+    analyze_btn.click(
+        classify_pipeline,
+        inputs=img_input,
+        outputs=cls_output
+    )
+
+    explain_output = gr.Textbox(
+        label="🧩 Detailed Recycling & Disposal Advice",
+        elem_id="explainbox",
+        lines=6
+    )
+
+    # BUTTON
+    analyze_result_btn = gr.Button("Analyze", variant="primary")
+
+    analyze_result_btn.click(
+        analyze_pipeline,
+        inputs=None,
+        outputs=explain_output
+    )
 
 
-@app.post("/api/analyze")
-def api_analyze():
-    result = analyze_pipeline()
-    return {"result": result}
+demo.launch(css=custom_css)

requirements.txt

@@ -6,6 +6,3 @@ transformers
 sentencepiece
 accelerate
 
-fastapi
-uvicorn
-pydantic

spaces.yaml

@@ -1,3 +0,0 @@
-sdk: "python"
-python_version: "3.10"
-app_file: "app.py"