Upload 4 files

main.py

@@ -4,7 +4,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
 import io
 import json
-from utils import get_text
+from utils import get_text, translate_text
 from json_flatten import flatten
 
 app = FastAPI(
@@ -107,3 +107,27 @@ async def list_models():
     ]
 
     return JSONResponse(content={"models": models})
+
+
+@app.get("/translate")
+async def translate_text_endpoint(text: str, target_language: str):
+    """
+    ### Endpoint Description:
+    Translate text to a specified language.
+
+    #### Request Parameters:
+    - `text`: The text to translate. (Required)
+    - `target_language`: The language to translate the text to. (Required)
+
+    #### Response:
+    - The translated text in the specified language.
+    """
+
+    try:
+        # print(target_language)
+        response = translate_text(text, target_language)
+        # print(response) #{"translated_text":"नमस्ते","translated_language":"नेपाली"}
+        response = json.loads(response)
+        return JSONResponse(content=response)
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=400)

models.py

@@ -5,3 +5,7 @@ from typing import Optional, List
 class FormDetails(BaseModel):
     fields: List[str]
     values: List[str]
+
+class TranslateDetails(BaseModel):
+    translated_text: str
+    translated_language: str

prompts.py

@@ -25,3 +25,17 @@ prompt = """Please extract the [{}] details from this image, and then output int
 
 # # Print or use the resulting system_prompt
 # print(system_prompt)
+
+translate_system_prompt = """
+        You are a translation tool that translates the following text into [{}] language.
+        1. Please translate the text into the language specified.
+        2. The text may contain technical terms, proper nouns, and other specialized vocabulary.
+        3. The text may contain multiple sentences.
+        4. The text may contain spelling errors.
+        5. The text may contain grammatical errors.
+        6. The text may contain punctuation errors.
+        7. The text may contain formatting errors.
+        8. The output should be like [{}] formate json.
+    """
+
+translation_prompt = """Please translate the following into [{}] language :- \n\n [{}] ."""

utils.py

@@ -2,8 +2,8 @@ from dotenv import load_dotenv
 import os
 # import openai
 from openai import OpenAI
-from models import FormDetails
-from prompts import system_prompt_template, prompt
+from models import FormDetails, TranslateDetails
+from prompts import system_prompt_template, prompt, translate_system_prompt, translation_prompt
 import base64
 from io import BytesIO
 import anthropic
@@ -145,4 +145,28 @@ def get_text(image, filename, model, fields="ALL"):
                     )
             response = response.choices[0].message.content
     # print(response)
+    return response
+
+def translate_text(text, target_language):
+    # print("translating")
+    client = OpenAI(api_key = OPENAI_API_KEY)
+    # print("hi")
+    response = client.beta.chat.completions.parse(
+        model="gpt-4o",
+        messages=[
+                {
+                    "role":"system",
+                    "content":translate_system_prompt.format(target_language,TranslateDetails.schema_json())
+                },
+                {
+                "role": "user",
+                "content": translation_prompt.format(target_language,text)
+                },
+                ],
+
+        response_format=TranslateDetails,
+        temperature=0.0,
+    )
+    # print(response.choices[0].message.content)
+    response = response.choices[0].message.content
     return response