Add bidirectional translation (FR ↔ DYU)

- New endpoint /api/translate/fr-dyu for French → Dioula
- Renamed /api/translate to /api/translate/dyu-fr for clarity
- Standardized JSON response format for both endpoints
- Updated documentation with workflow examples
- Version 1.1.0

app.py

@@ -143,14 +143,23 @@ def home():
         </div>
 
         <div class="endpoint">
-            <p><span class="method post">POST</span> <code>/api/translate</code></p>
-            <p><strong>Traduction</strong> - Traduit du Dioula vers le Français</p>
+            <p><span class="method post">POST</span> <code>/api/translate/dyu-fr</code></p>
+            <p><strong>Traduction Dioula → Français</strong></p>
             <p><strong>Entrée:</strong> Texte en Dioula (paramètre <code>text</code>)</p>
-            <p><strong>Sortie:</strong> <code>{"texte_dioula": "...", "traduction_francais": "..."}</code></p>
-            <pre>curl -X POST https://votre-space.hf.space/api/translate \\
+            <p><strong>Sortie:</strong> JSON avec traduction française</p>
+            <pre>curl -X POST https://votre-space.hf.space/api/translate/dyu-fr \\
   -F "text=Sanji bɛna kɛ bi"</pre>
         </div>
 
+        <div class="endpoint">
+            <p><span class="method post">POST</span> <code>/api/translate/fr-dyu</code></p>
+            <p><strong>Traduction Français → Dioula</strong></p>
+            <p><strong>Entrée:</strong> Texte en Français (paramètre <code>text</code>)</p>
+            <p><strong>Sortie:</strong> JSON avec traduction dioula</p>
+            <pre>curl -X POST https://votre-space.hf.space/api/translate/fr-dyu \\
+  -F "text=Il va pleuvoir aujourd'hui"</pre>
+        </div>
+
         <h2>🔗 Documentation interactive</h2>
         <p>
             <a href="/docs">Swagger UI</a> |
@@ -161,8 +170,16 @@ def home():
         <ul>
             <li><strong>STT:</strong> facebook/mms-1b-all (adapter Dioula)</li>
             <li><strong>TTS:</strong> facebook/mms-tts-dyu</li>
-            <li><strong>Traduction:</strong> facebook/nllb-200-distilled-600M (Dioula → Français)</li>
+            <li><strong>Traduction:</strong> facebook/nllb-200-distilled-600M (Dioula ↔ Français)</li>
         </ul>
+
+        <h2>🔄 Flux de travail complet</h2>
+        <p><strong>Exemple :</strong> Audio Dioula → Texte Dioula → Traduction Français → Audio Français</p>
+        <ol>
+            <li><code>/api/stt</code> : Convertit audio en texte dioula</li>
+            <li><code>/api/translate/dyu-fr</code> : Traduit en français</li>
+            <li>(Optionnel) Utiliser un TTS français pour générer l'audio</li>
+        </ol>
     </body>
     </html>
     """
@@ -296,7 +313,7 @@ async def text_to_speech(text: str = Form(...)):
         print(f"Erreur TTS: {e}")
         raise HTTPException(status_code=500, detail=f"Erreur lors de la génération audio: {str(e)}")
 
-@app.post("/api/translate")
+@app.post("/api/translate/dyu-fr")
 async def translate_dioula_to_french(text: str = Form(...)):
     """
     Traduit du texte du Dioula vers le Français
@@ -328,14 +345,60 @@ async def translate_dioula_to_french(text: str = Form(...)):
         traduction = translate_tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
 
         return {
-            "texte_dioula": text,
-            "traduction_francais": traduction
+            "texte_source": text,
+            "langue_source": "dioula",
+            "texte_traduit": traduction,
+            "langue_cible": "français"
+        }
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        print(f"Erreur Traduction DYU→FR: {e}")
+        raise HTTPException(status_code=500, detail=f"Erreur lors de la traduction: {str(e)}")
+
+@app.post("/api/translate/fr-dyu")
+async def translate_french_to_dioula(text: str = Form(...)):
+    """
+    Traduit du texte du Français vers le Dioula
+
+    - **text**: Texte en Français à traduire
+    """
+    try:
+        if not text.strip():
+            raise HTTPException(status_code=400, detail="Le texte ne peut pas être vide")
+
+        # Définir la langue source : Français
+        translate_tokenizer.src_lang = "fra_Latn"
+
+        # Préparation des tokens
+        inputs = translate_tokenizer(text, return_tensors="pt").to(device)
+
+        # Récupérer l'ID de la langue cible : Dioula
+        target_lang_id = translate_tokenizer.convert_tokens_to_ids("dyu_Latn")
+
+        # Génération de la traduction
+        with torch.no_grad():
+            translated_tokens = translate_model.generate(
+                **inputs,
+                forced_bos_token_id=target_lang_id,
+                max_length=200
+            )
+
+        # Décodage
+        traduction = translate_tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
+
+        return {
+            "texte_source": text,
+            "langue_source": "français",
+            "texte_traduit": traduction,
+            "langue_cible": "dioula"
         }
 
     except HTTPException:
         raise
     except Exception as e:
-        print(f"Erreur Traduction: {e}")
+        print(f"Erreur Traduction FR→DYU: {e}")
         raise HTTPException(status_code=500, detail=f"Erreur lors de la traduction: {str(e)}")
 
 if __name__ == "__main__":