Update app.py

app.py

@@ -281,13 +281,20 @@ def read_srt_file(file_path):
     except Exception as e:
         print(f"An error occurred: {e}")
 
+def clean_text(text):
+    # Remove 'srt ' from the start of each line
+    # Remove ''' from the start and end
+    text = re.sub(r"^```|```$", '', text)
+    text = re.sub(r'^srt', '', text, flags=re.MULTILINE)
+    return text
+
 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}"
+        "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}. in response dont add anything and only return the translation will keeping the srt format"
     },
     {
         "role": "assistant",
@@ -305,17 +312,26 @@ def enhance_text(api_key, text, google):
     }
 
     # 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["result"][0])
-        return result["result"][0]
-    else:
-        print(f"Error: {response.status_code}, {response.text}")
-        return None
+    attempts = 0
+    max_attempts = 3
+
+    while attempts < max_attempts:
+        response = requests.post(url, headers=headers, json=request_body)
+        if response.status_code == 200:
+            result = response.json()
+            if result["status"] == 200:
+                print("status: ", result["status"])
+                te = clean_text(result["result"][0])
+                print("result: ", te)
+                return te
+            else:
+                print(f"Error: status {result['status']}, retrying in 30 seconds...")
+        else:
+            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/"
@@ -337,11 +353,11 @@ def write_google(google_translate):
 
 def time_to_seconds(time_obj):
     return time_obj.hours * 3600 + time_obj.minutes * 60 + time_obj.seconds + time_obj.milliseconds / 1000
-
+    
 def create_subtitle_clips(subtitles, videosize, fontsize, font, color, debug):
     subtitle_clips = []
     for subtitle in subtitles:
-        start_time = time_to_seconds(subtitle.start)
+        start_time = time_to_seconds(subtitle.start) # Add 2 seconds offset
         end_time = time_to_seconds(subtitle.end)
         duration = end_time - start_time
         video_width, video_height = videosize
@@ -349,18 +365,11 @@ def create_subtitle_clips(subtitles, videosize, fontsize, font, color, debug):
         max_height = video_height * 0.2
         #reshaped_text = arabic_reshaper.reshape(subtitle.text)
         #bidi_text = get_display(reshaped_text)
-        text_clip = TextClip(font, subtitle.text, font_size=fontsize, size=(int(video_width * 0.68), int(video_height * 0.2)) ,text_align="center" ,
-                             color=color, bg_color='black', method='caption').with_start(start_time).with_duration(duration)
-        color_clip = ColorClip(size=(int(max_width), int(max_height)),
-                       color=(0, 255, 255)).with_start(start_time).with_duration(duration)
-        color_clip = color_clip.with_opacity(.6)
+        text_clip = TextClip(font, subtitle.text, font_size=fontsize, size=(int(video_width * 0.8), int(video_height * 0.2)) ,text_align="center" ,color=color, method='caption').with_start(start_time).with_duration(duration)
         subtitle_x_position = 'center'
-        subtitle_y_position = video_height * 0.74
+        subtitle_y_position = video_height * 0.68
         text_position = (subtitle_x_position, subtitle_y_position)
-        clip_to_overlay = CompositeVideoClip([color_clip.with_position(text_position), text_clip.with_position(text_position)])
-        clip_to_overlay = clip_to_overlay.with_position('center')
-        subtitle_clips.append(clip_to_overlay)
-    
+        subtitle_clips.append(text_clip.with_position(text_position))
     return subtitle_clips