| from flask import Flask, request, jsonify |
| from transformers import pipeline |
|
|
| app = Flask(__name__) |
|
|
| |
| sw_to_en = pipeline("translation", model="Bildad/Swahili-English_Translation") |
| en_to_sw = pipeline("translation", model="Bildad/English-Swahili_Translation") |
|
|
|
|
| @app.route("/", methods=["GET"]) |
| def home(): |
|
|
| html_content = f""" |
| <!DOCTYPE html> |
| <html> |
| <head> |
| <meta charset="UTF-8"> |
| <meta name="viewport" content="width=device-width, initial-scale=1.0"> |
| <title>ChatEasy swa-eng Translation API</title> |
| <style> |
| body {{ |
| font-family: Arial, sans-serif; |
| line-height: 1.6; |
| margin: 0; |
| padding: 0; |
| background-color: #f4f4f4; |
| }} |
| .container {{ |
| width: 80%; |
| margin: auto; |
| margin-top: 50px; |
| overflow: hidden; |
| background: #fff; |
| padding: 20px; |
| border-radius: 8px; |
| box-shadow: 0 0 10px rgba(0, 0, 0, 0.1); |
| }} |
| h1 {{ |
| color: #333; |
| }} |
| h2 {{ |
| color: #555; |
| margin-top: 20px; |
| }} |
| pre {{ |
| background-color: #f9f9f9; |
| border: 1px solid #ddd; |
| padding: 10px; |
| border-radius: 4px; |
| overflow-x: auto; |
| }} |
| code {{ |
| background-color: #f4f4f4; |
| padding: 2px 4px; |
| border-radius: 4px; |
| color: #d14; |
| }} |
| .payload-key {{ |
| color: #d14; |
| }} |
| .payload-value {{ |
| color: green; |
| }} |
| </style> |
| </head> |
| <body> |
| <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>/translate</code>.</p> |
| <h2>Request Payload</h2> |
| <pre> |
| { |
| "<span class="payload-key">text</span>" : "<span class="payload-value">Siku njema</span>", |
| "<span class="payload-key">target</span>" : "<span class="payload-value">en</span>" |
| } |
| </pre> |
| <h2>Expected Response</h2> |
| <pre>{ "<span class="payload-key">translated_text</span>" : "<span class="payload-value">A good day</span>" }</pre> |
| </div> |
| </body> |
| </html> |
| """ |
|
|
| return html_content |
|
|
|
|
| @app.route("/translate", methods=["POST"]) |
| def translate(): |
| data = request.get_json() |
|
|
| if "text" not in data: |
| return jsonify({"error": "No text provided"}), 400 |
| if "target" not in data: |
| return jsonify({"error": "No Target Language Provided"}), 400 |
|
|
| text_to_translate = data["text"] |
| target_language = data["target"] |
|
|
| if target_language == "sw": |
| translation = en_to_sw(text_to_translate)[0] |
| elif target_language == "en": |
| translation = sw_to_en(text_to_translate)[0] |
| else: |
| return jsonify({"error": "Invalid Target Language"}), 400 |
|
|
| translated_text = translation["translation_text"] |
|
|
| return jsonify({"translated_text": translated_text}) |
|
|
|
|
| if __name__ == "__main__": |
| app.run(debug=True) |
|
|
