add-translate

docs/tranlate.md

@@ -0,0 +1,66 @@
+
+
+curl -X 'POST' \
+  'https://gaganyatri-translate-indic-server-cpu.hf.space/translate?src_lang=eng_Latn&tgt_lang=kan_Knda' \
+  -H 'accept: application/json' \
+  -H 'Content-Type: application/json' \
+  -d '{
+  "sentences": [
+    "Hello, how are you?", "Good morning!"
+  ],
+  "src_lang": "eng_Latn",
+  "tgt_lang": "kan_Knda"
+}'
+
+
+{
+  "translations": [
+    "ಹಲೋ, ಹೇಗಿದ್ದೀರಿ? ",
+    "ಶುಭೋದಯ! "
+  ]
+}
+
+
+
+
+curl -X 'POST' \
+  'https://gaganyatri-translate-indic-server-cpu.hf.space/translate?src_lang=kan_Knda&tgt_lang=eng_Latn' \
+  -H 'accept: application/json' \
+  -H 'Content-Type: application/json' \
+  -d '{
+  "sentences": [
+    "ನಮಸ್ಕಾರ, ಹೇಗಿದ್ದೀರಾ?", "ಶುಭೋದಯ!"
+  ],
+  "src_lang": "kan_Knda",
+  "tgt_lang": "eng_Latn"
+}'
+
+
+{
+  "translations": [
+    "Hello, how are you?",
+    "Good morning!"
+  ]
+}
+
+
+
+curl -X 'POST' \
+  'https://gaganyatri-translate-indic-server-cpu.hf.space/translate?src_lang=kan_Knda&tgt_lang=hin_Deva' \
+  -H 'accept: application/json' \
+  -H 'Content-Type: application/json' \
+  -d '{
+  "sentences": [
+    "ನಮಸ್ಕಾರ, ಹೇಗಿದ್ದೀರಾ?", "ಶುಭೋದಯ!"
+  ],
+  "src_lang": "kan_Knda",
+  "tgt_lang": "hin_Deva"
+}'
+
+
+{
+  "translations": [
+    "हैलो, कैसा लग रहा है? ",
+    "गुड मॉर्निंग! "
+  ]
+}

src/server/main.py

@@ -307,6 +307,66 @@ async def chat_v2(
     except Exception as e:
         logger.error(f"Chat_v2 processing failed: {str(e)}", exc_info=True)
         raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
+    
+class TranslationRequest(BaseModel):
+    sentences: list[str]
+    src_lang: str
+    tgt_lang: str
+
+class TranslationResponse(BaseModel):
+    translations: list[str]
+
+@app.post("/v1/translate", response_model=TranslationResponse)
+async def translate(request: TranslationRequest):
+    logger.info(f"Received translation request: {request.dict()}")
+    
+    # External API endpoint
+    external_url = f"https://gaganyatri-translate-indic-server-cpu.hf.space/translate?src_lang={request.src_lang}&tgt_lang={request.tgt_lang}"
+    
+    # Prepare the payload matching the external API's expected format
+    payload = {
+        "sentences": request.sentences,
+        "src_lang": request.src_lang,
+        "tgt_lang": request.tgt_lang
+    }
+    
+    try:
+        # Make the POST request to the external API
+        response = requests.post(
+            external_url,
+            json=payload,
+            headers={
+                "accept": "application/json",
+                "Content-Type": "application/json"
+            },
+            timeout=10  # Set a timeout to avoid hanging
+        )
+        
+        # Raise an exception for bad status codes (4xx, 5xx)
+        response.raise_for_status()
+        
+        # Extract translations from the response
+        response_data = response.json()
+        translations = response_data.get("translations", [])
+        
+        if not translations or len(translations) != len(request.sentences):
+            logger.warning(f"Unexpected response format: {response_data}")
+            raise HTTPException(status_code=500, detail="Invalid response from translation service")
+        
+        logger.info(f"Translation successful: {translations}")
+        return TranslationResponse(translations=translations)
+    
+    except requests.Timeout:
+        logger.error("Translation request timed out")
+        raise HTTPException(status_code=504, detail="Translation service timeout")
+    except requests.RequestException as e:
+        logger.error(f"Error during translation: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Translation failed: {str(e)}")
+    except ValueError as e:
+        logger.error(f"Invalid JSON response: {str(e)}")
+        raise HTTPException(status_code=500, detail="Invalid response format from translation service")
+
+
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Run the FastAPI server.")