Update app.py

app.py

@@ -14,18 +14,17 @@ import fitz  # PyMuPDF
 from PIL import Image
 import io
 import requests
-from fastapi import FastAPI, Request, HTTPException
-from fastapi.responses import HTMLResponse, JSONResponse # استيراد JSONResponse
+from fastapi import FastAPI, Request
+from fastapi.responses import HTMLResponse, JSONResponse
 import uvicorn
 import random
-import docx  # لإضافة دعم .docx
+import docx
 
 # ========== إضافة المكتبات الصحيحة لحل مشكلة DNS + Lifespan + Typing ==========
 import httpx
 import dns.asyncresolver
-from httpx import AsyncClient, AsyncHTTPTransport, Request, Response
-import contextlib # <-- إضافة المكتبة الصحيحة
-from typing import Dict, Any # <-- إضافة Dict و Any
+from httpx import AsyncClient, AsyncHTTPTransport
+import contextlib
 # ==========================================================
 
 # ========== تكوين السجلات ==========
@@ -61,23 +60,20 @@ class CustomDNSTransport(AsyncHTTPTransport):
         self.resolver.nameservers = ['1.1.1.1', '8.8.8.8']
         logger.info("🔧 CustomDNSTransport initialized with 1.1.1.1 and 8.8.8.8")
 
-    async def handle_async_request(self, request: Request) -> Response:
+    async def handle_async_request(self, request):
         try:
             host = request.url.host
 
             # التحقق إذا كان Host هو أصلاً IP
             is_ip = True
             try:
-                # تحقق مبسط لـ IPv4
                 parts = host.split('.')
                 if len(parts) != 4 or not all(p.isdigit() and 0 <= int(p) <= 255 for p in parts):
                      is_ip = False
-                # يمكن إضافة تحقق لـ IPv6 إذا لزم الأمر
             except Exception:
                 is_ip = False
 
             if is_ip:
-                # logger.info(f"Host {host} is already an IP. Skipping resolution.") # يمكن إلغاء التعليق عند الحاجة للتتبع
                 pass
             else:
                 logger.info(f"🔧 Resolving host: {host}")
@@ -85,9 +81,7 @@ class CustomDNSTransport(AsyncHTTPTransport):
                 ip = result[0].address
                 logger.info(f"✅ Resolved {host} to {ip}")
 
-                # تعيين SNI (مهم جداً لـ SSL/HTTPS)
                 request.extensions["sni_hostname"] = host
-                # تحديث الـ URL لاستخدام الـ IP
                 request.url = request.url.copy_with(host=ip)
 
         except dns.resolver.NoAnswer:
@@ -96,10 +90,8 @@ class CustomDNSTransport(AsyncHTTPTransport):
             logger.error(f"❌ DNS NXDOMAIN for {host}. Trying request with original host...")
         except Exception as e:
             logger.error(f"❌ DNS resolution failed for {host}: {e}. Trying request with original host...")
-            # إذا فشل الحل لأي سبب، اسمح للطلب بالاستمرار مع اسم المضيف الأصلي
             pass
 
-        # استدعاء المعالج الأصلي (الأب) لإكمال الطلب
         return await super().handle_async_request(request)
 # ==========================================================
 
@@ -124,7 +116,7 @@ class MedicalLabBot:
         self.file_cache = {}
         self.application = None
         self.is_initialized = False
-        self.initialization_status = "pending" # تتبع حالة التهيئة
+        self.initialization_status = "pending"
         self.load_all_materials()
 
     async def initialize_application(self):
@@ -135,16 +127,13 @@ class MedicalLabBot:
 
             logger.info("🔄 جاري تهيئة تطبيق التليجرام...")
 
-            # ========== الحل الجذري (DNS) ==========
-            logger.info("🔧 إعداد عميل HTTP مخصص مع CustomDNSTransport...")
             custom_transport = CustomDNSTransport()
             custom_client = httpx.AsyncClient(transport=custom_transport)
-            # ============================================
 
             self.application = (
                 Application.builder()
                 .token(TELEGRAM_BOT_TOKEN)
-                .http_client(custom_client)  # <-- تمرير العميل المخصص هنا
+                .http_client(custom_client)
                 .build()
             )
             await self.setup_handlers()
@@ -155,11 +144,9 @@ class MedicalLabBot:
             for attempt in range(max_retries):
                 try:
                     logger.info(f"🚀 محاولة تهيئة الاتصال بـ Telegram (محاولة {attempt + 1}/{max_retries})...")
-                    # 1. الاتصال بـ Telegram (getMe)
                     await self.application.initialize()
                     logger.info("✅ تم تهيئة الاتصال بـ Telegram بنجاح.")
 
