Update app.py

app.py

@@ -5,31 +5,24 @@ import tempfile
 from pathlib import Path
 
 def image_to_video(input_file, fps=30):
-    # Create a temporary directory to store the image
     with tempfile.TemporaryDirectory() as temp_dir:
         file_path = os.path.join(temp_dir, "image.jpg")
         with open(file_path, 'wb') as file:
             file.write(input_file)
 
-
-        # Read the image to set the video size
         frame = cv2.imread(file_path)
         height, width, layers = frame.shape
         size = (width, height)
         
-        # Create a unique filename for the output video
         video_filename = os.path.join('/mnt/data', 'output_video.mp4')
         
-        # Initialize the video writer
         out = cv2.VideoWriter(video_filename, cv2.VideoWriter_fourcc(*'mp4v'), fps, size)
-        
-        # Add the image to the video
         out.write(frame)
-        
-        # Release the video writer
         out.release()
-        
-        # Return the path of the created video file
+
+        if not os.path.exists(video_filename):
+            raise FileNotFoundError(f"Failed to create video: {video_filename}")
+
         return video_filename
 
 iface = gr.Interface(