Update app.py

app.py

@@ -1,40 +1,21 @@
-from fastapi import FastAPI
-from transformers import pipeline
-
-# ساخت اپ FastAPI
-app = FastAPI(
-    title="English–Persian Translator API",
-    description="A lightweight translation API using Helsinki-NLP models.",
-    version="1.0.0"
-)
-
-# لود مدل ترجمه سبک
-# مدل‌های Helsinki-NLP بسیار سبک و سازگار با اسپیس‌های رایگان هستند
-translator_en_fa = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fa")
-translator_fa_en = pipeline("translation", model="Helsinki-NLP/opus-mt-fa-en")
-
-@app.get("/")
-def home():
-    return {
-        "message": "Welcome to the English–Persian Translator API!",
-        "usage": {
-            "English → Persian": "/translate/en2fa?text=Hello world",
-            "Persian → English": "/translate/fa2en?text=سلام دنیا"
-        }
-    }
-
-@app.get("/translate/en2fa")
-def translate_en_to_fa(text: str):
-    """
-    ترجمه از انگلیسی به فارسی
-    """
-    output = translator_en_fa(text)
-    return {"input": text, "translation": output[0]["translation_text"]}
-
-@app.get("/translate/fa2en")
-def translate_fa_to_en(text: str):
-    """
-    ترجمه از فارسی به انگلیسی
-    """
-    output = translator_fa_en(text)
-    return {"input": text, "translation": output[0]["translation_text"]}
+from fastapi import FastAPI, Query
+from transformers import pipeline
+
+# ساخت اپ FastAPI
+app = FastAPI(title="English–Persian Translator")
+
+# لود مدل سبک ترجمه (عمومی و بدون نیاز به لاگین)
+translator_en_fa = pipeline(
+    "translation_en_to_fa",
+    model="Helsinki-NLP/opus-mt-en-fa",
+    trust_remote_code=True
+)
+
+@app.get("/")
+def home():
+    return {"message": "Welcome to English–Persian Translator!"}
+
+@app.get("/translate")
+def translate(text: str = Query(..., description="English text to translate to Persian")):
+    result = translator_en_fa(text)
+    return {"input": text, "translation": result[0]["translation_text"]}