A-I4All's picture
Update app.py
d60d4f3 verified
import streamlit as st
from transformers import pipeline
import random
# Load the DistilGPT-2 model for text generation
model = pipeline("text-generation", model="distilgpt2")
def get_recycling_advice(product):
# Crafting a specific prompt for generating recycling advice
prompt = (f"As an environmental expert, provide detailed, actionable steps on how to recycle or "
f"properly dispose of the following product: '{product}'. "
"Include methods, potential hazards, and any specific recycling processes.")
# Generate the response from the model
response = model(prompt, max_length=150, num_return_sequences=1)
return response[0]['generated_text'].strip()
# List of daily sustainability tips
SUSTAINABILITY_TIPS = [
"Reduce single-use plastics by opting for reusable bags and containers.",
"Compost your kitchen waste to create nutrient-rich soil for gardening.",
"Choose energy-efficient appliances to reduce electricity consumption.",
"Always carry a reusable water bottle to minimize plastic waste.",
"Participate in local clean-up events to contribute to your community's cleanliness.",
]
# Get a random sustainability tip
daily_tip = random.choice(SUSTAINABILITY_TIPS)
# Streamlit UI
st.title("Recycling Advisor")
st.subheader("Get expert advice on how to recycle or dispose of your products!")
product = st.text_input("Enter a product you'd like to recycle:")
if st.button("Get Advice"):
if product:
advice = get_recycling_advice(product)
st.write(advice)
else:
st.error("Please enter a product name.")
# Additional features for a better user experience
st.sidebar.title("About")
st.sidebar.info("This application provides actionable suggestions on how to recycle and dispose of various products, promoting environmental sustainability.")
st.sidebar.subheader("Recycling Resources:")
st.sidebar.write("For proper recycling and disposal of waste, refer to the following resources:")
st.sidebar.write("[India: Central Pollution Control Board (CPCB)](https://cpcb.nic.in/recycling-and-disposal-of-waste/)")
st.sidebar.write("[Pakistan: Pakistan Environmental Protection Agency](http://www.environment.gov.pk/)")
st.sidebar.write("[Bangladesh: Department of Environment](http://www.doe.gov.bd/)")
st.sidebar.write("[USA: Environmental Protection Agency (EPA)](https://www.epa.gov/recycle)")
st.sidebar.write("[UK: Recycle Now](https://www.recyclenow.com/)")
st.sidebar.write("[Canada: Government of Canada - Recycling](https://www.canada.ca/en/services/environment/conservation/recycling.html)")
st.sidebar.write("[Germany: Federal Ministry for the Environment](https://www.bmu.de/en/topics/water-waste-soil/waste-management/recycling/)")
st.sidebar.write("[China: National Development and Reform Commission](http://en.ndrc.gov.cn/)")
st.sidebar.write("[Japan: Ministry of the Environment](https://www.env.go.jp/en/recycle/)")
st.sidebar.subheader("Daily Sustainability Tip:")
st.sidebar.write(daily_tip)