Update app.py

app.py

@@ -1,13 +1,18 @@
 import os
+import json
+import requests
 import gradio as gr
 import whisper
 import torch
-from transformers import pipeline
 
 # تهيئة النماذج
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 whisper_model = whisper.load_model("base")
 
+# مفتاح API لـ Gemini
+GEMINI_API_KEY = "AIzaSyDrHCW4FxrDt6amCTQvYPTdh2NE06p9YlQ"
+GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent"
+
 # قاموس للغات المدعومة
 SUPPORTED_LANGUAGES = {
     "ar": "العربية",
@@ -25,15 +30,40 @@ def transcribe_audio(audio_file, source_lang):
         return f"خطأ في التحويل: {str(e)}"
 
 def translate_text(text, source_lang, target_lang):
-    """ترجمة النص بين اللغات"""
+    """ترجمة النص باستخدام Gemini API"""
     if source_lang == target_lang:
         return text
     
     try:
-        translator = pipeline("translation", 
-                            model=f"Helsinki-NLP/opus-mt-{source_lang}-{target_lang}")
-        result = translator(text)[0]['translation_text']
-        return result
+        # تحضير الطلب للترجمة
+        prompt = f"Translate the following text from {SUPPORTED_LANGUAGES[source_lang]} to {SUPPORTED_LANGUAGES[target_lang]}. Only provide the translation without any additional text or explanation:\n\n{text}"
+        
+        payload = {
+            "contents": [{
+                "parts": [{
+                    "text": prompt
+                }]
+            }]
+        }
+        
+        # إضافة مفتاح API كمعامل URL
+        url = f"{GEMINI_API_URL}?key={GEMINI_API_KEY}"
+        
+        # إرسال الطلب
+        response = requests.post(
+            url,
+            headers={"Content-Type": "application/json"},
+            json=payload
+        )
+        
+        if response.status_code == 200:
+            result = response.json()
+            # استخراج النص المترجم من الاستجابة
+            translated_text = result['candidates'][0]['content']['parts'][0]['text']
+            return translated_text
+        else:
+            return f"خطأ في الترجمة: {response.status_code} - {response.text}"
+            
     except Exception as e:
         return f"خطأ في الترجمة: {str(e)}"