Fix: Added safety checks for secrets and updated UI

src/streamlit_app.py

@@ -57,16 +57,26 @@ st.markdown("""
     """, unsafe_allow_html=True)
 
 # --- CLOUD DATA LOGGING (GOOGLE SHEETS) ---
-if "GSHEETS_JSON" in st.secrets:
-    creds = json.loads(st.secrets["GSHEETS_JSON"])
-    conn = st.connection("gsheets", type=GSheetsConnection, credentials=creds)
+# Use .get() to prevent the "StreamlitSecretNotFoundError" crash
+gsheets_json = st.secrets.get("GSHEETS_JSON")
+sheet_url = st.secrets.get("GSHEETS_URL")
+
+if gsheets_json and sheet_url:
+    try:
+        creds = json.loads(gsheets_json)
+        conn = st.connection("gsheets", type=GSheetsConnection, credentials=creds)
+    except Exception as e:
+        st.error(f"Configuration Error: {e}")
+        st.stop()
 else:
-    conn = st.connection("gsheets", type=GSheetsConnection)
+    # This replaces the red crash box with a clean instructions message
+    st.warning("🚀 System Initializing... Please ensure GSHEETS_JSON and GSHEETS_URL are set in Hugging Face Settings.")
+    st.info("If you just added them, please wait 30 seconds and refresh.")
+    st.stop()
 
 def save_to_cloud(text, ai_label, ai_score, corrected_label=None):
     try:
-        sheet_url = st.secrets.get("GSHEETS_URL", "")
-        # Using "Sheet1" as per your previous spreadsheet setup
+        # Pull fresh data to append to
         existing_data = conn.read(spreadsheet=sheet_url, worksheet="Sheet1", ttl=0)
     except:
         existing_data = pd.DataFrame()