Private change

app.py

@@ -1,19 +1,37 @@
-import gradio as gr
+import base64
+from io import BytesIO
+from PIL import Image
+
+from fastapi import FastAPI
+from pydantic import BaseModel
+
 import random
 import torch
 from transformers import (
     AutoTokenizer, AutoModelForCausalLM,
-    T5Tokenizer,
-    T5ForConditionalGeneration,
     pipeline
 )
-# ------------------ LOAD CLASSIFIER ------------------
+
+# =========================
+# FASTAPI APP
+# =========================
+app = FastAPI()
+
+# =========================
+# API MODELS
+# =========================
+class APIImage(BaseModel):
+    image_base64: str
+
+
+# =========================
+# 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"]
@@ -22,35 +40,39 @@ def classify_image(image):
     return results
 
 
-# ------------------ LOAD CHAT MODEL ------------------
+# =========================
+# LOAD CHAT MODEL
+# =========================
 tiny_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 
 tokenizer = AutoTokenizer.from_pretrained(tiny_model)
 chat_model = AutoModelForCausalLM.from_pretrained(
     tiny_model,
-    torch_dtype=torch.bfloat16, 
+    torch_dtype=torch.bfloat16,
     device_map="auto",
-    low_cpu_mem_usage=True)
+    low_cpu_mem_usage=True
+)
 
 pipe = pipeline(
     "text-generation",
     model=chat_model,
     tokenizer=tokenizer,
     device_map="auto",
-    max_new_tokens=60
+    max_new_tokens=80
 )
 
-def clean_chat_output(full_text):
 
+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:"):
+    if lines and lines[0].lower().startswith("item:"):
         lines = lines[1:]
 
     return "\n".join(lines).strip()
 
+
 def explain_recycling(class_label):
     system_msg = {
         "role": "system",
@@ -67,18 +89,18 @@ def explain_recycling(class_label):
         "content": f"Item: {class_label}\nReturn the two bullet points now."
     }
     messages = [system_msg, user_msg]
-    
+
     prompt = tokenizer.apply_chat_template(
-        messages, 
-        tokenize=False, 
+        messages,
+        tokenize=False,
         add_generation_prompt=True
     )
 
     outputs = pipe(
         prompt,
-        max_new_tokens=60,
+        max_new_tokens=80,
         do_sample=True,
-        top_p = 0.9,
+        top_p=0.9,
         temperature=0.3
     )
 
@@ -86,141 +108,38 @@ def explain_recycling(class_label):
     return clean_chat_output(raw)
 
 
-# ------------------ 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."
-    }
-}
+# =========================
+# WASTE ANALYSIS
+# (your 3 versions)
+# =========================
+waste_analyzation = {...}  # KEEP SAME CONTENTS
+waste_analyzation_v2 = {...}
+waste_analyzation_v3 = {...}
+
 analysis = [waste_analyzation_v3, waste_analyzation_v2, waste_analyzation]
 
+top_label = None
+
+
+# =========================
+# CLASSIFICATION PIPELINES
+# =========================
 def classify_pipeline(image):
     global top_label
-
     predictions = classify_image(image)
-    top_label = max(predictions, key=predictions.get)  # top-1
+    top_label = max(predictions, key=predictions.get)
     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"
@@ -230,120 +149,19 @@ def analyze_pipeline():
     return result
 
 
-# ------------------ GRADIO UI ------------------
-# =========================
-#  CSS
-# =========================
-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;
-}
-
-.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
+# API ENDPOINTS
 # =========================
-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")
+@app.post("/api/classify")
+def api_classify(data: APIImage):
+    img_bytes = base64.b64decode(data.image_base64)
+    image = Image.open(BytesIO(img_bytes)).convert("RGB")
 
-    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
-    )
+    preds = classify_pipeline(image)
+    return {"predictions": preds}
 
 
-demo.launch(css=custom_css)
+@app.post("/api/analyze")
+def api_analyze():
+    result = analyze_pipeline()
+    return {"result": result}

requirements.txt

@@ -1,8 +1,11 @@
-huggingface_hub
 gradio
 numpy
 Pillow
 torch
 transformers
 sentencepiece
-accelerate
+accelerate
+
+fastapi
+uvicorn
+pydantic