Update translator.py

translator.py

@@ -13,7 +13,13 @@ from typing import Dict, List, Optional, Set, Tuple, Any
 from enum import Enum
 import requests
 from tqdm import tqdm
-
+# Add to existing imports
+try:
+    from gradio_client import Client
+    HAS_GRADIO_CLIENT = True
+except ImportError:
+    HAS_GRADIO_CLIENT = False
+    
 # --- Core Library Diagnostics ---
 print("🔍 System Check...")
 
@@ -81,6 +87,7 @@ class TranslationBackend(Enum):
     OPENAI = "openai"
     ANTHROPIC = "anthropic"
     OLLAMA = "ollama"
+    SAUERKRAUT = "sauerkraut"
     AUTO = "auto"
 
 
@@ -517,6 +524,11 @@ class LLMTranslator:
                 f"Provide a natural, fluent translation. "
                 f"Return ONLY the translation:\n\n{text}"
             )
+            
+        # Sauerkraut API
+        if hasattr(self, 'sauerkraut') and self.sauerkraut and self.sauerkraut.available:
+            result = self.sauerkraut.translate_batch([text])[0]
+            if result and result.strip() != text.strip(): return result
         
         for provider_name, provider in self.providers.items():
             try:
@@ -1000,6 +1012,41 @@ class SlideMetadata:
 # DOCUMENT TRANSLATOR
 # ============================================================================
 
+class SauerkrautTranslator:
+    """External API Backend: SauerkrautLM-Translator via Gradio Client."""
+    def __init__(self, target_lang: str):
+        self.target_lang = target_lang
+        self.available = False
+        if not HAS_GRADIO_CLIENT:
+            return
+            
+        try:
+            logger.info("NMT | Connecting to SauerkrautLM API...")
+            self.client = Client("VAGOsolutions/SauerkrautLM-Translator-LFM2.5-1.2b-Demo")
+            self.available = True
+            logger.info("✓ NMT | SauerkrautLM API Ready")
+        except Exception as e:
+            logger.error(f"NMT | SauerkrautLM connection failed: {e}")
+
+    def translate_batch(self, texts: List[str]) -> List[str]:
+        if not self.available or not texts:
+            return texts
+            
+        results = []
+        for text in texts:
+            try:
+                # The API returns (Prompt, Raw Response, Translation)
+                res = self.client.predict(
+                    text=text,
+                    target_lang=self.target_lang,
+                    api_name="/translate"
+                )
+                results.append(res[2])
+            except Exception as e:
+                logger.error(f"Sauerkraut API Error: {e}")
+                results.append(text) # Fallback to original
+        return results
+
 class UltimateDocumentTranslator:
     """Document translator with configurable backends and M1 optimization."""
     
@@ -1039,7 +1086,12 @@ class UltimateDocumentTranslator:
             
             elif nmt_backend == "opus":
                 self.opus = OpusMTTranslator(src_lang, tgt_lang)
-            
+                
+            elif nmt_backend == "sauerkraut":
+                self.sauerkraut = SauerkrautTranslator(tgt_lang)
+                if not self.sauerkraut.available:
+                    logger.error("INIT | Sauerkraut API requested but client not available.")
+        
             elif nmt_backend == "madlad":
                 self.madlad = Madlad400Translator(src_lang, tgt_lang, "3b")