Update app.py

app.py

@@ -28,7 +28,7 @@ logger = logging.getLogger(__name__)
 # ========== التوكنات ومتغيرات البيئة ==========
 TELEGRAM_BOT_TOKEN = os.environ.get('TELEGRAM_BOT_TOKEN')
 NVAPI_API_KEY = os.environ.get('NVAPI_API_KEY')
-WEBHOOK_URL = os.environ.get('SPACE_URL', '')
+SPACE_URL = os.environ.get('SPACE_URL', '')
 
 # ========== تكوين عميل NVIDIA ==========
 nvidia_client = OpenAI(
@@ -58,39 +58,82 @@ class MedicalLabBot:
         """تهيئة تطبيق التليجرام"""
         try:
             if not self.application:
-                # استخدام تكوين مبسط لتجنب مشكلة proxies
-                self.application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
+                # استخدام تكوين مبسط مع إعدادات محددة
+                self.application = (
+                    Application.builder()
+                    .token(TELEGRAM_BOT_TOKEN)
+                    .build()
+                )
                 self.setup_handlers()
                 logger.info("✅ تم تهيئة تطبيق التليجرام بنجاح")
         except Exception as e:
             logger.error(f"❌ خطأ في تهيئة التطبيق: {e}")
-            # محاولة بديلة
-            self.application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
-            self.setup_handlers()
+            raise
     
     def setup_handlers(self):
         """إعداد معالجات التليجرام"""
+        # إعداد Conversation Handler مع إعدادات محسنة
         conv_handler = ConversationHandler(
             entry_points=[CommandHandler('start', self.start)],
             states={
                 SELECTING_SUBJECT: [
-                    CallbackQueryHandler(self.handle_subject_selection, pattern='^subject_|general_help|refresh_materials$')
+                    CallbackQueryHandler(self.handle_subject_selection, pattern='^(subject_|general_help|refresh_materials)')
                 ],
                 SELECTING_ACTION: [
-                    CallbackQueryHandler(self.handle_action_selection),
-                    CallbackQueryHandler(self.handle_subject_selection, pattern='^subject_')
+                    CallbackQueryHandler(self.handle_action_selection, pattern='^(explain_lecture|browse_files|generate_questions|summarize_content|explain_concept|main_menu)$')
                 ],
                 WAITING_FOR_QUESTION: [
-                    MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_message)
+                    MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_message),
+                    CallbackQueryHandler(self.handle_back_actions, pattern='^back_to_actions$')
                 ]
             },
-            fallbacks=[CommandHandler('start', self.start)]
+            fallbacks=[
+                CommandHandler('start', self.start),
+                CallbackQueryHandler(self.handle_main_menu, pattern='^main_menu$')
+            ],
+            name="medical_lab_conversation",
+            persistent=False,
+            per_message=False  # هذا الإعداد يقلل التحذيرات
         )
         
+        # إضافة الـ handlers
         self.application.add_handler(conv_handler)
-        self.application.add_handler(CallbackQueryHandler(self.handle_callback_query, pattern='^more_questions|change_subject|back_to_actions$'))
+        
+        # إضافة 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$'))
+        
         logger.info("✅ تم إعداد معالجات التليجرام")
     
