Update backend/models.py

backend/models.py

@@ -41,14 +41,28 @@ def qa_pipeline(question, context):
     response = requests.post(url, headers=headers, json=payload)
     return response.json()
 
-# دالة الترجمة
-def translator(text):
-    url = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-fr"
+# دالة الترجمة بين اللغات المدعومة
+def translator(text, source_lang, target_lang):
+    # تحقق من أن اللغات مدعومة
+    supported_langs = ['en', 'fr', 'es', 'ar', 'zh']
+    if source_lang not in supported_langs or target_lang not in supported_langs:
+        return {"error": "اللغة غير مدعومة. اللغات المدعومة: en, fr, es, ar, zh"}
+
+    # توليد اسم النموذج بناءً على اللغات
+    model_name = f"Helsinki-NLP/opus-mt-{source_lang}-{target_lang}"
+    url = f"https://api-inference.huggingface.co/models/{model_name}"
     headers = {"Authorization": f"Bearer {HUGGING_FACE_API_KEY}"}
     payload = {"inputs": text}
 
+    # إرسال الطلب
     response = requests.post(url, headers=headers, json=payload)
-    return response.json()
+
+    # معالجة الأخطاء المحتملة
+    try:
+        return response.json()
+    except Exception as e:
+        return {"error": str(e)}
+
 
 # دالة توليد كود التصور البياني
 def code_generator(prompt):