Update app.py

app.py

@@ -336,40 +336,43 @@ async def test_concurrent_download():
         ]
         await asyncio.gather(*downloads)
 
+def run_async_function(func):
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    return loop.run_until_complete(func)
 
 @app.route('/list_models', methods=['GET'])
 def list_models():
-    include_remote = request.args.get('include_remote', type=bool, default=False)
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    result = loop.run_until_complete(get_build().list_models(include_remote=include_remote))
-    return jsonify(result)
+    include_remote = request.args.get('include_remote', 'false').lower() == 'true'
 
-@app.route('/translate', methods=['POST'])
-def translate():
-    data = request.get_json()
-    text = data.get('text')
-    src = data.get('src')
-    trg = data.get('trg')
-    model = data.get('model')
-    pivot = data.get('pivot')
-    html = data.get('html', False)
+    async def async_func():
+        async with get_build() as tl:
+            models = await tl.list_models(include_remote=include_remote)
+            return models
 
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    result = loop.run_until_complete(get_build().translate(text, src, trg, model=model, pivot=pivot, html=html))
+    result = run_async_function(async_func())
     return jsonify(result)
 
-@app.route('/download_model', methods=['POST'])
-def download_model():
+@app.route('/translate', methods=['POST'])
+def translate():
     data = request.get_json()
-    model_id = data.get('model_id')
 
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    result = loop.run_until_complete(get_build().download_model(model_id))
+    async def async_func():
+        async with get_build() as tl:
+            result = await tl.translate(
+                data['text'],
+                src=data.get('src'),
+                trg=data.get('trg'),
+                model=data.get('model'),
+                pivot=data.get('pivot'),
+                html=data.get('html', False)
+            )
+            return result
+
+    result = run_async_function(async_func())
     return jsonify(result)
 
+# Define more routes for other operations like download_model, etc.
 if __name__ == "__main__":
    config = Config()
    config.bind = ["0.0.0.0:7860"]  # You can specify the host and port here