-                    # 2. ضبط الـ Webhook (فقط بعد نجاح الاتصال)
                     if SPACE_URL:
                         webhook_url = f"{SPACE_URL.rstrip('/')}/telegram"
                         logger.info(f"���️ جاري إعداد الويب هوك على: {webhook_url}")
@@ -172,11 +159,10 @@ class MedicalLabBot:
                     else:
                         logger.warning("⚠️ SPACE_URL not set. Webhook cannot be set.")
 
-                    # 3. ضبط الحالة على "ناجح"
                     self.is_initialized = True
                     self.initialization_status = "success"
                     logger.info("✅✅✅ التطبيق جاهز لاستقبال الطلبات.")
-                    return True # نجح، اخرج من الدالة
+                    return True
 
                 except Exception as e:
                     logger.warning(f"⚠️ فشلت محاولة التهيئة {attempt + 1}: {e}")
@@ -184,22 +170,20 @@ class MedicalLabBot:
                         logger.info(f"⏳ الانتظار {retry_delay} ثواني قبل إعادة المحاولة...")
                         await asyncio.sleep(retry_delay)
                     else:
-                        logger.error(f"❌ فشل تهيئة التطبيق نهائياً بعد {max_retries} محاولات (حتى مع DNS المخصص).")
+                        logger.error(f"❌ فشل تهيئة التطبيق نهائياً بعد {max_retries} محاولات.")
 
-            # فشل بعد كل المحاولات
             self.is_initialized = False
             self.initialization_status = "failed"
             return False
 
         except Exception as e:
-            logger.error(f"❌ خطأ فادح في تهيئة التطبيق: {e}", exc_info=True) # إضافة exc_info=True
+            logger.error(f"❌ خطأ فادح في تهيئة التطبيق: {e}", exc_info=True)
             self.is_initialized = False
             self.initialization_status = "failed"
             return False
 
     async def setup_handlers(self):
         """إعداد معالجات التليجرام"""
