Fix

Dockerfile

@@ -17,6 +17,6 @@ EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.enableXsrfProtection=false"]
 
 # If secrets need to be mounted: https://huggingface.co/docs/hub/spaces-sdks-docker#secrets-and-variables-management

src/streamlit_app.py

@@ -3,45 +3,43 @@ import streamlit as st
 
 from plant_identifier.identifier import identify_plant
 
-st.write("XSRF:", st.get_option("server.enableXsrfProtection"))
-
-# st.set_page_config(page_title="Greenmorrow Plant Identification", page_icon="🌿")
-
-# st.title("🌱🌲 Greenmorrow Plant Identification Agent")
-# st.write("Upload a plant photograph and the app will attempt to identify the species.")
-
-# # Hugging Face Spaces best practice: store keys in Space Secrets.
-# # They are available via st.secrets and/or environment variables.
-# openai_key = os.getenv("OPENAI_KEY", os.getenv("OPENAI_KEY", ""))
-# plantnet_key = os.getenv("PLANTNET_API_KEY", os.getenv("PLANTNET_API_KEY", ""))
-
-# if not openai_key or not plantnet_key:
-#     st.error(
-#         "Missing API keys. Please define the following Space Secrets:"
-#         "- OPENAI_KEY"
-#         "- PLANTNET_API_KEY"
-#     )
-#     st.stop()
-
-# st.subheader("📷 Upload plant photo(s)")
-# uploaded_files = st.file_uploader(
-#     "Upload one or more images",
-#     type=["png", "jpg", "jpeg", "webp"],
-#     accept_multiple_files=True,
-# )
-
-# if st.button("Identify plant", type="primary", disabled=not uploaded_files):
-#     with st.spinner("Identifying plant..."):
-#         try:
-#             # Streamlit UploadedFile behaves like a file-like object
-#             result = identify_plant(
-#                 images=[f for f in uploaded_files],
-#                 openai_api_key=openai_key,
-#                 plantnet_api_key=plantnet_key,
-#             )
-#             st.success("Done!")
-#             st.markdown("### 🌿 Identification result")
-#             st.write(result)
-
-#         except Exception as e:
-#             st.error(f"Error: {e}")
+st.set_page_config(page_title="Greenmorrow Plant Identification", page_icon="🌿")
+
+st.title("🌱🌲 Greenmorrow Plant Identification Agent")
+st.write("Upload a plant photograph and the app will attempt to identify the species.")
+
+# Hugging Face Spaces best practice: store keys in Space Secrets.
+# They are available via st.secrets and/or environment variables.
+openai_key = os.getenv("OPENAI_KEY", os.getenv("OPENAI_KEY", ""))
+plantnet_key = os.getenv("PLANTNET_API_KEY", os.getenv("PLANTNET_API_KEY", ""))
+
+if not openai_key or not plantnet_key:
+    st.error(
+        "Missing API keys. Please define the following Space Secrets:"
+        "- OPENAI_KEY"
+        "- PLANTNET_API_KEY"
+    )
+    st.stop()
+
+st.subheader("📷 Upload plant photo(s)")
+uploaded_files = st.file_uploader(
+    "Upload one or more images",
+    type=["png", "jpg", "jpeg", "webp"],
+    accept_multiple_files=True,
+)
+
+if st.button("Identify plant", type="primary", disabled=not uploaded_files):
+    with st.spinner("Identifying plant..."):
+        try:
+            # Streamlit UploadedFile behaves like a file-like object
+            result = identify_plant(
+                images=[f for f in uploaded_files],
+                openai_api_key=openai_key,
+                plantnet_api_key=plantnet_key,
+            )
+            st.success("Done!")
+            st.markdown("### 🌿 Identification result")
+            st.write(result)
+
+        except Exception as e:
+            st.error(f"Error: {e}")