add update and delete stream endpoints

main.py

@@ -37,57 +37,48 @@ description = """
 
 <br>
 
-### 📽️ Section 1: Live Stream Monitoring
+###  Section 1: Live Stream Monitoring
 *Use these endpoints to manage and watch live AI security feeds.*
 
 1. **`POST /stream/create` (Register Camera)**
-   - **What it does:** Saves your camera's RTSP address and location into the system.
-   - **How to use:** Enter a name (e.g., "Cashier-1"), the RTSP link from your camera, and the location.
+   - **What it does:** Saves your camera's RTSP address into the system.
+   - **How to use:** Enter a name, the RTSP link, and the location.
 
 2. **`POST /stream/start/{id}` (Activate AI)**
-   - **What it does:** Powers up the AI "Brain" for a specific camera.
-   - **How to use:** Enter the ID (like `cam-1`). It returns a **view_url**. Paste this URL into a new browser tab to see the live AI.
+   - **What it does:** Starts the AI processing. 
+   - **How to use:** Returns stream URL to view the stream.
 
-3. **`GET /cameras/{id}/frame` (Main Video Feed)**
-   - **What it does:** The actual live video stream with AI boxes and the security card.
-   - **How to use:** This is the URL used by your browser or dashboard to display the video.
+3. **`PUT /stream/update/{id}` (Change Settings)**
+   - **What it does:** Updates the name or RTSP link of an existing camera.
 
-4. **`GET /stream/list_cameras` (Camera Directory)**
-   - **What it does:** Displays a list of all added cameras.
-   - **How to use:** Use this to see which cameras are currently "Online" or to find a Camera ID.
+4. **`DELETE /stream/delete/{id}` (Remove Camera)**
+   - **What it does:** Stops the stream and removes the camera from the list.
 
-5. **`POST /stream/stop/{id}` (Deactivate AI)**
-   - **What it does:** Shuts down the AI processing for a camera to save computer resources.
-   - **How to use:** Enter the ID and click Execute. The camera will move to "Offline" status.
+5. **`GET /cameras/{id}/frame` (Main Video Feed)**
+   - **What it does:** The actual live video stream.
+
+6. **`GET /stream/list_cameras` (Camera Directory)**
+   - **What it does:** Displays all added cameras and their status.
+
+7. **`POST /stream/stop/{id}` (Deactivate AI)**
+   - **What it does:** Shuts down the AI for a camera.
 
 <br>
 
-### 📁 Section 2: Recorded Video Analysis
+###  Section 2: Recorded Video Analysis
 *Use these endpoints to scan uploaded files for theft incidents.*
 
 1. **`POST /video/detect` (Upload for Analysis)**
-   - **What it does:** Uploads a video file (MP4/MOV) to the server for a full AI scan.
-   - **How to use:** Select your video file and click Execute. You will receive a **job_id**.
+   - **What it does:** Uploads a video file for a full AI scan. Returns a **job_id**.
 
 2. **`GET /video/status/{job_id}` (Check Progress)**
-   - **What it does:** Provides the scan percentage (0% to 100%) and current state.
-   - **How to use:** Enter your **job_id**. Once it reaches 100%, your file is ready.
+   - **What it does:** Provides the scan percentage (0% to 100%).
 
 3. **`GET /video/jobs` (Task History)**
-   - **What it does:** Lists every video ever uploaded and whether the scan is finished or failed.
-   - **How to use:** Helpful for finding old `job_id`s or checking multiple scans at once.
+   - **What it does:** Lists every video ever uploaded.
 
 4. **`GET /video/download/{job_id}` (Get Result)**
-   - **What it does:** Allows you to download the final processed video.
-   - **How to use:** Once status is 100%, enter the `job_id` here to save the analyzed video to your PC.
-
-<br>
-
-### 🔔 Section 3: Discord Notifications
-*Automatic alerts for security teams.*
-
-- **How it works:** Whenever the AI detects a theft (Live or Recorded), it takes a snapshot of the person and sends a high-priority alert card to your Discord channel.
-- **Requirement:** Ensure your `DISCORD_WEBHOOK_URL` is set in the system settings.
+   - **What it does:** Download the final processed video once status is 100%.
 
 </details>
 
@@ -110,6 +101,7 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 app = FastAPI(title="AI Theft Detection Backend",
                description=description,
+               openapi_tags=tags_metadata,
                 version="1.0.0")
 
 # --- DATABASE & STATE ---
@@ -201,51 +193,34 @@ def draw_corner_rect(img, pt1, pt2, color, thickness=2, r=15, d=25):
     cv2.ellipse(img, (x2-r, y2-r), (r,r), 0, 0, 90, color, thickness)
 
 def draw_security_card(frame, avg_prob, theft_flag, title="AI ANALYZER"):
-    # 1. Card Layout Configuration
     card_x, card_y = 35, 35
     padding = 30
     line_spacing = 45
-    card_w, card_h = 620, 260 # Increased size for better spacing
-    
+    card_w, card_h = 620, 260
     orange = (0, 165, 255)
-    bg_color = (25, 25, 25) # Slightly darker for better contrast
-    
-    # 2. Draw Background with Transparency
+    bg_color = (25, 25, 25)
     overlay = frame.copy()
     cv2.rectangle(overlay, (card_x, card_y), (card_x + card_w, card_y + card_h), bg_color, -1)
     cv2.addWeighted(overlay, 0.85, frame, 0.15, 0, frame)
-    
-    # 3. Draw Bold Orange Border
     cv2.rectangle(frame, (card_x, card_y), (card_x + card_w, card_y + card_h), orange, 3)
-
-    # 4. Prepare Content
     status_label = "ALERT: THEFT DETECTED" if theft_flag else "STATUS: SYSTEM NORMAL"
     status_color = (0, 0, 255) if theft_flag else (0, 255, 0)
     now = datetime.datetime.now().strftime("%b %d, %Y  |  %I:%M:%S %p")
     confidence = f"AI CONFIDENCE: {int(avg_prob * 100)}%"
-    
-    # 5. Render Text with Padding and Anti-Aliasing (cv2.LINE_AA)
     curr_y = card_y + padding + 15
-    
-    # Header - Centered Bold
     header_text = "THEFT DETECTION LIVE MONITORING"
     (tw, th), _ = cv2.getTextSize(header_text, cv2.FONT_HERSHEY_DUPLEX, 0.9, 2)
     cv2.putText(frame, header_text, (card_x + (card_w - tw) // 2, curr_y), 
                 cv2.FONT_HERSHEY_DUPLEX, 0.9, (255, 255, 255), 2, cv2.LINE_AA)
-    
-    # Rows with Spacing
     curr_y += line_spacing + 5
     cv2.putText(frame, f"SOURCE: {title.upper()}", (card_x + padding, curr_y), 
                 cv2.FONT_HERSHEY_DUPLEX, 0.8, (200, 200, 200), 1, cv2.LINE_AA)
-    
     curr_y += line_spacing
     cv2.putText(frame, status_label, (card_x + padding, curr_y), 
                 cv2.FONT_HERSHEY_DUPLEX, 1.0, status_color, 2, cv2.LINE_AA)
-    
     curr_y += line_spacing
     cv2.putText(frame, confidence, (card_x + padding, curr_y), 
                 cv2.FONT_HERSHEY_DUPLEX, 0.8, (255, 255, 255), 1, cv2.LINE_AA)
-    
     curr_y += line_spacing
     cv2.putText(frame, now, (card_x + padding, curr_y), 
                 cv2.FONT_HERSHEY_SIMPLEX, 0.6, (150, 150, 150), 1, cv2.LINE_AA)
@@ -368,7 +343,7 @@ class CameraPipeline:
                                 with torch.no_grad():
                                     probs = torch.softmax(slowfast_model([clip_ts[:,:,::4,:,:], clip_ts]), dim=1)
                                 self.prediction_buffer.append(probs[0][1].item())
-                                avg_prob = np.mean(self.prediction_buffer)
+                                avg_prob = np.mean(prediction_buffer)
                         is_theft = avg_prob > THEFT_THRESHOLD
                         draw_corner_rect(frame, (x1,y1), (x2,y2), (0,0,255) if is_theft else (0,255,0))
                         if is_theft:
@@ -429,18 +404,32 @@ async def start_camera(id: str, request: Request):
 
     # Generate URLs for the user
     local_url = f"{request.base_url}cameras/{id}/frame"
-    hf_host = request.headers.get("x-forwarded-host")
-    hf_url = f"https://{hf_host}/cameras/{id}/frame" if hf_host else "Not running on Hugging Face"
 
     return {
         "success": started,
         "job_id": id,
         "view_urls": {
-            "huggingface_url": hf_url,
-            "local_machine_url": str(local_url)
+            "stream_url": str(local_url)
         }
     }
 
+@app.put("/stream/update/{id}", tags=["Live Stream Monitoring"])
+def update_camera(id: str, cam_data: CameraUpdate):
+    if id not in MOCK_DB["cameras"]:
+        raise HTTPException(status_code=404, detail="Camera not found")
+    current_cam = MOCK_DB["cameras"][id]
+    update_dict = cam_data.dict(exclude_unset=True)
+    current_cam.update(update_dict)
+    return {"message": "Camera updated successfully", "camera": current_cam}
+
+@app.delete("/stream/delete/{id}", tags=["Live Stream Monitoring"])
+def delete_camera(id: str):
+    if id not in MOCK_DB["cameras"]:
+        raise HTTPException(status_code=404, detail="Camera not found")
+    manager.stop_camera(id)
+    del MOCK_DB["cameras"][id]
+    return {"message": f"Camera {id} deleted successfully"}
+
 @app.post("/stream/stop/{id}",  tags=["Live Stream Monitoring"])
 def stop_camera(id: str):
     if id not in MOCK_DB["cameras"]:
@@ -451,19 +440,6 @@ def stop_camera(id: str):
         MOCK_DB["cameras"][id]["isStreaming"] = False
     return {"success": stopped}
 
-# @app.get("/stream/frame/{id}",  tags=["Live Stream Monitoring"])
-# def get_frame(id: str):
-#     if id not in manager.active_pipelines:
-#         raise HTTPException(404)
-#     def generate():
-#         while id in manager.active_pipelines:
-#             frame = manager.active_pipelines[id].latest_frame
-#             if frame is not None:
-#                 _, buffer = cv2.imencode('.jpg', frame)
-#                 yield (b'--frame\r\nContent-Type: image/jpeg\r\n\r\n' + buffer.tobytes() + b'\r\n')
-#             time.sleep(0.05)
-#     return StreamingResponse(generate(), media_type="multipart/x-mixed-replace; boundary=frame")
-
 @app.get("/cameras/{id}/frame",  tags=["Live Stream Monitoring"])
 def get_frame_legacy(id: str):
     if id not in manager.active_pipelines: