change theme

app.py

@@ -38,21 +38,25 @@ def explain_recycling(class_label):
 
         Explain the followings:
         1. What recycling type this is (e.g., plastic, organic, glass, paper, metal).
-        2. How to properly dispose of or recycle it.
+        2. How to properly dispose of or recycle it correctly.
+        3. Environmental impact if disposed improperly
+        4. Optional tips for reducing waste or reusing the item
 
+        Write the response in friendly, clear, detailed paragraphs.
+        
         Use this format:
 
         Recycling type: <type>
-        Recommendation: <instruction>
+        Recommendation: <Detailed instructions>
         """
 
     inputs = tokenizer(prompt, return_tensors="pt").input_ids
 
     outputs = chat_model.generate(
         inputs,
-        max_length=180,
+        max_length=400,
         do_sample=True,
-        temperature=0.7,
+        temperature=0.75,
         top_p=0.9
     )
 
@@ -61,30 +65,50 @@ def explain_recycling(class_label):
 # ------------------ GRADIO APP ------------------
 
 def full_pipeline(image):
-    """Classifier → get top label → Flan-T5 explanation."""
+    "Classification → get top label → explanation"
     predictions = classify_image(image)
 
-    # Get the top class name
-    top_label = max(predictions, key=predictions.get)
-
+    top_label = max(predictions, key=predictions.get) # Get the top class name
     explanation = explain_recycling(top_label)
-
     return predictions, explanation
 
-
-with gr.Blocks() as demo:
-    gr.Markdown("# ♻️ AI Waste Classifier + Disposal Advisor")
+custom_theme = gr.themes.Base(
+    primary_hue="green",
+    secondary_hue="blue",
+    neutral_hue="slate",
+).set(
+    body_background_fill="#F8FFF4",
+    block_background_fill="#FFFFFF",
+    button_primary_background_fill="#3BAF4A",   # Green Analyze button
+    button_primary_background_fill_hover="#2F8C3A",
+    button_primary_text_color="white",
+)
+
+with gr.Blocks(theme=custom_theme, css="#explainbox {height: 350px;}") as demo:
+    gr.Markdown(
+        "<h1 style='text-align:center;'>♻️ AI Waste Classifier + Eco Advisor</h1>"
+    )
 
     with gr.Row():
         img_input = gr.Image(type="pil", label="Upload waste image")
 
     with gr.Column():
         cls_output = gr.Label(num_top_classes=3, label="Classifier Prediction")
-        explain_output = gr.Textbox(label="Recycling Advice", lines=6)
 
-    run_btn = gr.Button("Analyze")
+        explain_output = gr.Textbox(
+            label="Detailed Recycling & Disposal Advice",
+            elem_id="explainbox",
+            lines=16,
+            show_copy_button=True,
+        )
+
+    analyze_btn = gr.Button("Analyze", variant="primary")
 
-    run_btn.click(full_pipeline, inputs=img_input, outputs=[cls_output, explain_output])
+    analyze_btn.click(
+        full_pipeline,
+        inputs=img_input,
+        outputs=[cls_output, explain_output]
+    )
 
 
 demo.launch()