Update app.py

app.py

@@ -121,6 +121,10 @@ 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(
         """
@@ -160,10 +164,14 @@ if action == "Upload Image":
             for item in 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):
+                        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))
                 total_carbon_reduction += carbon_reduction
 
-                st.write(response)
                 st.markdown(
                     f"""<p style="color: #2e7d32;">🌍 Carbon Footprint Reduction: {carbon_reduction:.2f} kg CO₂</p>""",
                     unsafe_allow_html=True,
@@ -179,74 +187,11 @@ if action == "Upload Image":
         else:
             st.error("No recognizable waste items detected.")
 
-elif action == "Get Suggestions for Items":
+if st.session_state.clicked_diy:
     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>
-        """,
+        f"""<div style="padding: 10px; background-color: #f0f4c3; color: #33691e; border-radius: 5px;">
+            <h4>📝 DIY Instructions:</h4>
+            {st.session_state.clicked_diy}
+        </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,
-        )
-
-    # Add session state for DIY instructions
-    if "diy_suggestion" not in st.session_state:
-        st.session_state.diy_suggestion = ""
-
-    suggestion = st.text_input("DIY Guide: Please cCopy and Paste any Idea generated above you wish to materilize:", key="diy_input")
-    if st.button("Generate DIY Instructions"):
-        if suggestion:
-            st.session_state.diy_suggestion = get_diy_steps_from_groq(suggestion)
-
-    if st.session_state.diy_suggestion:
-        st.markdown(
-            f"""<div style="padding: 10px; background-color: #f0f4c3; color: #33691e; border-radius: 5px;">
-                <h4>📝 DIY Instructions:</h4>
-                {st.session_state.diy_suggestion}
-            </div>""",
-            unsafe_allow_html=True,
-        )
-
-# Motivational Message
-st.markdown(
-    """
-    <div style="text-align: center; padding: 20px; background-color: #dcedc8; border-radius: 10px;">
-        <h3 style="color: #33691e;">🌍 Let's Keep Our Planet Green!</h3>
-        <p style="color: #2e7d32;">Recycling is not just an action but a responsibility. Together, we can make a difference. ♻️💚</p>
-    </div>
-    """,
-    unsafe_allow_html=True,
-)
+    )