Update app.py

app.py

@@ -8,15 +8,31 @@ from datetime import datetime, timedelta
 import threading
 from queue import Queue
 import logging
-from typing import Dict, List, Tuple
-from fastapi import FastAPI, HTTPException
+from typing import Dict, List, Tuple, Optional
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
 import uvicorn
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Pydantic models for request/response
+class TranslationRequest(BaseModel):
+    text: str
+    source_lang: str
+    target_lang: str
+    api_key: Optional[str] = None
+
+class TranslationResponse(BaseModel):
+    translation: str
+    source_language: str
+    target_language: str
+    processing_time: float
+    character_count: int
+    status: str
+
 class TranslationCache:
     def __init__(self, cache_duration_minutes: int = 60):
         self.cache = {}
@@ -236,8 +252,52 @@ async def root():
     return {"message": "Multilingual Translation API", "status": "active"}
 
 @app.post("/api/translate")
-async def api_translate(text: str, source_lang: str, target_lang: str, api_key: str = None):
+async def api_translate(request: TranslationRequest):
     """API endpoint for translation"""
+    if not request.text.strip():
+        raise HTTPException(status_code=400, detail="No text provided")
+    
+    source_code = LANGUAGE_MAP.get(request.source_lang)
+    target_code = LANGUAGE_MAP.get(request.target_lang)
+    
+    if not source_code or not target_code:
+        raise HTTPException(status_code=400, detail="Invalid language codes")
+    
+    try:
+        translation, processing_time = translator.translate_text(request.text, source_code, target_code)
+        
+        return TranslationResponse(
+            translation=translation,
+            source_language=request.source_lang,
+            target_language=request.target_lang,
+            processing_time=processing_time,
+            character_count=len(request.text),
+            status="success"
+        )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Translation error: {str(e)}")
+
+# Alternative endpoint for form data (compatibility with WordPress)
+@app.post("/api/translate/form")
+async def api_translate_form(request: Request):
+    """Alternative endpoint that accepts form data"""
+    try:
+        form_data = await request.form()
+        text = form_data.get("text", "")
+        source_lang = form_data.get("source_lang", "")
+        target_lang = form_data.get("target_lang", "")
+        api_key = form_data.get("api_key", None)
+    except:
+        try:
+            # Try to get JSON data if form data fails
+            json_data = await request.json()
+            text = json_data.get("text", "")
+            source_lang = json_data.get("source_lang", "")
+            target_lang = json_data.get("target_lang", "")
+            api_key = json_data.get("api_key", None)
+        except:
+            raise HTTPException(status_code=400, detail="Invalid request format")
+    
     if not text.strip():
         raise HTTPException(status_code=400, detail="No text provided")