fix: start processing thread

This commit is contained in:
Akbar Rahman 2023-12-27 23:48:37 +00:00
parent 2b844a899f
commit 109e06a5d6
Signed by: alvierahman90
GPG Key ID: 6217899F07CA2BDF
6 changed files with 12 additions and 9 deletions

View File

@ -1,3 +1,4 @@
env
.git
**/__pycache__
q

1
.gitignore vendored
View File

@ -1,2 +1,3 @@
env
**/__pycache__
q

View File

@ -13,4 +13,4 @@ RUN apt update && apt install -y ffmpeg libsm6 libxext6
COPY src .
COPY best.pt /best.pt
CMD [ "gunicorn", "-b 0.0.0.0:80", "app:app" ]
CMD [ "gunicorn", "-b 0.0.0.0:80", "--workers", "1", "app:app" ]

View File

@ -5,4 +5,4 @@ services:
build: .
restart: unless-stopped
ports:
- 8759:8000
- 8759:80

View File

@ -10,7 +10,7 @@
<img id="img" src="" />
<p id="statusline">submit an image to get started!</p>
<form id="form" action="http://localhost:5000/submit" method="post">
<form id="form" action="http://localhost:8759/submit" method="post">
<input type="file" id="job" name="image" accept="image/png,image/jpeg" /><br>
<button type="submit" id="submit">submit</button>
</form>
@ -38,11 +38,11 @@
console.log(id);
while (true) {
const resp = await fetch('http://localhost:5000/state/' + id);
const resp = await fetch('http://localhost:8759/state/' + id);
const state = (await resp.json()).state;
statusline.innerHTML = state;
if (state == "READY") {
img.src = 'http://localhost:5000/result/' + id
img.src = 'http://localhost:8759/result/' + id
return;
}
}

View File

@ -44,7 +44,7 @@ def process():
print("waiting for/getting job")
job = q.get()
filepath = root.joinpath(job)
print("got job")
print("got job", flush=True)
state[job] = {
'state': QUEUE_STATE_PROCESSING,
}
@ -65,6 +65,10 @@ def process():
last = time.time()
p = Thread(target=process)
p.start()
@app.route("/submit", methods=['POST'])
def render():
id = str(uuid.uuid4())
@ -87,7 +91,4 @@ def get_result(i):
if __name__ == '__main__':
p = Thread(target=process)
p.start()
app.run(debug=True)
p.join()