Update app.py

app.py

@@ -1,26 +1,21 @@
 import gradio as gr
-from googletrans import Translator
+import google.generativeai as genai
 import re
 import os
-import time
 
-translator = Translator()
+API_KEY = "YOUR_API_KEY_HERE"
+genai.configure(api_key=API_KEY)
+model = genai.GenerativeModel("gemini-1.5-flash")
 
-def translate_with_googletrans(text):
+def translate_with_gemini(text):
     if not text.strip():
-        print("Text is empty, returning empty string")
         return ""
-    print(f"Starting translation for: '{text}' at {time.ctime()}")
-    start_time = time.time()
     try:
-        translated = translator.translate(text, src='en', dest='fa').text
-        elapsed_time = time.time() - start_time
-        print(f"Translated to: '{translated}' (took {elapsed_time:.2f} seconds)")
-        return translated
+        prompt = f"Translate this to Persian: {text}"
+        response = model.generate_content(prompt)
+        return response.text
     except Exception as e:
-        error_msg = f"خطا در ترجمه: {str(e)} (after {time.time() - start_time:.2f} seconds)"
-        print(error_msg)
-        return error_msg
+        return f"خطا: {str(e)}"
 
 def process_srt(file):
     encodings = ['utf-8', 'windows-1252', 'latin1']
@@ -29,7 +24,6 @@ def process_srt(file):
         try:
             with open(file.name, 'r', encoding=encoding) as f:
                 content = f.read()
-            print(f"File read with encoding: {encoding}")
             break
         except UnicodeDecodeError:
             continue
@@ -38,30 +32,26 @@ def process_srt(file):
         raise ValueError("نمی‌تونم فایل رو با انکودینگ‌های رایج بخونم. لطفاً فایل رو با UTF-8 ذخیره کن.")
 
     blocks = re.split(r'\n\n', content.strip())
-    print(f"Found {len(blocks)} blocks in SRT")
     subtitles = ["خروجی‌ها توسط هوش مصنوعی انجام شده\n"]
-    for i, block in enumerate(blocks):
-        print(f"Processing block {i + 1}/{len(blocks)}")
+    for block in blocks:
         lines = block.split('\n')
         if len(lines) >= 2:
             index = lines[0].strip()
             timing = lines[1].strip()
             text = '\n'.join(lines[2:]).strip()
-            translated_text = translate_with_googletrans(text)
+            translated_text = translate_with_gemini(text)
             subtitles.append(f"{index}\n{timing}\n{translated_text}")
-            time.sleep(0.5)  # تاخیر 0.5 ثانیه بین درخواست‌ها
     output_content = '\n\n'.join(subtitles)
     output_path = "translated_srt.srt"
     with open(output_path, 'w', encoding='utf-8') as f:
         f.write(output_content)
-    print("File written successfully")
     return output_path
 
 interface = gr.Interface(
     fn=process_srt,
     inputs=gr.File(label="فایل SRT انگلیسی رو آپلود کن"),
     outputs=gr.File(label="فایل SRT ترجمه‌شده به فارسی"),
-    title="ترجمه زیرنویس با Google Translate",
+    title="ترجمه زیرنویس با Gemini API",
     description="فایل SRT انگلیسی رو آپلود کن تا به فارسی ترجمه بشه."
 )