adds ugly ui

main.py

@@ -22,6 +22,8 @@ from utilities.constants import UUID
 from utilities.constants import VALUE_APP
 from utilities.constants import VALUE_JOB_PENDING
 from utilities.constants import VALUE_JOB_RUNNING
+from utilities.constants import VALUE_JOB_DONE
+from utilities.web import web
 from utilities.envvar import get_env_var_with_default
 from utilities.envvar import get_env_var
 from utilities.times import wait_for_seconds
@@ -179,7 +181,7 @@ def get_jobs():
             jobs.append(job)
         all_matching_completed_jobs = local_completed_jobs.get(req[API_KEY], {})
         if UUID in req:
-            all_matching_completed_jobs = all_matching_completed_jobs.get(req[UUID])
+            all_matching_completed_jobs = all_matching_completed_jobs.get(req[UUID], {})
         for key in all_matching_completed_jobs.keys():
             jobs.append(all_matching_completed_jobs[key])
 
@@ -190,6 +192,10 @@ def get_jobs():
         )
     return jsonify({"jobs": jobs})
 
+@app.route("/")
+def index():
+    return web()
+
 
 def backend(event_termination):
     model = load_model(logger)
@@ -240,7 +246,7 @@ def main():
     # ugly solution for now
     # TODO: use a database to track instead of internal memory
     try:
-        app.run()
+        app.run(host='0.0.0.0')
         thread.join()
     except KeyboardInterrupt:
         event_termination.set()

utilities/BUILD

@@ -82,3 +82,8 @@ py_test(
     srcs=["times_test.py"],
     deps=[":times"],
 )
+
+py_library(
+    name="web",
+    srcs=["web.py"],
+)

utilities/web.py

@@ -0,0 +1,156 @@
+def javascript():
+    return """
+    <script>
+
+function waitForImage(uuidValue) {
+  // Wait until image is done
+  $.ajax({
+    type: 'POST',
+    url: '/get_jobs',
+    contentType: 'application/json; charset=utf-8',
+    dataType:'json',
+    data: JSON.stringify({'api_key': 'demo', 'uuid': uuidValue}),
+    success: function(response) {
+        console.log(response);
+        $('#newjobresult').html('<p>waiting for result...<p>');
+        if (response.jobs.length == 1) {
+            if (response.jobs[0].status == "done") {
+                $('#newjobresultimg').attr('src', response.jobs[0].img);
+                return;
+            } else {
+                $('#newjobresult').append("<p>current status: " + response.jobs[0].status + "</p>")
+            }
+        }
+        setTimeout(waitForImage, 1000);  // refresh every second
+    },
+    error: function(xhr, status, error) {
+        // Handle error response
+        console.log(xhr.responseText);
+    }
+  });
+}
+
+$( document ).ready(function() {
+  console.log( "--- csrf token set ---" );
+
+  var csrftoken = $("[name=csrfmiddlewaretoken]").val();
+  function csrfSafeMethod(method) {
+      // these HTTP methods do not require CSRF protection
+      return (/^(GET|HEAD|OPTIONS|TRACE)$/.test(method));
+  }
+  $.ajaxSetup({
+      beforeSend: function(xhr, settings) {
+          if (!csrfSafeMethod(settings.type) && !this.crossDomain) {
+              xhr.setRequestHeader("X-CSRFToken", csrftoken);
+          }
+      }
+  });
+
+  $('#newjob').click(function(e) {
+    e.preventDefault(); // Prevent the default form submission
+
+    // Gather input field values
+    var promptVal = $('#prompt').val();
+    var negPromptVal = $('#negprompt').val();
+
+    if (promptVal == "") {
+        alert("needs to write a prompt!");
+        return;
+    }
+
+    // Send POST request using Ajax
+    $.ajax({
+      type: 'POST',
+      url: '/add_job',
+      contentType: 'application/json; charset=utf-8',
+      dataType:'json',
+      data: JSON.stringify({'api_key': 'demo', 'prompt': promptVal, 'neg_prompt': negPromptVal}),
+      success: function(response) {
+        console.log(response);
+        if (response.uuid) {
+            $('#jobuuid').val(response.uuid);
+        }
+        waitForImage(response.uuid);
+      },
+      error: function(xhr, status, error) {
+        // Handle error response
+        console.log(xhr.responseText);
+      }
+    });
+  });
+
+
+});
+</script>
+    """
+
+def stylesheet():
+    return """
+    """
+
+def content():
+    return """
+    <form>
+  <div class="mb-3">
+    <label for="prompt" class="form-label">Prompt</label>
+    <input type="prompt" class="form-control" id="prompt" aria-describedby="promptHelp">
+    <div id="promptHelp" class="form-text">Less than 77 words otherwise it'll be truncated</div>
+  </div>
+  <div class="mb-3">
+    <label for="negprompt" class="form-label">Negative Prompt</label>
+    <input type="negprompt" class="form-control" id="negprompt" aria-describedby="negpromptHelp">
+    <div id="negpromptHelp" class="form-text">Less than 77 words otherwise it'll be truncated</div>
+  </div>
+  <button id="newjob" type="submit" class="btn btn-primary">Submit New Job</button>
+</form>
+
+<div class="card mb-3">
+<div class="card-body" id="newjobresult"></div>
+<img id="newjobresultimg" class="card-img-bottom" />
+</div>
+
+<form>
+  <div class="mb-3">
+    <label for="jobuuid" class="form-label">Job UUID</label>
+    <input type="jobuuid" class="form-control" id="jobuuid" aria-describedby="">
+  </div>
+  <button id="getjob" type="submit" class="btn btn-primary">Get Jobs</button>
+  <button id="canceljob" type="submit" class="btn btn-primary">Cancel Job</button>
+</form>
+
+<div class="mb-3" id="joblist">
+</div>
+    """
+
+def web():
+    return """
+<html lang="en">
+
+<head>
+  <meta charset="utf-8">
+
+  <title>Happy Diffusion (Private Access) | 9pm</title>
+  <meta name="description" content="Stable Diffusion Online">
+  <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
+
+  <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.2.2/dist/css/bootstrap.min.css" rel="stylesheet"
+    integrity="sha384-Zenh87qX5JnK2Jl0vWa8Ck2rdkQ2Bzep5IDxbcnCeuOxjzrPF/et3URy9Bv1WTRi" crossorigin="anonymous">
+{css}
+</head>
+
+<body>
+  <div class="container">{content}</div>
+
+  <script src="https://code.jquery.com/jquery-3.6.1.min.js"
+    integrity="sha256-o88AwQnZB+VDvE9tvIXrMQaPlFFSUTR+nldQm1LuPXQ=" crossorigin="anonymous"></script>
+  <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.2.2/dist/js/bootstrap.bundle.min.js"
+    integrity="sha384-OERcA2EqjJCMA+/3y+gxIOqMEjwtxJY7qPCqsdltbNJuaOe923+mo//f6V8Qbsw3"
+    crossorigin="anonymous"></script>
+{js}
+</body>
+</html>
+    """.format(
+        content=content(),
+        css=stylesheet(),
+        js=javascript(),
+    )