sugoi-v4 / server /app.py
playmak3r's picture
fix: /filename path
557abd4
raw
history blame
2.37 kB
from threading import Thread
import redis, time
from flask import Flask, jsonify, send_from_directory, Response, request
from waitress import serve
from flask_cors import CORS, cross_origin
from model import SugoiTranslator
from typing import List
queue_key = "translation_queue"
translated_key = "translated_temp_store"
redis_client = redis.Redis(host='localhost', port=6379, db=0)
def queue_process():
while True:
task_list = []
for _ in range(5):
task = redis_client.rpop(queue_key) # right pop
if task: task_list.append(task)
else: break
if len(task_list): task_process(task_list)
time.sleep(0.5)
def task_process(input_text_list: List[str]):
sugoiTranslator = SugoiTranslator()
translations = sugoiTranslator.translate(input_text_list)
for index, translation in enumerate(translations):
redis_client.hset(translated_key, input_text_list[index], translation)
def query_translation(input_text: str):
for _ in range(30):
translated_text = redis_client.hget(translated_key, input_text)
if translated_text is not None:
redis_client.hdel(translated_key, input_text)
try: return translated_text.decode('utf-8')
except AttributeError: return translated_text
time.sleep(1)
app = Flask(__name__)
CORS(app)
@app.route('/')
def index():
return send_from_directory('dist', "index.html")
@app.route('/<path:filename>')
def download_file(filename):
return send_from_directory('dist', filename)
@app.route('/assets/<path:filename>')
def download_assets(filename):
return send_from_directory('dist/assets', filename)
@app.route('/api/translate', methods= ['POST'])
@cross_origin()
def translate_api():
input_text = request.args.get('text')
if isinstance(input_text, str) and len(input_text) > 0:
redis_client.lpush(queue_key, input_text) # left push
result = query_translation(input_text)
if result is not None: return jsonify({ "text": result })
else: return Response(status= 529, response= "Error 529: Server overloaded")
return Response(status= 400)
if __name__ == '__main__':
Thread(target=queue_process, daemon=True).start()
print("Starting server...")
try: serve(app, port= 7860)
except BaseException as e:
print(e)