-        # إعداد Conversation Handler مع إعدادات محسنة
         conv_handler = ConversationHandler(
             entry_points=[CommandHandler('start', self.start)],
             states={
@@ -208,7 +192,6 @@ class MedicalLabBot:
                 ],
                 SELECTING_ACTION: [
                     CallbackQueryHandler(self.handle_action_selection, pattern='^(explain_lecture|browse_files|generate_questions|summarize_content|explain_concept|main_menu)$'),
-                    # إضافة معالج العودة هنا أيضاً
                     CallbackQueryHandler(self.handle_back_actions, pattern='^back_to_actions$')
                 ],
                 WAITING_FOR_QUESTION: [
@@ -225,10 +208,7 @@ class MedicalLabBot:
             per_message=False
         )
 
-        # إضافة الـ handlers
         self.application.add_handler(conv_handler)
-
-        # إضافة handlers منفصلة للاستعلامات العامة
         self.application.add_handler(CallbackQueryHandler(self.handle_more_questions, pattern='^more_questions$'))
         self.application.add_handler(CallbackQueryHandler(self.handle_change_subject, pattern='^change_subject$'))
 
@@ -261,7 +241,6 @@ class MedicalLabBot:
                         materials[subject]['file_details'][file_name] = file_info
 
                     else:
-                        # ملفات في الجذر الرئيسي للمستودع (إن وجدت)
                         if 'general' not in materials:
                             materials['general'] = {
                                 'files': [],
@@ -275,19 +254,15 @@ class MedicalLabBot:
                     logger.error(f"خطأ في معالجة الملف {file_path}: {e}")
                     continue
 
-            # فرز الملفات داخل كل مادة (اختياري، للترتيب)
             for subject in materials:
                 materials[subject]['files'].sort(key=lambda x: (x['lecture_number'] if x['lecture_number'] is not None else float('inf'), x['name']))
 
-
             self.available_materials = materials
             logger.info(f"✅ تم تحميل {len(materials)} مادة بنجاح")
 
         except Exception as e:
             logger.error(f"❌ خطأ في تحميل المواد: {e}")
-            self.available_materials = {'Biochemistry': {'files': [], 'file_details': {}}} # مادة افتراضية عند الفشل
-
-    # ========== (الدوال التي تم إكمالها) ==========
+            self.available_materials = {'Biochemistry': {'files': [], 'file_details': {}}}
 
     def get_user_memory(self, user_id):
         """الحصول على ذاكرة المستخدم أو إنشاؤها"""
@@ -323,20 +298,15 @@ class MedicalLabBot:
         lecture_num = None
         file_type = 'unknown'
 
-        # استخراج رقم المحاضرة (محاولة أكثر مرونة)
-        # يبحث عن "lec" أو "lecture" أو "محاضرة" متبوعة برقم
         match = re.search(r'(?:lecture|lec|محاضرة)\s*(\d+)', name_lower)
         if not match:
-            # إذا لم يجد الصيغة السابقة، يبحث عن رقم لوحده في بداية أو نهاية الاسم
             match = re.search(r'^(\d+)\s*-|[\s_-](\d+)$', name_lower)
             if match:
-                # يأخذ الرقم الثاني إذا وجد (المجموعة الثانية)، وإلا الأول
                 lecture_num_str = match.group(2) or match.group(1)
                 lecture_num = int(lecture_num_str) if lecture_num_str else None
         else:
             lecture_num = int(match.group(1))
 
-        # تحديد نوع الملف
         if 'lab' in name_lower or 'عملي' in name_lower:
             file_type = 'lab'
         elif 'exam' in name_lower or 'امتحان' in name_lower or 'اسئلة' in name_lower:
@@ -345,7 +315,6 @@ class MedicalLabBot:
             file_type = 'summary'
         elif 'lecture' in name_lower or 'محاضرة' in name_lower:
             file_type = 'lecture'
-        # إذا لم يكن أي مما سبق وكان هناك رقم محاضرة، افترضه محاضرة
         elif lecture_num is not None:
             file_type = 'lecture'
 
@@ -359,10 +328,9 @@ class MedicalLabBot:
     async def _call_nvidia_api(self, messages, max_tokens=1500):
         """دالة مساعدة لاستدعاء NVIDIA API"""
         try:
-            # استخدام asyncio.to_thread لتشغيل الكود المتزامن في thread
             completion = await asyncio.to_thread(
                 nvidia_client.chat.completions.create,
-                model="meta/llama3-70b-instruct", # استخدام موديل قوي
+                model="meta/llama3-70b-instruct",
                 messages=messages,
                 temperature=0.5,
                 top_p=1,
@@ -378,19 +346,16 @@ class MedicalLabBot:
         files = self.available_materials[subject]['files']
         query_lower = query.lower().strip()
 
-        # 1. محاولة البحث بالرقم التسلسلي في القائمة (مثل "1", "2")
         try:
-            # استخراج الرقم فقط
-            match_num = re.findall(r'^\d+$', query_lower) # يبحث عن رقم فقط في الاستعلام
+            match_num = re.findall(r'^\d+$', query_lower)
             if match_num:
                 index = int(match_num[0])
                 if 1 <= index <= len(files):
                     logger.info(f"Found file by index {index}")
                     return files[index - 1]
         except (IndexError, ValueError):
-            pass # ليس استعلام رقمي
+            pass
 
-        # 2. محاولة البحث برقم المحاضرة الدقيق (مثل "محاضرة 3", "lec 5")
         match_lec_num = re.search(r'(?:lecture|lec|محاضرة)\s*(\d+)', query_lower)
         if match_lec_num:
             num = int(match_lec_num.group(1))
@@ -399,21 +364,18 @@ class MedicalLabBot:
                     logger.info(f"Found file by exact lecture number {num}")
                     return file_info
 
-        # 3. محاولة البحث بجزء من الاسم
         best_match = None
         highest_score = 0
         for file_info in files:
             name_lower = file_info['name'].lower()
-            # حساب درجة التشابه (بسيط) - عدد الكلمات المشتركة
             query_words = set(query_lower.split())
-            name_words = set(re.findall(r'\w+', name_lower)) # استخراج الكلمات من اسم الملف
+            name_words = set(re.findall(r'\w+', name_lower))
             common_words = query_words.intersection(name_words)
             score = len(common_words)
 
             if score > highest_score:
                 highest_score = score
                 best_match = file_info
-            # إذا تطابق الرقم المستخرج من الاستعلام مع رقم المحاضرة، نعتبره تطابق جيد
             elif highest_score > 0 and file_info['lecture_number'] is not None:
                 num_in_query = re.findall(r'\d+', query_lower)
                 if num_in_query and file_info['lecture_number'] == int(num_in_query[0]):
@@ -425,17 +387,15 @@ class MedicalLabBot:
             return best_match
 
         logger.warning(f"Could not find file matching query: '{query}' in subject: {subject}")
-        return None # لم يتم العثور
+        return None
 
     async def download_and_extract_content(self, file_path, subject):
         """تحميل الملف من HF واستخراج النص منه"""
         if file_path in self.file_cache:
-            # logger.info(f"💾 Using cached content for {file_path}") # إلغاء التعليق للتتبع
             return self.file_cache[file_path]
 
         logger.info(f"⏳ Downloading {file_path} from Hugging Face...")
         try:
-            # استخدام asyncio.to_thread لتشغيل الدالة المتزامنة في thread منفصل
             local_path = await asyncio.to_thread(
                 hf_hub_download,
                 repo_id=REPO_ID,
@@ -447,18 +407,16 @@ class MedicalLabBot:
             logger.info(f"📄 Extracting content from {local_path}")
 
             if local_path.lower().endswith('.pdf'):
-                # استخدام PyMuPDF (fitz) لاستخراج أدق
                 with fitz.open(local_path) as doc:
                     for page_num, page in enumerate(doc):
-                        text_content += page.get_text("text", sort=True) # محاولة فرز النص
-                        # text_content += f"\n\n--- Page {page_num + 1} ---\n\n" # إضافة فاصل صفحات (اختياري)
+                        text_content += page.get_text("text", sort=True)
 
             elif local_path.lower().endswith('.txt'):
-                with open(local_path, 'r', encoding='utf-8', errors='ignore') as f: # تجاهل أخطاء الترميز
+                with open(local_path, 'r', encoding='utf-8', errors='ignore') as f:
                     text_content = f.read()
 
             elif local_path.lower().endswith('.docx'):
-                doc_obj = await asyncio.to_thread(docx.Document, local_path) # تشغيل في thread
+                doc_obj = await asyncio.to_thread(docx.Document, local_path)
                 full_text = []
                 for para in doc_obj.paragraphs:
                     full_text.append(para.text)
@@ -468,24 +426,20 @@ class MedicalLabBot:
                 logger.warning(f"Unsupported file type: {local_path}")
                 return f"Error: Unsupported file type ({os.path.basename(file_path)})."
 
-            # تنظيف أساسي ومتقدم
-            text_content = re.sub(r'\s+\n', '\n', text_content) # إزالة المسافات الزائدة قبل سطر جديد
-            text_content = re.sub(r'\n{3,}', '\n\n', text_content) # تقليل الأسطر الفارغة المتتالية
-            text_content = re.sub(r' +', ' ', text_content).strip() # تقليل المسافات المتتالية
+            text_content = re.sub(r'\s+\n', '\n', text_content)
+            text_content = re.sub(r'\n{3,}', '\n\n', text_content)
+            text_content = re.sub(r' +', ' ', text_content).strip()
 
             if len(text_content) < 50:
                 logger.warning(f"File {file_path} content is very short or empty after extraction.")
 
             self.file_cache[file_path] = text_content
-            # logger.debug(f"Content for {file_path}: {text_content[:500]}...") # إلغاء التعليق للتتبع
             return text_content
 
         except Exception as e:
             logger.error(f"❌ Error downloading/extracting {file_path}: {e}", exc_info=True)
             return f"Error: Could not retrieve or process file {os.path.basename(file_path)}."
 
-    # ========== (الدوال الرئيسية) ==========
-
     async def explain_lecture(self, user_query, subject, user_id):
         """شرح محاضرة بناءً على استعلام المستخدم"""
         file_info = self._find_file_by_query(user_query, subject)
@@ -508,8 +462,7 @@ class MedicalLabBot:
         memory = self.get_user_memory(user_id)
         memory['history'].append({"role": "user", "content": f"اشرح لي النقاط الأساسية في هذه المحاضرة: {file_name}"})
 
-        # قص المحتوى ليلائم نافذة السياق (مع مراعاة التوكنات التقريبية)
-        max_content_chars = 7000 # تقليل لترك مساحة للبرومبت والرد
+        max_content_chars = 7000
         if len(content) > max_content_chars:
             content_snippet = content[:max_content_chars] + "\n\n[... المحتوى مقطوع ...]"
             logger.warning(f"Content for {file_name} truncated to {max_content_chars} chars.")
@@ -536,12 +489,11 @@ class MedicalLabBot:
 
         messages = [
             {"role": "system", "content": system_prompt},
-            # إضافة آخر تفاعلين للسياق إذا كانت ذات صلة بالمفهوم
-            *memory['history'][-3:-1], # يأخذ العنصرين قبل الأخير (آخر سؤال وجواب)
+            *memory['history'][-3:-1],
             {"role": "user", "content": f"اشرح لي المفهوم التالي: {user_query}"}
         ]
 
-        response = await self._call_nvidia_api(messages, 1000) # تقليل التوكنات للردود المركزة
+        response = await self._call_nvidia_api(messages, 1000)
         memory['history'].append({"role": "assistant", "content": response})
         return f"🧪 **شرح مفهوم: {user_query}**\n\n{response}"
 
@@ -553,7 +505,6 @@ class MedicalLabBot:
         system_prompt = f"أنت مساعد ذكي متخصص في المختبرات الطبية. المادة الدراسية الحالية التي يركز عليها الطالب هي '{subject}'. أجب على سؤال الطالب ({user_message}) إجابة واضحة ومباشرة. استخدم سياق المحادثة السابق إذا كان ضرورياً لفهم السؤال. إذا كان السؤال خارج نطاق المادة أو التخصص، اعتذر بلطف."
 
         messages = [{"role": "system", "content": system_prompt}]
-        # إضافة آخر 4 تفاعلات للسياق (سؤالين وجوابين)
         messages.extend(history[-4:])
         messages.append({"role": "user", "content": user_message})
 
@@ -568,12 +519,11 @@ class MedicalLabBot:
             return "❌ لا توجد ملفات في هذه المادة لتوليد أسئلة منها."
 
         files = self.available_materials[subject]['files']
-        # تفضيل ملفات المحاضرات أو الملخصات إن وجدت
         preferred_files = [f for f in files if f['type'] in ('lecture', 'summary')]
         if preferred_files:
             file_info = random.choice(preferred_files)
         else:
-            file_info = random.choice(files) # اختر أي ملف إذا لم توجد المفضلة
+            file_info = random.choice(files)
 
         file_path = file_info['path']
         file_name = file_info['name']
@@ -583,7 +533,6 @@ class MedicalLabBot:
 
         if content.startswith("Error:") or not content.strip():
             logger.error(f"Failed to get content for question generation from {file_name}. Content: {content[:100]}")
-            # محاولة اختيار ملف آخر مرة واحدة
             if len(files) > 1:
                 logger.info("Retrying with another file for question generation...")
                 remaining_files = [f for f in files if f['path'] != file_path]
@@ -620,12 +569,9 @@ class MedicalLabBot:
 
         files = self.available_materials[subject]['files']
 
-        # إعطاء أولوية للملفات المسماة "summary" أو "ملخص"
         summary_file = next((f for f in files if f['type'] == 'summary'), None)
-        # إذا لم يوجد، اختر أول ملف محاضرة
         if not summary_file:
             summary_file = next((f for f in files if f['type'] == 'lecture'), None)
-        # إذا لم يوجد محاضرة أيضاً، اختر أول ملف
         if not summary_file:
             summary_file = files[0]
 
@@ -652,8 +598,6 @@ class MedicalLabBot:
         response = await self._call_nvidia_api(messages)
         return f"📋 **ملخص ملف: {file_name}**\n\n{response}"
 
-    # ========== (باقي دوال المعالجة الأصلية) ==========
-
     async def start(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """بدء المحادثة وعرض القائمة الرئيسية"""
         user_id = update.effective_user.id
@@ -670,7 +614,7 @@ class MedicalLabBot:
         if not self.available_materials:
             welcome_text += "\n\n⚠️ عذراً، لم أتمكن من تحميل أي مواد دراسية حالياً. حاول تحديث القائمة بالضغط على (🔄)."
         else:
-            for subject in sorted(self.available_materials.keys()): # فرز المواد أبجدياً
+            for subject in sorted(self.available_materials.keys()):
                 file_count = len(self.available_materials[subject]['files'])
                 welcome_text += f"\n• {subject} ({file_count} ملف)"
 
@@ -679,13 +623,11 @@ class MedicalLabBot:
         keyboard = self.create_subjects_keyboard()
         reply_markup = InlineKeyboardMarkup(keyboard)
 
-        # إذا كانت استجابة لـ callback query (مثل العودة للقائمة), عدّل الرسالة
         if update.callback_query:
             try:
                 await update.callback_query.edit_message_text(welcome_text, reply_markup=reply_markup)
             except Exception as e:
                 logger.error(f"Error editing message in start: {e}")
-                # إذا فشل التعديل (ربما الرسالة قديمة), أرسل رسالة جديدة
                 await update.effective_message.reply_text(welcome_text, reply_markup=reply_markup)
         else:
             await update.message.reply_text(welcome_text, reply_markup=reply_markup)
@@ -695,7 +637,6 @@ class MedicalLabBot:
     def create_subjects_keyboard(self):
         """إنشاء لوحة مفاتيح للمواد المتاحة"""
         keyboard = []
-        # عرض المواد في عمودين إذا كان عددها أكبر من 4
         subjects = sorted(self.available_materials.keys())
         row = []
         max_cols = 2 if len(subjects) > 4 else 1
@@ -724,7 +665,7 @@ class MedicalLabBot:
             return await self.handle_general_help(query, context)
         elif callback_data == "refresh_materials":
             await query.edit_message_text("🔄 جاري تحديث قائمة المواد...")
-            self.load_all_materials() # إعادة تحميل المواد
+            self.load_all_materials()
             keyboard = self.create_subjects_keyboard()
             reply_markup = InlineKeyboardMarkup(keyboard)
             await query.edit_message_text("✅ تم تحديث قائمة المواد.\nاختر المادة:", reply_markup=reply_markup)
@@ -736,11 +677,10 @@ class MedicalLabBot:
         memory = self.get_user_memory(user_id)
         memory['last_subject'] = subject
 
-        # التأكد من أن المادة موجودة (احتياطي)
         if subject not in self.available_materials:
             logger.error(f"Selected subject '{subject}' not found in available materials.")
             await query.edit_message_text("❌ خطأ: المادة المحددة غير موجودة. ربما تحتاج لتحديث القائمة؟")
-            return await self.start(update, context) # العودة للقائمة الرئيسية
+            return await self.start(update, context)
 
         subject_files = self.available_materials[subject]['files']
         subject_name = subject.replace('_', ' ').title()
@@ -771,7 +711,7 @@ class MedicalLabBot:
         subject = context.user_data.get('current_subject')
         if not subject or subject not in self.available_materials:
             logger.warning("Subject context lost or invalid in handle_back_actions. Returning to main menu.")
-            return await self.start(update, context) # إذا ضاع السياق، عد للبداية
+            return await self.start(update, context)
 
         subject_files = self.available_materials[subject]['files']
         subject_name = subject.replace('_', ' ').title()
@@ -797,7 +737,7 @@ class MedicalLabBot:
         query = update.callback_query
         await query.answer()
         logger.info(f"User {query.from_user.id} requested main menu.")
-        context.user_data.clear() # تنظيف السياق عند العودة للرئيسية
+        context.user_data.clear()
         return await self.start(update, context)
 
     async def handle_more_questions(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
@@ -814,7 +754,7 @@ class MedicalLabBot:
         reply_markup = InlineKeyboardMarkup(keyboard)
 
         await query.edit_message_text(f"💬 تفضل، اكتب سؤالك أو طلبك الجديد المتعلق بمادة '{subject}':", reply_markup=reply_markup)
-        context.user_data['waiting_for'] = 'general' # ضبط الحالة لانتظار استعلام عام
+        context.user_data['waiting_for'] = 'general'
         return WAITING_FOR_QUESTION
 
     async def handle_change_subject(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
@@ -842,7 +782,6 @@ class MedicalLabBot:
             await query.edit_message_text("❌ حدث خطأ في السياق. يرجى البدء من جديد.")
             return await self.start(update, context)
 
-        # زر العودة للقائمة الرئيسية
         back_button = InlineKeyboardButton("🔙 رجوع", callback_data="back_to_actions")
         keyboard_with_back = [[back_button]]
         reply_markup_back = InlineKeyboardMarkup(keyboard_with_back)
@@ -853,18 +792,16 @@ class MedicalLabBot:
         elif action == "browse_files":
             files_text = await self.get_files_list_text(subject)
             keyboard = [
-                # يمكن إضافة زر لشرح ملف من هنا مباشرة إذا أردت
                 [InlineKeyboardButton("📖 طلب شرح ملف محدد", callback_data="explain_lecture")],
                 [back_button]
             ]
             reply_markup = InlineKeyboardMarkup(keyboard)
-            await query.edit_message_text(files_text, reply_markup=reply_markup, parse_mode='Markdown') # إضافة parse_mode
-            # البقاء في نفس الحالة لعرض الخيارات مرة أخرى
+            await query.edit_message_text(files_text, reply_markup=reply_markup, parse_mode='Markdown')
             return SELECTING_ACTION
 
         elif action == "explain_lecture":
             files_list = await self.get_files_list_text(subject)
-            if files_list.startswith("❌"): # إذا لم توجد ملفات
+            if files_list.startswith("❌"):
                 await query.edit_message_text(files_list, reply_markup=reply_markup_back)
                 return SELECTING_ACTION
 
@@ -873,7 +810,7 @@ class MedicalLabBot:
                 f"{files_list}\n\n"
                 f"📝 اكتب رقم الملف من القائمة أعلاه أو جزءاً من اسمه:",
                 reply_markup=reply_markup_back,
-                parse_mode='Markdown' # إضافة parse_mode
+                parse_mode='Markdown'
             )
             context.user_data['waiting_for'] = 'lecture_explanation'
             return WAITING_FOR_QUESTION
@@ -881,9 +818,8 @@ class MedicalLabBot:
         elif action == "generate_questions":
             await query.edit_message_text("⏳ حسناً، جاري توليد بعض الأسئلة للمراجعة...", reply_markup=reply_markup_back)
             questions = await self.generate_questions_for_subject(subject, user_id)
-            # await query.edit_message_text(questions, reply_markup=reply_markup_back) # لا تعدل الرسالة، أرسل الأسئلة كرد
-            await query.message.reply_text(questions, reply_markup=reply_markup_back, parse_mode='Markdown') # ارسال رد جديد + parse_mode
-            return SELECTING_ACTION # العودة لقائمة الخيارات
+            await query.message.reply_text(questions, reply_markup=reply_markup_back, parse_mode='Markdown')
+            return SELECTING_ACTION
 
         elif action == "explain_concept":
             await query.edit_message_text(
@@ -898,20 +834,13 @@ class MedicalLabBot:
         elif action == "summarize_content":
             await query.edit_message_text("⏳ تمام، جاري تلخيص ملف مهم من المادة...", reply_markup=reply_markup_back)
             summary = await self.generate_summary(subject, user_id)
-            # await query.edit_message_text(summary, reply_markup=reply_markup_back) # لا تعدل، أرسل كرد
-            await query.message.reply_text(summary, reply_markup=reply_markup_back, parse_mode='Markdown') # ارسال رد + parse_mode
-            return SELECTING_ACTION # العودة لقائمة الخيارات
+            await query.message.reply_text(summary, reply_markup=reply_markup_back, parse_mode='Markdown')
+            return SELECTING_ACTION
 
-        # إذا لم يتعرف على الإجراء (احتياطي)
         logger.warning(f"Unknown action selected: {action}")
         await query.message.reply_text("عذراً، لم أتعرف على هذا الخيار.")
         return SELECTING_ACTION
 
-    async def browse_available_files(self, query, context):
-        """عرض الملفات المتاحة للمادة - تم دمجه في handle_action_selection"""
-        # هذه الدالة لم تعد مستخدمة بشكل مباشر، الكود موجود في handle_action_selection
-        pass
-
     async def get_files_list_text(self, subject):
         """إنشاء نص لقائمة الملفات"""
         if subject not in self.available_materials:
@@ -924,7 +853,6 @@ class MedicalLabBot:
             return "❌ لا توجد ملفات متاحة لهذه المادة بعد."
 
         files_text = f"📁 **الملفات المتاحة لمادة {subject}:**\n\n"
-        # عرض عدد محدود من الملفات لتجنب رسالة طويلة جداً
         max_files_to_show = 25
         for i, file_info in enumerate(files[:max_files_to_show], 1):
             file_name = file_info['name']
@@ -940,8 +868,6 @@ class MedicalLabBot:
             }.get(file_type, '📄')
 
             num_text = f" (محاضرة {lecture_num})" if lecture_num else ""
-            # تنسيق السطر: الرقم. ايموجي اسم_الملف (رقم المحاضرة إن وجد) - استخدام Markdown للأسماء
-            # استبدال الشرطة السفلية بمسافة في اسم الملف للعرض فقط
             display_name = file_name.replace("_", " ")
             files_text += f"{i}. {type_emoji} `{display_name}`{num_text}\n"
 
@@ -953,7 +879,7 @@ class MedicalLabBot:
     async def handle_message(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """معالجة الرسائل النصية من المستخدم"""
         if not update.message or not update.message.text:
-            return # تجاهل الرسائل غير النصية أو الفارغة
+            return
 
         user_message = update.message.text
         user_id = update.effective_user.id
@@ -963,26 +889,24 @@ class MedicalLabBot:
 
         await update.message.chat.send_action(action="typing")
 
-        response = "عذراً، لم أفهم طلبك. هل يمكنك توضيحه؟" # رد افتراضي
-        next_state = WAITING_FOR_QUESTION # البقاء في حالة الانتظار افتراضياً
+        response = "عذراً، لم أفهم طلبك. هل يمكنك توضيحه؟"
+        next_state = WAITING_FOR_QUESTION
 
         try:
             if waiting_for == 'lecture_explanation':
                 response = await self.explain_lecture(user_message, subject, user_id)
-                next_state = SELECTING_ACTION # تم الرد، العودة للخيارات
+                next_state = SELECTING_ACTION
             elif waiting_for == 'concept_explanation':
                 response = await self.explain_concept(user_message, subject, user_id)
-                next_state = SELECTING_ACTION # تم الرد، العودة للخيارات
-            else: # يتضمن 'general' أو أي شيء آخر (مثل سؤال عام بعد الضغط على "أسئلة أخرى")
+                next_state = SELECTING_ACTION
+            else:
                 response = await self.process_general_query(user_message, subject, user_id)
-                next_state = SELECTING_ACTION # تم الرد، العودة للخيارات
+                next_state = SELECTING_ACTION
 
-            # إرسال الرد
             await update.message.reply_text(response, parse_mode='Markdown')
 
-            # إذا نجح الرد، أرسل قائمة الخيارات التالية
             if next_state == SELECTING_ACTION:
-                context.user_data['waiting_for'] = None # إنهاء حالة الانتظار
+                context.user_data['waiting_for'] = None
                 keyboard = [
                     [InlineKeyboardButton("🔄 طرح سؤال آخر", callback_data="more_questions")],
                     [InlineKeyboardButton("📚 تغيير المادة", callback_data="change_subject")],
@@ -991,13 +915,13 @@ class MedicalLabBot:
                 reply_markup = InlineKeyboardMarkup(keyboard)
                 await update.message.reply_text("هل تحتاج مساعدة أخرى؟", reply_markup=reply_markup)
 
-            return next_state # العودة للحالة المناسبة (إما خيارات أو انتظار سؤال آخر)
+            return next_state
 
         except Exception as e:
             logger.error(f"❌ Error processing message from user {user_id}: {e}", exc_info=True)
             await update.message.reply_text("❌ حدث خطأ غير متوقع أثناء معالجة طلبك. لقد تم تسجيل الخطأ. يرجى المحاولة مرة أخرى أو اختيار خيار آخر.")
-            context.user_data['waiting_for'] = None # إنهاء الانتظار عند حدوث خطأ فادح
-            return SELECTING_ACTION # العودة لقائمة الخيارات كحل احتياطي آمن
+            context.user_data['waiting_for'] = None
+            return SELECTING_ACTION
 
 # ========== إنشاء كائن البوت ==========
 bot = MedicalLabBot()
@@ -1010,17 +934,16 @@ async def root():
     materials_count = len(bot.available_materials)
     total_files = sum(len(material['files']) for material in bot.available_materials.values())
 
-    # تحديد رسالة الحالة بناءً على حالة التهيئة
     status_message = "⏳ جاري التهيئة (الاتصال بـ Telegram)..."
-    status_color = "#ffc107" # أصفر
+    status_color = "#ffc107"
     init_details = ""
     if bot.initialization_status == "success":
         status_message = "✅ نشط ومهيأ"
-        status_color = "#28a745" # أخضر
+        status_color = "#28a745"
         init_details = "تم الاتصال بـ Telegram وضبط الويب هوك بنجاح."
     elif bot.initialization_status == "failed":
         status_message = "❌ فشل التهيئة"
-        status_color = "#dc3545" # أحمر
+        status_color = "#dc3545"
         init_details = "فشل الاتصال بـ Telegram أو ضبط الويب هوك بعد عدة محاولات."
 
     return f"""
@@ -1072,23 +995,18 @@ async def root():
 async def handle_telegram_update(request: Request):
     """معالجة تحديثات Telegram"""
     try:
-        # التحقق من أن التطبيق مهيأ
         if not bot.is_initialized or not bot.application:
             logger.error("❌ التطبيق غير مهيأ، لا يمكن معالجة التحديث (ربما لا يزال قيد التهيئة).")
-            # إرجاع 503 Service Unavailable لإخبار Telegram بإعادة المحاولة لاحقًا
             return JSONResponse(
                 status_code=503,
                 content={"status": "error", "detail": "Application not initialized or still initializing"}
             )
 
         update_data = await request.json()
-        # logger.debug(f"Received update: {update_data}") # إلغاء التعليق للتتبع المكثف
         update = Update.de_json(update_data, bot.application.bot)
 
-        # معالجة التحديث في مهمة منفصلة لتجنب حظر الحلقة الرئيسية
         asyncio.create_task(bot.application.process_update(update))
 
-        # إرجاع استجابة سريعة لـ Telegram
         return JSONResponse(content={"status": "ok"})
 
     except json.JSONDecodeError:
@@ -1099,7 +1017,6 @@ async def handle_telegram_update(request: Request):
         )
     except Exception as e:
         logger.error(f"❌ Error processing update: {e}", exc_info=True)
-        # إرجاع خطأ عام 500 Internal Server Error
         return JSONResponse(
             status_code=500,
             content={"status": "error", "detail": "Internal server error while processing update"}
@@ -1111,18 +1028,16 @@ async def health_check():
     materials_count = len(bot.available_materials)
     total_files = sum(len(material['files']) for material in bot.available_materials.values())
 
-    status_code = 503 # Service Unavailable افتراضياً
+    status_code = 503
     service_status = "unhealthy"
     if bot.initialization_status == "pending":
         service_status = "initializing"
-        # يمكن إرجاع 200 هنا إذا أردت أن تعتبره "صحي" أثناء التهيئة
-        # status_code = 200
     elif bot.initialization_status == "success":
         service_status = "healthy"
-        status_code = 200 # OK
+        status_code = 200
     elif bot.initialization_status == "failed":
         service_status = "unhealthy_failed_init"
-        status_code = 500 # Internal Server Error
+        status_code = 500
 
     response_payload = {
         "status": service_status,
@@ -1133,19 +1048,17 @@ async def health_check():
         "timestamp": datetime.now().isoformat()
     }
 
-    # استخدام JSONResponse لتحديد status_code بشكل صحيح
     return JSONResponse(content=response_payload, status_code=status_code)
 
 # ========== التشغيل الرئيسي ==========
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
-    # يمكن تحديد مستوى السجل من متغير بيئة أيضاً
     log_level = os.environ.get("LOG_LEVEL", "info").lower()
     logger.info(f"🚀 Starting Medical Lab Bot on port {port} with log level {log_level}")
     uvicorn.run(
-        "app:app", # تغيير لاستخدام صيغة uvicorn القياسية
+        "app:app",
         host="0.0.0.0",
         port=port,
         log_level=log_level,
-        reload=False # تعطيل إعادة التحميل التلقائي في الإنتاج
+        reload=False
     )