UI change

app.py

@@ -34,21 +34,31 @@ def explain_recycling(class_label):
     prompt = f"""
     You are an expert in waste management.
 
-    The item has been identified as: {class_label}
+    Item: {class_label}
 
-    Your task:
-    1. Give 2 to 3 clear bullet points explaining how to recycle this item correctly.
-    2. Give 2 to 3 clear bullet points explaining how to reduce waste from this item.
+    Your task is to produce EXACTLY TWO sections:
 
-    Formatting rules:
-    - Use simple hyphen bullet points ("- ").
-    - Put each bullet point on a new line.
-    - Do NOT add introductions, explanations, or extra text.
-    - Output only the bullet points.
+    SECTION 1: "Recycling Instructions"
+    - Write exactly 3 bullet points.
+    - Each bullet point MUST start with "- ".
+    - Each bullet point MUST be on its own line.
 
-    Now provide the bullet points for: {class_label}
+    SECTION 2: "Waste Reduction Tips"
+    - Write exactly 3 bullet points.
+    - Each bullet point MUST start with "- ".
+    - Each bullet point MUST be on its own line.
+
+    Important rules:
+    - Do NOT add anything before or after the two sections.
+    - Do NOT explain.
+    - Do NOT merge bullet points.
+    - Do NOT write paragraphs.
+    - Only output bullet points.
+
+    Now produce the two sections for: {class_label}
     """
 
+
     inputs = tokenizer(prompt, return_tensors="pt").input_ids
 
     outputs = chat_model.generate(