feat: Allow for manual generation of videos

Author: noahpistilli

templates/video_list.html

@@ -4,6 +4,38 @@
 {% block type_name_singular %}movie{% endblock %}
 
 {% block table_listing %}
+    <script src="https://code.jquery.com/jquery-3.6.0.min.js"></script>
+    <script>
+                function runProcess() {
+                    // Disable the button while processing
+                    $('#run-btn').prop('disabled', true);
+                    $('#status').text('Processing...');
+
+                    // Start the process
+                    $.post('/thegateway/generate', function(data) {
+                        if (data.in_progress) {
+                            alert("Generation is already in progress by another user.");
+                        } else {
+                            // Do the check
+                            checkStatus();
+                        }
+                    });
+                }
+
+                function checkStatus() {
+                    $.get('/thegateway/check_status', function(data) {
+                        if (data.completed) {
+                            $('#status').text(data.message);
+                            alert(data.message);  // Show popup
+                            $('#run-btn').prop('disabled', false);
+                        } else {
+                            $('#status').text('Processing... ' + data.message);
+                            // Check again after a delay
+                            setTimeout(checkStatus, 1000);
+                        }
+                    });
+                }
+    </script>
     <br>
     <a href="{{ url_for('thegateway.admin') }}">Return to the Main Menu</a>
     <table class="table is-fullwidth is-hoverable is-striped">
@@ -79,4 +111,8 @@
             <span>Add a Video</span>
         </button>
     </a>
+
+    <button id="run-btn" class="button is-info" onclick="runProcess()">
+        <span>Generate Videos</span>
+    </button>
 {% endblock %}

thegateway/videos.py

@@ -1,3 +1,5 @@
+from time import sleep
+
 from models import db, Videos
 from flask import (
     url_for,
@@ -6,12 +8,22 @@
     redirect,
     request,
     send_from_directory,
+    jsonify,
 )
 from thegateway import thegateway_blueprint
 from thegateway.mobiclip import validate_mobiclip, save_video_data, get_mobiclip_length
 from thegateway.form import VideoForm
 from thegateway.admin import oidc
 from werkzeug.utils import redirect
+import threading
+import subprocess
+
+
+generate_status = {
+    "completed": False,
+    'message': "",
+    "in_progress": False,
+}
 
 
 @thegateway_blueprint.route("/thegateway/videos/")
@@ -77,3 +89,36 @@ def add_video():
 @oidc.require_login
 def get_video_thumbnail(movie_id):
     return send_from_directory("./assets/videos/", f"{movie_id}.img")
+
+
+@thegateway_blueprint.post("/thegateway/generate")
+@oidc.require_login
+def generate_videos():
+    def actually_generate_videos():
+        message = "Successfully generated thumbnails and videos!"
+        # Sleep for a second to give the web app time to process the fact that another user isn't generating.
+        sleep(1)
+        generate_status["in_progress"] = True
+
+        # Generate videos first
+        if subprocess.run(["./cli", "2"]).returncode != 0:
+            message = "Error generating videos."
+        else:
+            # Now thumbnails
+            if subprocess.run(["./cli", "3"]).returncode != 0:
+                message = "Error generating thumbnails."
+
+        generate_status["completed"] = True
+        generate_status["message"] = message
+        generate_status["in_progress"] = False
+
+    if not generate_status["in_progress"]:
+        threading.Thread(target=actually_generate_videos, daemon=True).start()
+
+    return jsonify(generate_status)
+
+
+@thegateway_blueprint.route("/thegateway/check_status")
+def check_status():
+    """Endpoint to check the current process status"""
+    return jsonify(generate_status)