Update app.py

app.py

@@ -6,6 +6,7 @@ import json
 import random
 import logging
 import matplotlib.pyplot as plt
+import shutil
 from datetime import datetime
 from collections import Counter
 from typing import Any, Dict, List, Optional, Tuple
@@ -72,11 +73,15 @@ static_image: Optional[np.ndarray] = None
 
 # Constants
 DEFAULT_VIDEO_PATH = "sample.mp4"
-TEMP_IMAGE_PATH = "temp.jpg"
-CAPTURED_FRAMES_DIR = "captured_frames"
-OUTPUT_DIR = "outputs"
-os.makedirs(CAPTURED_FRAMES_DIR, exist_ok=True)
-os.makedirs(OUTPUT_DIR, exist_ok=True)
+TEMP_IMAGE_PATH = os.path.abspath("temp.jpg")
+CAPTURED_FRAMES_DIR = os.path.abspath("captured_frames")
+OUTPUT_DIR = os.path.abspath("outputs")
+TEMP_MEDIA_DIR = os.path.abspath("temp_media")
+
+# Ensure directories exist with write permissions
+for directory in [CAPTURED_FRAMES_DIR, OUTPUT_DIR, TEMP_MEDIA_DIR]:
+    os.makedirs(directory, exist_ok=True)
+    os.chmod(directory, 0o777)  # Ensure write permissions
 
 def initialize_media(media_file: Optional[Any] = None) -> str:
     global media_loaded, is_video, static_image, log_entries, frame_count
@@ -84,22 +89,37 @@ def initialize_media(media_file: Optional[Any] = None) -> str:
     static_image = None
     frame_count = 0  # Reset frame count on new media load
 
-    # Validate media_file
+    # If no media file is provided, try the default video
     if media_file is None:
         media_path = DEFAULT_VIDEO_PATH
-    elif not hasattr(media_file, 'name') or not media_file.name:
-        status = "Error: Invalid media file uploaded."
-        log_entries.append(status)
-        logging.error(status)
-        media_loaded = False
-        return status
+        log_entries.append(f"No media uploaded, attempting to load default: {media_path}")
+        logging.info(f"No media uploaded, attempting to load default: {media_path}")
     else:
-        media_path = media_file.name
+        # Validate media file
+        if not hasattr(media_file, 'name') or not media_file.name:
+            status = "Error: Invalid media file uploaded."
+            log_entries.append(status)
+            logging.error(status)
+            media_loaded = False
+            return status
 
-    # Log the media path for debugging
-    log_entries.append(f"Attempting to load media from path: {media_path}")
-    logging.info(f"Attempting to load media from path: {media_path}")
+        # Copy the uploaded file to a known location to avoid path issues
+        original_path = media_file.name
+        file_extension = os.path.splitext(original_path)[1].lower()
+        temp_media_path = os.path.join(TEMP_MEDIA_DIR, f"uploaded_media{file_extension}")
+        try:
+            shutil.copy(original_path, temp_media_path)
+            media_path = temp_media_path
+            log_entries.append(f"Copied uploaded file to: {media_path}")
+            logging.info(f"Copied uploaded file to: {media_path}")
+        except Exception as e:
+            status = f"Error copying uploaded file: {str(e)}"
+            log_entries.append(status)
+            logging.error(status)
+            media_loaded = False
+            return status
 
+    # Verify the file exists
     if not os.path.exists(media_path):
         status = f"Error: Media file '{media_path}' not found."
         log_entries.append(status)
@@ -109,12 +129,12 @@ def initialize_media(media_file: Optional[Any] = None) -> str:
 
     try:
         # Determine if the file is a video or image
-        if media_path.lower().endswith((".mp4", ".avi")):
+        if file_extension in (".mp4", ".avi"):
             is_video = True
             preload_video(media_path)
             media_loaded = True
             status = f"Successfully loaded video: {media_path}"
-        elif media_path.lower().endswith((".jpg", ".jpeg", ".png")):
+        elif file_extension in (".jpg", ".jpeg", ".png"):
             is_video = False
             static_image = cv2.imread(media_path)
             if static_image is None:
@@ -222,22 +242,36 @@ def monitor_feed() -> Tuple[
         frame = last_frame.copy()
         metrics = last_metrics.copy()
     else:
-        try:
-            if is_video:
-                frame = get_next_video_frame()
-                if frame is None:
-                    # Reset video if it ends
-                    log_entries.append("Video ended, resetting to start.")
-                    logging.info("Video ended, resetting to start.")
-                    reset_video_index()
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                if is_video:
                     frame = get_next_video_frame()
                     if frame is None:
-                        raise RuntimeError("Failed to retrieve frame after reset.")
-            else:
-                frame = static_image.copy()
-        except RuntimeError as e:
-            log_entries.append(f"Error: {str(e)}")
-            logging.error(f"Frame retrieval error: {str(e)}")
+                        log_entries.append(f"Frame retrieval failed on attempt {attempt + 1}, resetting video.")
+                        logging.warning(f"Frame retrieval failed on attempt {attempt + 1}, resetting video.")
+                        reset_video_index()
+                        continue
+                    break
+                else:
+                    frame = static_image.copy()
+                    break
+            except Exception as e:
+                log_entries.append(f"Frame retrieval error on attempt {attempt + 1}: {str(e)}")
+                logging.error(f"Frame retrieval error on attempt {attempt + 1}: {str(e)}")
+                if attempt == max_retries - 1:
+                    return (
+                        None,
+                        json.dumps(last_metrics, indent=2),
+                        "\n".join(log_entries[-10:]),
+                        detected_plants,
+                        detected_issues,
+                        None,
+                        None
+                    )
+        else:
+            log_entries.append("Failed to retrieve frame after maximum retries.")
+            logging.error("Failed to retrieve frame after maximum retries.")
             return (
                 None,
                 json.dumps(last_metrics, indent=2),