Update app.py

app.py

@@ -11,26 +11,26 @@ from telegram.ext import ApplicationBuilder, MessageHandler, filters
 from transformers import pipeline, AutoTokenizer, VitsModel
 from huggingface_hub import login
 
-# ===== تهيئة التوكن وتسجيل الدخول =====
+# ===== تهيئة التوكن =====
 login(token=os.getenv("HF_TOKEN"))
 
-# ===== تهيئة نظام التسجيل =====
+# ===== إعدادات النظام =====
 logging.basicConfig(
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
     level=logging.INFO
 )
 logger = logging.getLogger(__name__)
 
-# ===== تهيئة النماذج مع التعامل مع الأخطاء =====
+# ===== تحميل النماذج =====
 try:
-    # 1. نموذج التعرف على الكلام
+    # 1. نموذج التعرف على الصوت (ASR)
     asr_pipeline = pipeline(
         "automatic-speech-recognition",
         model="jonatasgrosman/wav2vec2-large-xlsr-53-arabic",
         token=os.getenv("HF_TOKEN")
     )
 
-    # 2. نموذج توليف الصوت
+    # 2. نموذج توليف الصوت (TTS)
     tts_tokenizer = AutoTokenizer.from_pretrained(
         "facebook/mms-tts-ara",
         token=os.getenv("HF_TOKEN")
@@ -46,6 +46,7 @@ except Exception as e:
 
 # ===== دوال معالجة الصوت =====
 def enhance_audio(input_path: str, output_path: str) -> bool:
+    """تحسين جودة الملف الصوتي"""
     try:
         audio = AudioSegment.from_wav(input_path)
         audio = audio.low_pass_filter(3000)
@@ -55,10 +56,11 @@ def enhance_audio(input_path: str, output_path: str) -> bool:
         audio.export(output_path, format="wav")
         return True
     except Exception as e:
-        logger.error(f"فشل تحسين الصوت: {str(e)}")
+        logger.error(f"خطأ في تحسين الصوت: {str(e)}")
         return False
 
 async def speech_to_text(audio_path: str) -> str:
+    """تحويل الصوت إلى نص"""
     try:
         audio, sr = librosa.load(audio_path, sr=16000)
         sf.write("temp.wav", audio, sr)
@@ -69,6 +71,7 @@ async def speech_to_text(audio_path: str) -> str:
         return ""
 
 async def generate_response(text: str) -> str:
+    """توليد رد الذكاء الاصطناعي"""
     try:
         chatbot = pipeline(
             "text-generation",
@@ -84,9 +87,10 @@ async def generate_response(text: str) -> str:
         return response[0]['generated_text']
     except Exception as e:
         logger.error(f"فشل توليد الرد: {str(e)}")
-        return "عذرًا، حدث خطأ في الرد."
+        return "حدث خطأ في توليد الرد."
 
 async def text_to_speech(text: str) -> None:
+    """تحويل النص إلى صوت"""
     try:
         inputs = tts_tokenizer(text, return_tensors="pt")
         with torch.no_grad():
@@ -96,26 +100,29 @@ async def text_to_speech(text: str) -> None:
     except Exception as e:
         logger.error(f"فشل تحويل النص إلى صوت: {str(e)}")
 
-# ===== الدالة الرئيسية للمحادثة =====
+# ===== الدالة الرئيسية =====
 async def process_voice(update: Update, context):
     try:
+        # تحميل الصوت
         voice_file = await update.message.voice.get_file()
         await voice_file.download_to_drive("user_voice.ogg")
         
+        # معالجة الصوت
         user_text = await speech_to_text("user_voice.ogg")
         bot_response = await generate_response(user_text)
         await text_to_speech(bot_response)
         
+        # إرسال الرد
         if enhance_audio("bot_response.wav", "bot_response_enhanced.wav"):
             await update.message.reply_voice("bot_response_enhanced.wav")
         else:
             await update.message.reply_voice("bot_response.wav")
             
     except Exception as e:
-        logger.error(f"خطأ رئيسي: {str(e)}")
-        await update.message.reply_text("⚠️ حدث خطأ غير متوقع.")
+        logger.error(f"خطأ غير متوقع: {str(e)}")
+        await update.message.reply_text("⚠️ عذرًا، حدث خطأ في المعالجة.")
 
-# ===== التشغيل الرئيسي مع إصلاح حدث اللوب =====
+# ===== التشغيل الرئيسي =====
 async def main():
     application = ApplicationBuilder().token(os.getenv("TELEGRAM_TOKEN")).build()
     application.add_handler(MessageHandler(filters.VOICE, process_voice))