Update app.py

app.py

@@ -136,7 +136,7 @@ class MedicalLabBot:
             logger.info("🔄 جاري تهيئة تطبيق التليجرام...")
 
             # ========== الحل الجذري (DNS) ==========
-            logger.info("🔧 إعداد عملي HTTP مخصص مع CustomDNSTransport...")
+            logger.info("🔧 إعداد عميل HTTP مخصص مع CustomDNSTransport...")
             custom_transport = CustomDNSTransport()
             custom_client = httpx.AsyncClient(transport=custom_transport)
             # ============================================
@@ -328,14 +328,13 @@ class MedicalLabBot:
         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
+            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))
-
+            lecture_num = int(match.group(1))
 
         # تحديد نوع الملف
         if 'lab' in name_lower or 'عملي' in name_lower:
@@ -348,8 +347,7 @@ class MedicalLabBot:
             file_type = 'lecture'
         # إذا لم يكن أي مما سبق وكان هناك رقم محاضرة، افترضه محاضرة
         elif lecture_num is not None:
-             file_type = 'lecture'
-
+            file_type = 'lecture'
 
         return {
             'name': file_name,
@@ -395,11 +393,11 @@ class MedicalLabBot:
         # 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))
-             for file_info in files:
-                  if file_info['lecture_number'] == num:
-                        logger.info(f"Found file by exact lecture number {num}")
-                        return file_info
+            num = int(match_lec_num.group(1))
+            for file_info in files:
+                if file_info['lecture_number'] == num:
+                    logger.info(f"Found file by exact lecture number {num}")
+                    return file_info
 
         # 3. محاولة البحث بجزء من الاسم
         best_match = None
@@ -417,16 +415,14 @@ class MedicalLabBot:
                 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]):
-                      logger.info(f"Found file by partial name match with lecture number heuristic: {file_info['name']}")
-                      return file_info
-
+                num_in_query = re.findall(r'\d+', query_lower)
+                if num_in_query and file_info['lecture_number'] == int(num_in_query[0]):
+                    logger.info(f"Found file by partial name match with lecture number heuristic: {file_info['name']}")
+                    return file_info
 
         if best_match and highest_score > 0:
-             logger.info(f"Found file by best partial name match: {best_match['name']} (score: {highest_score})")
-             return best_match
-
+            logger.info(f"Found file by best partial name match: {best_match['name']} (score: {highest_score})")
+            return best_match
 
         logger.warning(f"Could not find file matching query: '{query}' in subject: {subject}")
         return None # لم يتم العثور
@@ -452,21 +448,21 @@ class MedicalLabBot:
 
             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" # إضافة فاصل صفحات (اختياري)
+                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" # إضافة فاصل صفحات (اختياري)
 
             elif local_path.lower().endswith('.txt'):
                 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
-                 full_text = []
-                 for para in doc_obj.paragraphs:
-                     full_text.append(para.text)
-                 text_content = '\n'.join(full_text)
+                doc_obj = await asyncio.to_thread(docx.Document, local_path) # تشغيل في thread
+                full_text = []
+                for para in doc_obj.paragraphs:
+                    full_text.append(para.text)
+                text_content = '\n'.join(full_text)
 
             else:
                 logger.warning(f"Unsupported file type: {local_path}")
@@ -478,7 +474,7 @@ class MedicalLabBot:
             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.")
+                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]}...") # إلغاء التعليق للتتبع
@@ -507,7 +503,7 @@ class MedicalLabBot:
             return f"❌ خطأ في معالجة الملف: {file_name}\n{content}"
 
         if not content.strip():
-             return f"❌ المحتوى فارغ للملف: {file_name}"
+            return f"❌ المحتوى فارغ للملف: {file_name}"
 
         memory = self.get_user_memory(user_id)
         memory['history'].append({"role": "user", "content": f"اشرح لي النقاط الأساسية في هذه المحاضرة: {file_name}"})
@@ -518,8 +514,7 @@ class MedicalLabBot:
             content_snippet = content[:max_content_chars] + "\n\n[... المحتوى مقطوع ...]"
             logger.warning(f"Content for {file_name} truncated to {max_content_chars} chars.")
         else:
