Update app.py

app.py

@@ -9,6 +9,14 @@ import imageio
 from ultralytics import YOLO
 from huggingface_hub import hf_hub_download
 
+# Move set_page_config to the very top as it must be the first Streamlit command
+st.set_page_config(
+    page_title="Fire Detection: Original vs. Processed Video",
+    page_icon="🔥",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
 ###############################################################################
 # Helper: Embed an HTML5 video that autoplays (muted) with controls.
 ###############################################################################
@@ -39,25 +47,24 @@ if "shortened_video_ready" not in st.session_state:
 ###############################################################################
 # Download YOLO Model from Hugging Face
 ###############################################################################
-repo_id = "tstone87/ccr-colorado"
+repo_id = "tstone87/ccr-colorado"  # Verify this is correct
 model_filename = "best.pt"
 
 try:
-    local_model_path = hf_hub_download(repo_id=repo_id, filename=model_filename)
+    # Add error handling for repository access
+    local_model_path = hf_hub_download(
+        repo_id=repo_id, 
+        filename=model_filename,
+        repo_type="model"  # Explicitly specify repo_type
+    )
     model = YOLO(local_model_path)
 except Exception as ex:
-    st.error(f"Unable to load model. Check model path: {model_filename}")
-    st.error(ex)
-
-###############################################################################
-# Configure Streamlit Page Layout
-###############################################################################
-st.set_page_config(
-    page_title="Fire Detection: Original vs. Processed Video",
-    page_icon="🔥",
-    layout="wide",
-    initial_sidebar_state="expanded"
-)
+    st.error(f"Unable to load model. Error details: {str(ex)}")
+    st.error("Please verify:")
+    st.error("1. Repository 'tstone87/ccr-colorado' exists and is accessible")
+    st.error("2. 'best.pt' file is uploaded to the repository")
+    st.error("3. You have proper access permissions")
+    model = None  # Set model to None if loading fails
 
 ###############################################################################
 # SIDEBAR: Video input options
@@ -79,7 +86,7 @@ with st.sidebar:
     )
     progress_text = st.empty()
     progress_bar = st.progress(0)
-    download_placeholder = st.empty()  # Placeholder for download button
+    download_placeholder = st.empty()
 
 ###############################################################################
 # MAIN TITLE
@@ -93,7 +100,6 @@ original_video_data = None
 processed_video_data = None  
 
 if example_option != "None":
-    # Use example videos from remote URLs.
     example_videos = {
         "T Example": ("T1.mp4", "T2.mpg"),
         "LA Example": ("LA1.mp4", "LA2.mp4")
@@ -107,9 +113,8 @@ if example_option != "None":
             original_video_data = requests.get(orig_url).content
             processed_video_data = requests.get(proc_url).content
         except Exception as ex:
-            st.error("Error loading example videos. Check your URLs.")
+            st.error(f"Error loading example videos: {str(ex)}")
 else:
-    # No example selected. If a file is uploaded, use it.
     if source_file:
         file_type = source_file.type.split('/')[0]
         if file_type == 'image':
@@ -150,7 +155,7 @@ with col2:
 # Process Video if No Example is Selected
 ###############################################################################
 if example_option == "None" and source_file and source_file.type.split('/')[0] != 'image':
-    if st.sidebar.button("Let's Detect Wildfire"):
+    if st.sidebar.button("Let's Detect Wildfire") and model is not None:
         st.session_state["processed_frames"] = []
         processed_frames = st.session_state["processed_frames"]
 
@@ -176,7 +181,7 @@ if example_option == "None" and source_file and source_file.type.split('/')[0] !
 
         if processed_frames:
             temp_video_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
-            fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Use widely supported codec
+            fourcc = cv2.VideoWriter_fourcc(*'mp4v')
             out = cv2.VideoWriter(temp_video_file.name, fourcc, fps, (width, height))
 
             for frame in processed_frames:
@@ -199,4 +204,4 @@ if st.session_state["shortened_video_ready"]:
         data=st.session_state["shortened_video_data"],
         file_name="processed_video.mp4",
         mime="video/mp4"
-    )
+    )