Update app.py

app.py

@@ -26,42 +26,48 @@ def get_google_token():
     return access_token
 
 
-def dubpro_english_to_hindi(text):
-    API_URL = os.environ.get("GEMINI_FINETUNED_ENG_HINDI_API")
-    BEARER_TOKEN = get_google_token()
+def clean(result):
+    text = result["choices"][0]['message']["content"]
+    text = re.sub(r"\(.*?\)|\[.*?\]","", text)
+    text = text.strip("'").replace('"', "")
+    if "\n" in text.strip("\n"):
+        text = text.split("\n")[-1]
+    return text
+
+
+def dubpro_hindi_to_tamil(text):
+    API_URL = "https://api.openai.com/v1/chat/completions"
+    prompt = f"""Please convert the following Hindi text into Tamil,  where it is important to maintain context, so please translate it in a colloquial manner, like that of a YouTube video. Just print the translation.
+___
+{text}
+___
+"""
+
+    messages = [
+        {"role": "system", "content": f"You are a language translation assistant."},
+        {"role": "user", "content": prompt}
+    ]
+    
     headers = {
-        "Authorization": f"Bearer {BEARER_TOKEN}",
         "Content-Type": "application/json",
+        "Authorization": f"Bearer {os.environ.get('OPENAI_API_KEY')}"
     }
+    
     payload = {
-        "contents": [
-            {
-                "parts": [{"text": f"text: {text}"}],
-                "role": "user",
-            }
-        ],
-        "generationConfig": {
-            "maxOutputTokens": 8192,
-            "temperature": 0.85,
-        },
-        "safetySettings": [
-            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
-            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
-            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
-            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
-        ],
+        "model": "gpt-4",
+        "messages": messages
     }
+    
     result = requests.post(
         url=API_URL,
         headers=headers,
         json=payload
     )
     response = result.json()
-    response_content = response['candidates'][0]['content']['parts'][0]['text'].replace("translated:", "").strip()
-    return response_content
+    return clean(response_content)
 
 
-def gemini_english_to_hindi(text):
+def gemini_hindi_to_tamil(text):
     API_URL = os.environ.get("GEMINI_FINETUNED_ENG_HINDI_API")
     BEARER_TOKEN = get_google_token()
     headers = {
@@ -71,7 +77,7 @@ def gemini_english_to_hindi(text):
     payload = {
         "contents": [
             {
-                "parts": [{"text": f"Translate the following text to Hindi: `{text}` Output: "}],
+                "parts": [{"text": f"Translate the following text to Tamil: `{text}` Output: "}],
                 "role": "user",
             }
         ],
@@ -105,8 +111,9 @@ def clean(result):
         text = text.split("\n")[-1]
     return text
 
-def openai_english_to_hindi(text, model):
-    prompt = f"Translate the following English text into Hindi such that the meaning in unchanged. Return only the translated text: `{text}`. Output: "
+def openai_hindi_to_tamil(text, model):
+    prompt = 
+    f"Translate the following Hindi text into Tamil such that the meaning in unchanged. Return only the translated text: `{text}`. Output: "
 
     headers = {
         "Content-Type": "application/json",
@@ -134,7 +141,7 @@ def openai_english_to_hindi(text, model):
 
 
 # Azure translate
-def azure_english_to_hindi(text):
+def azure_hindi_to_tamil(text):
     headers = {
         "Ocp-Apim-Subscription-Key": os.environ.get("AZURE_TRANSLATE_KEY"),
         "Ocp-Apim-Subscription-Region": os.environ.get("AZURE_TRANSLATE_REGION"),
@@ -144,8 +151,8 @@ def azure_english_to_hindi(text):
     ENDPOINT = "https://api.cognitive.microsofttranslator.com/translate"
     params = {
         "api-version": "3.0",
-        "from": "en-US",
-        "to": "hi-IN",
+        "from": "hi-IN",
+        "to": "ta-IN",
     }
     texts = [{"text": text}]
     request = requests.post(ENDPOINT, headers=headers, params=params, json=texts)
@@ -154,7 +161,7 @@ def azure_english_to_hindi(text):
 
 
 # Anthopic Claude 3 Haiku
-def claude_english_to_hindi(text):
+def claude_hindi_to_tamil(text):
     client = anthropic.Anthropic()
     message = client.messages.create(
         model="claude-3-haiku-20240307",
@@ -167,7 +174,7 @@ def claude_english_to_hindi(text):
                 "content": [
                     {
                         "type": "text",
-                        "text": f"Translate the following English text into Hindi such that the meaning in unchanged. Return only the translated text: `{text}`. Output: "
+                        "text": f"Translate the following Hindi text into Tamil such that the meaning in unchanged. Return only the translated text: `{text}`. Output: "
                     }
                 ]
             }