+    async def handle_back_actions(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """معالجة العودة إلى القائمة"""
+        query = update.callback_query
+        await query.answer()
+        return await self.handle_action_selection(update, context)
+    
+    async def handle_main_menu(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """معالجة العودة للقائمة الرئيسية"""
+        query = update.callback_query
+        await query.answer()
+        return await self.start(update, context)
+    
+    async def handle_more_questions(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """معالجة طلب المزيد من الأسئلة"""
+        query = update.callback_query
+        await query.answer()
+        await query.edit_message_text("💬 اكتب سؤالك أو طلبك:")
+        context.user_data['waiting_for'] = 'general'
+        return WAITING_FOR_QUESTION
+    
+    async def handle_change_subject(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
+        """معالجة تغيير المادة"""
+        query = update.callback_query
+        await query.answer()
+        keyboard = self.create_subjects_keyboard()
+        reply_markup = InlineKeyboardMarkup(keyboard)
+        await query.edit_message_text("اختر المادة:", reply_markup=reply_markup)
+        return SELECTING_SUBJECT
+
     def load_all_materials(self):
         """تحميل جميع المواد والملفات من Hugging Face"""
         try:
@@ -132,10 +175,10 @@ class MedicalLabBot:
                     continue
             
             self.available_materials = materials
-            logger.info(f"تم تحميل {len(materials)} مادة بنجاح")
+            logger.info(f"✅ تم تحميل {len(materials)} مادة بنجاح")
             
         except Exception as e:
-            logger.error(f"خطأ في تحميل المواد: {e}")
+            logger.error(f"❌ خطأ في تحميل المواد: {e}")
             self.available_materials = {'Biochemistry': {'files': [], 'file_details': {}}}
     
     def extract_file_info(self, file_name, file_path):
@@ -170,7 +213,7 @@ class MedicalLabBot:
             if file_path in self.file_cache:
                 return self.file_cache[file_path]
             
-            logger.info(f"جاري تحميل الملف: {file_path}")
+            logger.info(f"📥 جاري تحميل الملف: {file_path}")
             
             local_path = hf_hub_download(
                 repo_id=REPO_ID,
@@ -195,7 +238,7 @@ class MedicalLabBot:
             return content
             
         except Exception as e:
-            logger.error(f"خطأ في تحميل الملف {file_path}: {e}")
+            logger.error(f"❌ خطأ في تحميل الملف {file_path}: {e}")
             return f"❌ تعذر تحميل محتوى الملف: {str(e)}"
     
     async def extract_pdf_content(self, pdf_path):
@@ -231,7 +274,7 @@ class MedicalLabBot:
             return text_content
             
         except Exception as e:
-            logger.error(f"خطأ في استخراج محتوى PDF: {e}")
+            logger.error(f"❌ خطأ في استخراج محتوى PDF: {e}")
             return f"❌ حدث خطأ في قراءة ملف PDF: {str(e)}"
     
     async def extract_doc_content(self, doc_path):
@@ -242,7 +285,7 @@ class MedicalLabBot:
             content += "للاستفادة الكاملة، يرجى تحويل الملف إلى PDF أو طلب شرح محتوى معين."
             return content
         except Exception as e:
-            logger.error(f"خطأ في استخراج محتوى Word: {e}")
+            logger.error(f"❌ خطأ في استخراج محتوى Word: {e}")
             return f"❌ حدث خطأ في قراءة ملف Word: {str(e)}"
     
     def get_user_memory(self, user_id):
@@ -296,7 +339,11 @@ class MedicalLabBot:
         keyboard = self.create_subjects_keyboard()
         reply_markup = InlineKeyboardMarkup(keyboard)
         
-        await update.message.reply_text(welcome_text, reply_markup=reply_markup)
+        if update.callback_query:
+            await update.callback_query.edit_message_text(welcome_text, reply_markup=reply_markup)
+        else:
+            await update.message.reply_text(welcome_text, reply_markup=reply_markup)
+        
         return SELECTING_SUBJECT
     
     def create_subjects_keyboard(self):
@@ -366,10 +413,7 @@ class MedicalLabBot:
         subject = context.user_data.get('current_subject', 'general')
         
         if action == "main_menu":
-            keyboard = self.create_subjects_keyboard()
-            reply_markup = InlineKeyboardMarkup(keyboard)
-            await query.edit_message_text("اختر المادة:", reply_markup=reply_markup)
-            return SELECTING_SUBJECT
+            return await self.start(update, context)
         
         elif action == "browse_files":
             return await self.browse_available_files(query, context)
@@ -479,7 +523,7 @@ class MedicalLabBot:
             return SELECTING_ACTION
             
         except Exception as e:
-            logger.error(f"Error processing message: {e}")
+            logger.error(f"❌ Error processing message: {e}")
             await update.message.reply_text("❌ حدث خطأ أثناء معالجة طلبك. يرجى المحاولة مرة أخرى.")
             return SELECTING_ACTION
     
@@ -523,7 +567,7 @@ class MedicalLabBot:
             return f"**📖 شرح {file_name}**\n\n{response}"
             
         except Exception as e:
-            logger.error(f"Error explaining lecture: {e}")
+            logger.error(f"❌ Error explaining lecture: {e}")
             return "❌ عذراً، حدث خطأ في شرح المحاضرة. يرجى المحاولة مرة أخرى."
     
     async def find_requested_file(self, request, subject):
@@ -582,7 +626,7 @@ class MedicalLabBot:
             return response
             
         except Exception as e:
-            logger.error(f"Error explaining concept: {e}")
+            logger.error(f"❌ Error explaining concept: {e}")
             return await self.fallback_explain_concept(concept, subject, user_id)
     
     async def fallback_explain_concept(self, concept, subject, user_id):
@@ -629,7 +673,7 @@ class MedicalLabBot:
             return f"**❓ أسئلة {subject}**\n\n{response}"
             
         except Exception as e:
-            logger.error(f"Error generating questions: {e}")
+            logger.error(f"❌ Error generating questions: {e}")
             return await self.fallback_generate_questions(subject, user_id)
     
     async def fallback_generate_questions(self, subject, user_id):
@@ -716,12 +760,12 @@ class MedicalLabBot:
             
             reasoning = getattr(completion.choices[0].message, "reasoning_content", None)
             if reasoning:
-                logger.info(f"Reasoning content for user {user_id}: {reasoning}")
+                logger.info(f"🤔 Reasoning content for user {user_id}: {reasoning}")
             
             return completion.choices[0].message.content
             
         except Exception as e:
-            logger.error(f"Error calling NVIDIA AI: {e}")
+            logger.error(f"❌ Error calling NVIDIA AI: {e}")
             return "❌ عذراً، حدث خطأ في الاتصال بالذكاء الاصطناعي. يرجى المحاولة مرة أخرى لاحقاً."
     
     async def handle_general_help(self, query, context):
@@ -761,43 +805,6 @@ class MedicalLabBot:
         
         await query.edit_message_text(help_text, reply_markup=reply_markup)
         return SELECTING_SUBJECT
-    
-    async def handle_callback_query(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
-        """معالجة استعلامات الـ Callback"""
-        query = update.callback_query
-        await query.answer()
-        
-        if query.data == "more_questions":
-            await query.edit_message_text("💬 اكتب سؤالك أو طلبك:")
-            context.user_data['waiting_for'] = 'general'
-            return WAITING_FOR_QUESTION
-        
-        elif query.data == "change_subject":
-            keyboard = self.create_subjects_keyboard()
-            reply_markup = InlineKeyboardMarkup(keyboard)
-            await query.edit_message_text("اختر المادة:", reply_markup=reply_markup)
-            return SELECTING_SUBJECT
-        
-        elif query.data == "back_to_actions":
-            subject = context.user_data.get('current_subject')
-            subject_name = subject.replace('_', ' ').title() if subject else "المادة"
-            
-            keyboard = [
-                [InlineKeyboardButton("📖 شرح محاضرة محددة", callback_data="explain_lecture")],
-                [InlineKeyboardButton("🔍 استعراض جميع الملفات", callback_data="browse_files")],
-                [InlineKeyboardButton("❓ أسئلة عن المادة", callback_data="generate_questions")],
-                [InlineKeyboardButton("📝 ملخص المادة", callback_data="summarize_content")],
-                [InlineKeyboardButton("🧪 تفسير مفهوم", callback_data="explain_concept")],
-                [InlineKeyboardButton("🏠 القائمة الرئيسية", callback_data="main_menu")]
-            ]
-            reply_markup = InlineKeyboardMarkup(keyboard)
-            
-            await query.edit_message_text(
-                f"📚 **{subject_name}**\n\nاختر الخدمة التي تريدها:",
-                reply_markup=reply_markup,
-                parse_mode='Markdown'
-            )
-            return SELECTING_ACTION
 
 # ========== إنشاء كائن البوت ==========
 bot = MedicalLabBot()
@@ -810,14 +817,15 @@ async def on_startup():
         # تهيئة التطبيق
         bot.initialize_application()
         
-        if WEBHOOK_URL:
+        if SPACE_URL:
+            webhook_url = f"{SPACE_URL.rstrip('/')}/telegram"
             await bot.application.bot.set_webhook(
-                url=f"{WEBHOOK_URL}/telegram",
+                url=webhook_url,
                 allowed_updates=Update.ALL_TYPES
             )
-            logger.info(f"✅ Webhook set to: {WEBHOOK_URL}/telegram")
+            logger.info(f"✅ Webhook set to: {webhook_url}")
         else:
-            logger.warning("⚠️ WEBHOOK_URL not set, using polling instead")
+            logger.warning("⚠️ SPACE_URL not set, using polling instead")
             
     except Exception as e:
         logger.error(f"❌ Error during startup: {e}")
@@ -830,9 +838,12 @@ async def root():
         <head>
             <title>Medical Lab Bot</title>
             <style>
-                body { font-family: Arial, sans-serif; margin: 40px; }
-                .container { max-width: 800px; margin: 0 auto; }
-                .status { padding: 20px; background: #f0f8ff; border-radius: 10px; }
+                body { font-family: Arial, sans-serif; margin: 40px; background: #f5f5f5; }
+                .container { max-width: 800px; margin: 0 auto; background: white; padding: 30px; border-radius: 15px; box-shadow: 0 4px 6px rgba(0,0,0,0.1); }
+                .status { padding: 20px; background: #f0f8ff; border-radius: 10px; border-left: 5px solid #007bff; }
+                h1 { color: #2c3e50; text-align: center; }
+                ul { line-height: 1.8; }
+                li { margin: 10px 0; }
             </style>
         </head>
         <body>
@@ -841,17 +852,21 @@ async def root():
                 <div class="status">
                     <h2>✅ البوت يعمل بنجاح</h2>
                     <p>البوت متصل بـ Telegram وجاهز لاستقبال الرسائل.</p>
-                    <p>استخدم الرابط التالي للتواصل مع البوت:</p>
-                    <p><strong>https://t.me/your_bot_username</strong></p>
+                    <p><strong>الحالة:</strong> ✅ نشط</p>
+                    <p><strong>المواد المحملة:</strong> """ + str(len(bot.available_materials)) + """ مادة</p>
                 </div>
-                <h3>المميزات:</h3>
+                <h3>🎯 المميزات:</h3>
                 <ul>
                     <li>📚 شرح المواد الدراسية من ملفات PDF وWord</li>
                     <li>❓ توليد أسئلة متنوعة للمراجعة</li>
                     <li>📖 تلخيص المحتوى الدراسي</li>
                     <li>🧪 تفسير المفاهيم العلمية</li>
-                    <li>🎯 اختبار الفهم</li>
+                    <li>🎯 اختبار الفهم والاستيعاب</li>
+                    <li>💾 ذاكرة محادثة لكل مستخدم</li>
                 </ul>
+                <div class="status" style="margin-top: 20px;">
+                    <p><strong>ℹ️ معلومات:</strong> هذا البوت يستخدم الذكاء الاصطناعي من NVIDIA وملفات Hugging Face</p>
+                </div>
             </div>
         </body>
     </html>
@@ -872,9 +887,19 @@ async def handle_telegram_update(request: Request):
 @app.get("/health")
 async def health_check():
     """فحص صحة الخدمة"""
-    return {"status": "healthy", "service": "medical-lab-bot"}
+    materials_count = len(bot.available_materials)
+    total_files = sum(len(material['files']) for material in bot.available_materials.values())
+    
+    return {
+        "status": "healthy", 
+        "service": "medical-lab-bot",
+        "materials_loaded": materials_count,
+        "total_files": total_files,
+        "timestamp": datetime.now().isoformat()
+    }
 
 # ========== التشغيل الرئيسي ==========
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
+    logger.info(f"🚀 Starting Medical Lab Bot on port {port}")
     uvicorn.run(app, host="0.0.0.0", port=port)