Update app.py

app.py

@@ -1,4 +1,6 @@
 import streamlit as st
+from huggingface_hub import InferenceClient
+import os
 
 st.set_page_config(page_title="Smart Waste Segregation Advisor")
 
@@ -10,8 +12,11 @@ This tool helps users identify waste categories and provides guidance
 for responsible disposal using AI-based reasoning.
 """)
 
-# Image upload (UI purpose)
-uploaded_image = st.file_uploader("Upload an image of the waste item (optional)", type=["jpg", "png", "jpeg"])
+# Image upload
+uploaded_image = st.file_uploader(
+    "Upload an image of the waste item (optional)",
+    type=["jpg", "png", "jpeg"]
+)
 
 # Optional text input
 user_text = st.text_input("Optional: Describe the waste item")
@@ -19,47 +24,38 @@ user_text = st.text_input("Optional: Describe the waste item")
 st.divider()
 
 if st.button("Analyze Waste"):
-    # Simulated image interpretation
     if uploaded_image:
         image_description = "An image showing a waste item uploaded by the user"
     else:
         image_description = "No image provided"
 
+    # 🔐 Initialize IBM Granite via HF Inference API
+    client = InferenceClient(
+        model="ibm-granite/granite-3.0-2b-instruct",
+        token=os.getenv("HF_TOKEN")
+    )
+
     prompt = f"""
 You are an AI-powered Smart Waste Segregation Advisor.
 
-You will receive:
-1. A brief description of an uploaded image (assume the image shows a waste item)
-2. Optional text input from the user
-
-Your tasks:
-1. Identify the waste item based on the input.
-2. Classify the item into one of the following categories:
-   - Wet Waste
-   - Dry Waste
-   - E-Waste
-3. Explain briefly why the item belongs to this category.
-4. Suggest the correct and safe disposal method.
-5. Explain the environmental impact if the item is disposed of incorrectly.
+Tasks:
+1. Identify the waste item.
+2. Classify it as Wet Waste, Dry Waste, or E-Waste.
+3. Explain the reason.
+4. Suggest correct disposal.
+5. Explain environmental impact.
 
 User Input:
 Image description: {image_description}
-Additional text: {user_text}
+User text: {user_text}
 """
 
-    # Simulated AI response (semi-working prototype)
-    response = """
-**Waste Category:** Dry Waste
-
-**Reason:**  
-Plastic-based items are non-biodegradable and commonly recyclable.
-
-**Disposal Method:**  
-Dispose of the item in a dry waste or recycling bin.
-
-**Environmental Impact:**  
-Improper disposal can cause long-term pollution and harm wildlife.
-"""
+    with st.spinner("Analyzing using IBM Granite AI..."):
+        response = client.text_generation(
+            prompt,
+            max_new_tokens=250,
+            temperature=0.7
+        )
 
     st.markdown("### 🧠 AI Analysis Result")
-    st.markdown(response)
+    st.write(response)