Update app.py

app.py

@@ -121,10 +121,6 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
-# Session state to track clicked DIY suggestions
-if "clicked_diy" not in st.session_state:
-    st.session_state.clicked_diy = ""
-
 if action == "Upload Image":
     st.markdown(
         """
@@ -161,12 +157,12 @@ if action == "Upload Image":
             st.write(", ".join(combined_items))
 
             total_carbon_reduction = 0
-            for item in combined_items:
+            for idx, item in enumerate(combined_items):
                 st.markdown(f"**Recycling Idea for {item}:**")
                 response = get_recycling_suggestions_from_groq(item, 1)
                 suggestions = response.split("\n")
-                for suggestion in suggestions:
-                    if st.button(f"{suggestion}", key=suggestion):
+                for suggestion_idx, suggestion in enumerate(suggestions):
+                    if st.button(f"{suggestion}", key=f"{suggestion}_{idx}_{suggestion_idx}"):
                         st.session_state.clicked_diy = get_diy_steps_from_groq(suggestion)
 
                 carbon_reduction = max(0.5, min(2.5, carbon_reduction_data.get(item.lower(), 0) * 1))
@@ -187,6 +183,49 @@ if action == "Upload Image":
         else:
             st.error("No recognizable waste items detected.")
 
+elif action == "Get Suggestions for Items":
+    st.markdown(
+        """
+        <div style="text-align: center; background-color: #fff3e0; padding: 10px; border-radius: 5px;">
+            <h3 style="color: #ff6f00;">Select clutter items for recycling suggestions:</h3>
+        </div>
+        """,
+        unsafe_allow_html=True,
+    )
+    selected_items = []
+    quantities = {}
+
+    cols = st.columns(len(predefined_clutter_items))
+    for i, (item, emoji) in enumerate(predefined_clutter_items.items()):
+        with cols[i]:
+            if st.checkbox(f"{emoji} {item.title()}", key=item):
+                selected_items.append(item)
+                quantities[item] = st.number_input(f"{item} (kg):", min_value=0.0, step=0.1, key=f"qty_{item}")
+
+    if selected_items and st.button("Generate Suggestions"):
+        total_carbon_reduction = 0
+        st.write("### ♻️ Recycling Suggestions and Impact:")
+        for item, quantity in quantities.items():
+            if quantity > 0:
+                response = get_recycling_suggestions_from_groq(item, quantity)
+                carbon_reduction = max(0.5, min(2.5, carbon_reduction_data.get(item.lower(), 0) * quantity))
+                total_carbon_reduction += carbon_reduction
+
+                st.markdown(f"**{item} ({quantity} kg)**")
+                st.write(response)
+                st.markdown(
+                    f"""<p style="color: #2e7d32;">🌍 Carbon Footprint Reduction: {carbon_reduction:.2f} kg CO₂</p>""",
+                    unsafe_allow_html=True,
+                )
+                st.write("---")
+
+        st.markdown(
+            f"""<div style="padding: 15px; text-align: center; background-color: #004d40; color: #ffffff; border-radius: 5px;">
+                🌟 Total Carbon Footprint Reduction: <b>{total_carbon_reduction:.2f} kg CO₂ saved</b>
+            </div>""",
+            unsafe_allow_html=True,
+        )
+
 if st.session_state.clicked_diy:
     st.markdown(
         f"""<div style="padding: 10px; background-color: #f0f4c3; color: #33691e; border-radius: 5px;">
@@ -194,4 +233,4 @@ if st.session_state.clicked_diy:
             {st.session_state.clicked_diy}
         </div>""",
         unsafe_allow_html=True,
-    )
+    )