feat: add option to run the worker from translation_queue module

opus_api/__init__.py

@@ -1,2 +1,5 @@
-from opus_api.app_uvi import *
+# __init__.py → importei o pacote
+# __main__.py → executei o pacote
+from .app_uvi import main
+
 

opus_api/app_uvi.py

@@ -81,4 +81,6 @@ def main():
     uvicorn.run("opus_api.app_uvi:app", host="0.0.0.0", port=7860, reload=False)
 
 if __name__ == "__main__":
-    main()
+    main()
+
+

opus_api/translation_queue.py

@@ -1,6 +1,7 @@
 import time, os
 import asyncio
 from huey import SqliteHuey
+from huey.consumer import Consumer
 from opus_api.model import run, run_batch
 from typing import List
 
@@ -39,4 +40,10 @@ async def status(task_id: str):
         "status": "pending",
     }]
 
+def run_worker():
+    consumer = Consumer(huey, workers=4)
+    consumer.run()
 
+
+if __name__ == "__main__":
+    run_worker()

setup.py

@@ -1,13 +1,17 @@
 from setuptools import setup, find_packages
 
+
+with open('requirements.txt', 'r', encoding='utf-8') as fh:
+    install_requires = fh.read().splitlines()
+
 setup(
     name="opus_api",
     version="1.0",
     packages=["opus_api"],
-    #install_requires=[],
+    install_requires=install_requires,
     entry_points={
         "console_scripts": [
-            "app_uvi=opus_api.app_uvi.__main__:main",  # Substitua pelo ponto de entrada correto
+            "opus_api=opus_api.app_uvi.__main__:main",  # Substitua pelo ponto de entrada correto
         ],
     },
 )