Update app.py

app.py

@@ -11,20 +11,34 @@ import streamlink
 # Page Config
 st.set_page_config(page_title="AI Fire Watch", page_icon="🌍", layout="wide")
 
-# Lighter Background CSS
+# Lighter Background CSS with Darker Text
 st.markdown(
     """
     <style>
     .stApp {
         background-color: #f5f5f5;
-        color: #333333;
+        color: #1a1a1a;  /* Dark text for general content */
+    }
+    h1 {
+        color: #1a1a1a;  /* Darker title text */
     }
     .stTabs > div > button {
         background-color: #e0e0e0;
-        color: #333333;
+        color: #1a1a1a;  /* Darker tab text */
+        font-weight: bold;
     }
     .stTabs > div > button:hover {
         background-color: #d0d0d0;
+        color: #1a1a1a;
+    }
+    .stButton > button {
+        background-color: #e0e0e0;
+        color: #1a1a1a;  /* Darker button text */
+        font-weight: bold;
+    }
+    .stButton > button:hover {
+        background-color: #d0d0d0;
+        color: #1a1a1a;
     }
     </style>
     """,
@@ -39,6 +53,12 @@ except Exception as ex:
     st.error(f"Model loading failed: {ex}")
     st.stop()
 
+# Initialize Session State
+if 'monitoring' not in st.session_state:
+    st.session_state.monitoring = False
+if 'current_webcam_url' not in st.session_state:
+    st.session_state.current_webcam_url = None
+
 # Header
 st.title("AI Fire Watch")
 st.markdown("Monitor fire and smoke in real-time with AI precision.")
@@ -87,15 +107,25 @@ with tabs[1]:
         webcam_url = st.text_input("Webcam URL", "http://<your_webcam_ip>/current.jpg", label_visibility="collapsed")
         confidence = st.slider("Detection Threshold", 0.25, 1.0, 0.4, key="webcam_conf")
         start = st.button("Begin Monitoring", key="webcam_start")
-    with col2:
+        stop = st.button("Stop Monitoring", key="webcam_stop")
+
+        # Handle monitoring state
         if start:
+            st.session_state.monitoring = True
+            st.session_state.current_webcam_url = webcam_url
+        if stop or (st.session_state.monitoring and webcam_url != st.session_state.current_webcam_url):
+            st.session_state.monitoring = False
+            st.session_state.current_webcam_url = None
+
+    with col2:
+        if st.session_state.monitoring and st.session_state.current_webcam_url:
             image_placeholder = st.empty()
             timer_placeholder = st.empty()
             refresh_interval = 30  # Refresh every 30 seconds
             while True:
                 start_time = time.time()
                 try:
-                    response = requests.get(webcam_url, timeout=5)
+                    response = requests.get(st.session_state.current_webcam_url, timeout=5)
                     if response.status_code != 200:
                         st.error(f"Fetch failed: HTTP {response.status_code}")
                         break
@@ -118,6 +148,7 @@ with tabs[1]:
                     st.experimental_rerun()
                 except Exception as e:
                     st.error(f"Error: {e}")
+                    st.session_state.monitoring = False
                     break
 
 # Tab 3: YouTube