Update app.py

app.py

@@ -326,7 +326,20 @@ def split_srt_file(input_file, max_chars=3000):
         output_files.append(output_file_name)
 
     return output_files
-    
+
+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}
+    headers = {"one-api-token": api_key, "Content-Type": "application/json"}
+    response = requests.post(url, headers=headers, json=request_body)
+    if response.status_code == 200:
+        result = response.json()
+        enhanced_text = enhance_text(api_key, text, result['result'])
+        return enhanced_text
+    else:
+        print(f"Error: {response.status_code}, {response.text}")
+        return None
+
 def enhance_text(api_key, text):
     url = "https://api.one-api.ir/chatbot/v1/gpt4o/"
 
@@ -361,21 +374,8 @@ def enhance_text(api_key, text):
             print(f"Error: {response.status_code}, {response.text}, retrying in 30 seconds...")
         attempts += 1
         time.sleep(30)
-    
-    return {"Error": "Max attempts reached. Could not retrieve a successful response."}
-
-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}
-    headers = {"one-api-token": api_key, "Content-Type": "application/json"}
-    response = requests.post(url, headers=headers, json=request_body)
-    if response.status_code == 200:
-        result = response.json()
-        enhanced_text = enhance_text(api_key, text, result['result'])
-        return enhanced_text
-    else:
-        print(f"Error: {response.status_code}, {response.text}")
-        return None
+    print("Error Max attempts reached. Could not retrieve a successful response.")
+    return 0
 
 def write_google(google_translate):
     google = "google_translate.srt"
@@ -424,6 +424,8 @@ def process_video(url, type):
         srt_string = read_srt_file(f"{i}")
     #google_translate = translate_text(api_key, source_language, target_language, srt_string)
         google_translate = enhance_text(api_key, srt_string)
+        if google_translate == 0 :
+            google_translate = translate_text(api_key, source_language, target_language, srt_string)
         write_google(google_translate)
         time.sleep(15)
     srt = read_srt_file("google_translate.srt")