update: add detail timing and error handling

app.py

@@ -1,5 +1,6 @@
 import os
 import tempfile
+import time
 
 import requests
 from fastapi import FastAPI
@@ -19,9 +20,9 @@ RESOURCES_DIR = "/app/resources"
 os.makedirs(RESOURCES_DIR, exist_ok=True)
 TEMP_VIDEO_DIR = f"{RESOURCES_DIR}/temp_video"
 
-
 os.environ['MOVIEPY_TEMP_DIR'] = '/tmp/moviepy'
 
+
 def upload_to_supabase(video_path, bucket_name="JewelmirrorVideoGeneration"):
     try:
         with open(video_path, 'rb') as f:
@@ -76,6 +77,7 @@ class VideoGenerator(BaseModel):
 
 @app.post("/create-video/")
 async def create_video(request: VideoGenerator):
+    start_time = time.time()
     try:
         temp_files = {
             'necklace': download_image(request.necklace_image),
@@ -83,6 +85,7 @@ async def create_video(request: VideoGenerator):
             'clothing': [download_image(url) for url in request.clothing_output_images],
             'makeup': [download_image(url) for url in request.makeup_output_images]
         }
+        file_download_time = time.time() - start_time
 
         if any(file is None for files in temp_files.values() for file in
                (files if isinstance(files, list) else [files])):
@@ -114,13 +117,21 @@ async def create_video(request: VideoGenerator):
 
         # Generate video
         video_creator.create_final_video()
+        video_creation_time = time.time() - start_time - file_download_time
 
         # Upload to Supabase
         url = upload_to_supabase(video_path=output_path)
+        supabase_upload_time = time.time() - start_time - file_download_time - video_creation_time
+
         response = {
             "status": "success",
-            "message": "Video created successfully",
-            "public_url": url
+            "message": "Video created successfully.",
+            "video_url": url,
+            "timings": {
+                "file_download_time": file_download_time,
+                "video_creation_time": video_creation_time,
+                "supabase_upload_time": supabase_upload_time
+            }
         }
 
         for files in temp_files.values():
@@ -135,7 +146,7 @@ async def create_video(request: VideoGenerator):
         return JSONResponse(content=response, status_code=200)
 
     except Exception as e:
-        raise e
+        return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
 
 
 @app.get("/resources")