update: eng_sw model

app.py

@@ -4,7 +4,8 @@ from transformers import pipeline
 app = Flask(__name__)
 
 # Initialize the translation pipeline
-translator = pipeline("translation", model="Bildad/Swahili-English_Translation")
+sw_to_en = pipeline("translation", model="Bildad/Swahili-English_Translation")
+en_to_sw = pipeline("translation", model="Bildad/English-Swahili_Translation")
 
 
 @app.route("/", methods=["GET"])
@@ -28,7 +29,7 @@ def home():
             .container {{
                 width: 80%;
                 margin: auto;
-                margin-top: 50px;
+                margin-top: 100px;
                 overflow: hidden;
                 background: #fff;
                 padding: 20px;
@@ -67,9 +68,12 @@ def home():
         <div class="container">
             <h1>ChatEasy swa-eng Translation API</h1>
             <p>Welcome to the ChatEasy swa-eng Translation API documentation.</p>
-            <p>To use the API, send a POST request to <code>https://aboge-demo.hf.space/translate</code>.</p>
+            <p>To use the API, send a POST request to <code>/translate</code>.</p>
             <h2>Request Payload</h2>
-            <pre>&#123; "<span class="payload-key">text</span>" : "<span class="payload-value">Siku njema</span>" &#125;</pre>
+            <pre>&#123; 
+                "<span class="payload-key">text</span>" : "<span class="payload-value">Siku njema</span>",<br>
+                "<span class="payload-key">source_language</span>" : "<span class="payload-value">sw</span>"
+               &#125;</pre>
             <h2>Expected Response</h2>
             <pre>&#123; "<span class="payload-key">translated_text</span>" : "<span class="payload-value">A good day</span>" &#125;</pre>
         </div>
@@ -86,9 +90,19 @@ def translate():
 
     if "text" not in data:
         return jsonify({"error": "No text provided"}), 400
+    if "source_language" not in data:
+        return jsonify({"error": "No Source Language Provided"}), 400
 
     text_to_translate = data["text"]
-    translation = translator(text_to_translate)[0]
+    source_language = data["source_language"]
+
+    if source_language == "en":
+        translation = en_to_sw(text_to_translate)[0]
+    elif source_language == "sw":
+        translation = sw_to_en(text_to_translate)[0]
+    else:
+        return jsonify({"error": "Invalid Source Language"}), 400
+
     translated_text = translation["translation_text"]
 
     return jsonify({"translated_text": translated_text})