Update app.py

app.py

@@ -201,6 +201,42 @@ def read_srt_file(file_path):
     except Exception as e:
         print(f"An error occurred: {e}")
 
+def enhance_text(api_key, text, google):
+    url = "https://api.one-api.ir/chatbot/v1/gpt4o/"
+
+    # Prepare the request body
+    request_body = [{
+        "role": "user",
+        "content": f" i have a main English text that has been translated, use the main text and the translations to write a better translation of the main text in persian. main text: {text}"
+    },
+    {
+        "role": "assistant",
+        "content": "ok"
+    },
+    {
+        "role": "user",
+        "content": f"tarnslation: {google} in response only return the better version of the translation with maintaining the srt format of the text"
+    },]
+
+    # Add the API key to the request
+    headers = {
+        "one-api-token": api_key,
+        "Content-Type": "application/json"
+    }
+
+    # Make the POST request
+    response = requests.post(url, headers=headers, json=request_body)
+
+    # Check the response status
+    if response.status_code == 200:
+        result = response.json()
+        print("result: ")
+        print(result)
+        return result["result"]
+    else:
+        print(f"Error: {response.status_code}, {response.text}")
+        return None
+
 def translate_text(api_key, source_lang, target_lang, text):
     url = "https://api.one-api.ir/translate/v1/google/"
     request_body = {"source": source_lang, "target": target_lang, "text": text}
@@ -208,7 +244,8 @@ def translate_text(api_key, source_lang, target_lang, text):
     response = requests.post(url, headers=headers, json=request_body)
     if response.status_code == 200:
         result = response.json()
-        return result['result']
+        enhanced_text = enhance_text(api_key, text, result['result'])
+        return enhanced_text
     else:
         print(f"Error: {response.status_code}, {response.text}")
         return None