-             content_snippet = content
-
+            content_snippet = content
 
         system_prompt = f"أنت مساعد أكاديمي متخصص في مادة {subject}. مهمتك هي شرح النقاط الأساسية في محتوى المحاضرة المقدم لك. ركز على المفاهيم الجوهرية، التعريفات الهامة، النتائج الرئيسية، وأي معلومات ضرورية لفهم الموضوع. قدم الشرح بطريقة منظمة وواضحة باستخدام نقاط Markdown. تجنب التفاصيل الثانوية غير الضرورية."
 
@@ -590,21 +585,20 @@ class MedicalLabBot:
             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]
-                 if remaining_files:
-                     file_info = random.choice(remaining_files)
-                     file_path = file_info['path']
-                     file_name = file_info['name']
-                     content = await self.download_and_extract_content(file_path, subject)
-                     if content.startswith("Error:") or not content.strip():
-                           logger.error(f"Retry failed for question generation from {file_name}.")
-                           return f"❌ لم أتمكن من قراءة محتوى صالح لتوليد أسئلة من ملفات المادة."
-                 else:
-                      return f"❌ لم أتمكن من قراءة محتوى صالح لتوليد أسئلة من ملفات المادة."
+                logger.info("Retrying with another file for question generation...")
+                remaining_files = [f for f in files if f['path'] != file_path]
+                if remaining_files:
+                    file_info = random.choice(remaining_files)
+                    file_path = file_info['path']
+                    file_name = file_info['name']
+                    content = await self.download_and_extract_content(file_path, subject)
+                    if content.startswith("Error:") or not content.strip():
+                        logger.error(f"Retry failed for question generation from {file_name}.")
+                        return f"❌ لم أتمكن من قراءة محتوى صالح لتوليد أسئلة من ملفات المادة."
+                else:
+                    return f"❌ لم أتمكن من قراءة محتوى صالح لتوليد أسئلة من ملفات المادة."
             else:
-                 return f"❌ لم أتمكن من قراءة محتوى صالح لتوليد أسئلة من ملفات المادة."
-
+                return f"❌ لم أتمكن من قراءة محتوى صالح لتوليد أسئلة من ملفات المادة."
 
         max_content_chars = 7000
         content_snippet = content[:max_content_chars] if len(content) > max_content_chars else content
@@ -633,7 +627,7 @@ class MedicalLabBot:
             summary_file = next((f for f in files if f['type'] == 'lecture'), None)
         # إذا لم يوجد محاضرة أيضاً، اختر أول ملف
         if not summary_file:
-             summary_file = files[0]
+            summary_file = files[0]
 
         file_path = summary_file['path']
         file_name = summary_file['name']
@@ -690,11 +684,11 @@ class MedicalLabBot:
             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)
+                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)
+            await update.message.reply_text(welcome_text, reply_markup=reply_markup)
 
         return SELECTING_SUBJECT
 
@@ -713,7 +707,6 @@ class MedicalLabBot:
                 keyboard.append(row)
                 row = []
 
-
         keyboard.append([InlineKeyboardButton("🔄 تحديث قائمة المواد", callback_data="refresh_materials")])
         keyboard.append([InlineKeyboardButton("❓ مساعدة", callback_data="general_help")])
         return keyboard
@@ -777,8 +770,8 @@ 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) # إذا ضاع السياق، عد للبداية
+            logger.warning("Subject context lost or invalid in handle_back_actions. Returning to main menu.")
+            return await self.start(update, context) # إذا ضاع السياق، عد للبداية
 
         subject_files = self.available_materials[subject]['files']
         subject_name = subject.replace('_', ' ').title()
@@ -844,39 +837,36 @@ class MedicalLabBot:
         subject = context.user_data.get('current_subject')
         logger.info(f"User {user_id} selected action: {action} for subject: {subject}")
 
-
         if not subject or subject not in self.available_materials:
-             logger.error("Subject context lost or invalid in handle_action_selection.")
-             await query.edit_message_text("❌ حدث خطأ في السياق. يرجى البدء من جديد.")
-             return await self.start(update, context)
+            logger.error("Subject context lost or invalid in handle_action_selection.")
+            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)
 
