goodgute 0.1.0__tar.gz

goodgute-0.1.0/GoodGute/cli.py

@@ -0,0 +1,21 @@
+import argparse
+from translator import translate_text
+import warnings
+
+
+def main():
+
+    parser = argparse.ArgumentParser(description="GoodGute")
+    parser.add_argument("-t", "--textfile", required=True, help="")
+    args = parser.parse_args()
+
+    with open(args.textfile, "r", encoding="utf-8") as f:
+        text = f.read()
+
+    translation = translate_text(text)
+    print("Translation Result:\n", translation)
+    warnings.filterwarnings("ignore")
+
+
+if __name__ == "__main__":
+    main()

goodgute-0.1.0/GoodGute/translator.py

@@ -0,0 +1,24 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from transformers import MarianTokenizer, MarianMTModel
+
+tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-de", clean_up_tokenization_spaces=True)
+model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-de")
+
+def translate_text(text):
+    inputs = tokenizer(text, return_tensors="pt")
+    outputs = model.generate(**inputs)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+
+app = FastAPI()
+
+class TextIn(BaseModel):
+    text: str
+
+class TextOut(BaseModel):
+    translation: str
+
+@app.post("/translate", response_model=TextOut)
+async def translate_api(data: TextIn):
+    return {"translation": translate_text(data.text)}

goodgute-0.1.0/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: goodgute
+Version: 0.1.0
+Summary: English-to-German translation package
+Author: paddy
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown

goodgute-0.1.0/goodgute.egg-info/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: goodgute
+Version: 0.1.0
+Summary: English-to-German translation package
+Author: paddy
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown

goodgute-0.1.0/goodgute.egg-info/SOURCES.txt

@@ -0,0 +1,8 @@
+pyproject.toml
+GoodGute/__init__.py
+GoodGute/cli.py
+GoodGute/translator.py
+goodgute.egg-info/PKG-INFO
+goodgute.egg-info/SOURCES.txt
+goodgute.egg-info/dependency_links.txt
+goodgute.egg-info/top_level.txt

goodgute-0.1.0/goodgute.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

goodgute-0.1.0/goodgute.egg-info/top_level.txt

@@ -0,0 +1 @@
+GoodGute

goodgute-0.1.0/pyproject.toml

@@ -0,0 +1,14 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "goodgute"
+version = "0.1.0"
+description = "English-to-German translation package"
+authors = [{ name = "paddy" }]
+readme = "README.md"
+requires-python = ">=3.8"
+license = { text = "MIT" }
+
+dependencies = []

goodgute-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+