Update main.py

main.py

@@ -5,62 +5,83 @@ import numpy as np
 import requests
 import tempfile
 import os
+from typing import List
+import logging
+import urllib3
+
+# Suppress only the single InsecureRequestWarning
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
 app = FastAPI()
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 class VideoURLs(BaseModel):
-    urls: list[str]
+    urls: List[str]
+
+MAX_VIDEOS = 10
 
 def download_video(url):
     try:
-        response = requests.get(url, verify=False)  # Added verify=False
+        response = requests.get(url, verify=False, timeout=30)
         response.raise_for_status()
         with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmp_file:
             tmp_file.write(response.content)
             return tmp_file.name
     except requests.RequestException as e:
+        logger.error(f"Failed to download video from {url}: {str(e)}")
         raise Exception(f"Failed to download video from {url}: {str(e)}")
 
 def combine_videos(urls):
     if not urls:
         return None
 
+    if len(urls) > MAX_VIDEOS:
+        raise HTTPException(status_code=400, detail=f"Maximum of {MAX_VIDEOS} videos allowed")
+
     temp_files = []
     for url in urls:
         if url.strip():
             try:
                 temp_files.append(download_video(url.strip()))
             except Exception as e:
+                logger.error(f"Error downloading video: {str(e)}")
                 raise HTTPException(status_code=400, detail=str(e))
 
     if not temp_files:
         raise HTTPException(status_code=400, detail="No valid videos to combine")
 
-    captures = [cv2.VideoCapture(file) for file in temp_files]
+    try:
+        captures = [cv2.VideoCapture(file) for file in temp_files]
 
-    fps = captures[0].get(cv2.CAP_PROP_FPS)
-    frame_size = (int(captures[0].get(cv2.CAP_PROP_FRAME_WIDTH)),
-                  int(captures[0].get(cv2.CAP_PROP_FRAME_HEIGHT)))
+        fps = captures[0].get(cv2.CAP_PROP_FPS)
+        frame_size = (int(captures[0].get(cv2.CAP_PROP_FRAME_WIDTH)),
+                      int(captures[0].get(cv2.CAP_PROP_FRAME_HEIGHT)))
 
-    output_path = 'combined_video.mp4'
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    out = cv2.VideoWriter(output_path, fourcc, fps, frame_size)
+        output_path = 'combined_video.mp4'
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        out = cv2.VideoWriter(output_path, fourcc, fps, frame_size)
 
-    for cap in captures:
-        while True:
-            ret, frame = cap.read()
-            if not ret:
-                break
-            out.write(frame)
+        for cap in captures:
+            while True:
+                ret, frame = cap.read()
+                if not ret:
+                    break
+                out.write(frame)
 
-    for cap in captures:
-        cap.release()
-    out.release()
+        for cap in captures:
+            cap.release()
+        out.release()
 
-    for file in temp_files:
-        os.remove(file)
+        for file in temp_files:
+            os.remove(file)
 
-    return output_path
+        return output_path
+    except Exception as e:
+        logger.error(f"Error combining videos: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error combining videos: {str(e)}")
 
 @app.post("/combine_videos")
 async def process_urls(video_urls: VideoURLs):
@@ -71,10 +92,12 @@ async def process_urls(video_urls: VideoURLs):
         else:
             raise HTTPException(status_code=400, detail="Failed to combine videos")
     except HTTPException as he:
+        logger.error(f"HTTP Exception: {str(he)}")
         raise he
     except Exception as e:
+        logger.error(f"Unexpected error: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8080)  # Changed port to 8080
+    uvicorn.run(app, host="0.0.0.0", port=8080)