-
         if action == "main_menu":
             return await self.start(update, context)
 
         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
-             # البقاء في نفس الحالة لعرض الخيارات مرة أخرى
-             return SELECTING_ACTION
-
+            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
+            # البقاء في نفس الحالة لعرض الخيارات مرة أخرى
+            return SELECTING_ACTION
 
         elif action == "explain_lecture":
             files_list = await self.get_files_list_text(subject)
             if files_list.startswith("❌"): # إذا لم توجد ملفات
-                 await query.edit_message_text(files_list, reply_markup=reply_markup_back)
-                 return SELECTING_ACTION
+                await query.edit_message_text(files_list, reply_markup=reply_markup_back)
+                return SELECTING_ACTION
 
             await query.edit_message_text(
                 f"📖 **شرح محاضرة**\n\n"
@@ -917,13 +907,11 @@ class MedicalLabBot:
         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:
@@ -962,11 +950,10 @@ class MedicalLabBot:
 
         return files_text
 
-
     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
@@ -974,7 +961,6 @@ class MedicalLabBot:
         subject = context.user_data.get('current_subject', 'general')
         logger.info(f"User {user_id} sent message: '{user_message}' | waiting_for: {waiting_for} | subject: {subject}")
 
-
         await update.message.chat.send_action(action="typing")
 
         response = "عذراً، لم أفهم طلبك. هل يمكنك توضيحه؟" # رد افتراضي
@@ -996,14 +982,14 @@ class MedicalLabBot:
 
             # إذا نجح الرد، أرسل قائمة الخيارات التالية
             if next_state == SELECTING_ACTION:
-                 context.user_data['waiting_for'] = None # إنهاء حالة الانتظار
-                 keyboard = [
-                     [InlineKeyboardButton("🔄 طرح سؤال آخر", callback_data="more_questions")],
-                     [InlineKeyboardButton("📚 تغيير المادة", callback_data="change_subject")],
-                     [InlineKeyboardButton("🏠 القائمة الرئيسية", callback_data="main_menu")]
-                 ]
-                 reply_markup = InlineKeyboardMarkup(keyboard)
-                 await update.message.reply_text("هل تحتاج مساعدة أخرى؟", reply_markup=reply_markup)
+                context.user_data['waiting_for'] = None # إنهاء حالة الانتظار
+                keyboard = [
+                    [InlineKeyboardButton("🔄 طرح سؤال آخر", callback_data="more_questions")],
+                    [InlineKeyboardButton("📚 تغيير المادة", callback_data="change_subject")],
+                    [InlineKeyboardButton("🏠 القائمة الرئيسية", callback_data="main_menu")]
+                ]
+                reply_markup = InlineKeyboardMarkup(keyboard)
+                await update.message.reply_text("هل تحتاج مساعدة أخرى؟", reply_markup=reply_markup)
 
             return next_state # العودة للحالة المناسبة (إما خيارات أو انتظار سؤال آخر)
 
@@ -1013,7 +999,6 @@ class MedicalLabBot:
             context.user_data['waiting_for'] = None # إنهاء الانتظار عند حدوث خطأ فادح
             return SELECTING_ACTION # العودة لقائمة الخيارات كحل احتياطي آمن
 
-
 # ========== إنشاء كائن البوت ==========
 bot = MedicalLabBot()
 
@@ -1038,7 +1023,6 @@ async def root():
         status_color = "#dc3545" # أحمر
         init_details = "فشل الاتصال بـ Telegram أو ضبط الويب هوك بعد عدة محاولات."
 
-
     return f"""
     <html>
         <head>
@@ -1140,7 +1124,6 @@ async def health_check():
         service_status = "unhealthy_failed_init"
         status_code = 500 # Internal Server Error
 
-
     response_payload = {
         "status": service_status,
         "service": "medical-lab-bot",
@@ -1165,4 +1148,4 @@ if __name__ == "__main__":
         port=port,
         log_level=log_level,
         reload=False # تعطيل إعادة التحميل التلقائي في الإنتاج
-        )
+    )