Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,6 +6,12 @@ import os
 import platform
 from tensorflow.keras.preprocessing.image import img_to_array
 
+# Safely get the directory of the current script (works locally and on hosted platforms)
+try:
+    current_dir = os.path.dirname(__file__)
+except NameError:
+    current_dir = os.getcwd()
+
 # Constants
 IMG_SIZE = 128
 CLASS_NAMES = ['cardboard', 'glass', 'metal', 'paper', 'plastic']
@@ -99,3 +105,12 @@ def main():
 
 if __name__ == "__main__":
     main()
+
+# Footer
+st.markdown("---")
+st.markdown(
+    "<div style='text-align: center; color: gray;'>"
+    "© 2025 Trash Sort App. Developed by <b>Cherilyn</b>."
+    "</div>",
+    unsafe_allow_html=True
+)