Update modules/ai_assistant/ai_app.py

modules/ai_assistant/ai_app.py

@@ -1,8 +1,9 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-وحدة مساعد الذكاء الاصطناعي
-تقوم بتوفير فئة AIAssistantApp للتوافق مع التطبيق الرئيسي
+وحدة مساعد الذكاء الاصطناعي (ai_app.py)
+توفر فئة AIAssistantApp للتوافق مع التطبيق الرئيسي
+تم إصلاح مشكلة الوصول إلى مفاتيح API ومشكلة دالة render المفقودة
 """
 
 import os
@@ -36,11 +37,50 @@ def _get_huggingface_secret(secret_name):
     """
     # تحويل اسم السر إلى الصيغة المستخدمة في متغيرات البيئة في هجين فيس
     env_var_name = f"HF_SECRET_{secret_name.upper()}"
+    
+    # محاولة الحصول على القيمة من متغيرات البيئة
     secret_value = os.environ.get(env_var_name, '')
     
-    # إذا لم يتم العثور على السر، نعرض رسالة تحذير (في وضع التطوير فقط)
+    # إذا لم يتم العثور على القيمة، نحاول الحصول عليها من st.secrets
+    if not secret_value and hasattr(st, 'secrets'):
+        try:
+            # محاولة الوصول إلى السر مباشرة
+            if hasattr(st.secrets, secret_name):
+                secret_value = getattr(st.secrets, secret_name)
+            # محاولة الوصول إلى السر كعنصر في القاموس
+            elif secret_name in st.secrets:
+                secret_value = st.secrets[secret_name]
+            # محاولة الوصول إلى السر من خلال طريقة get
+            else:
+                secret_value = st.secrets.get(secret_name, '')
+        except:
+            pass
+    
+    # محاولة ثالثة: البحث عن المفتاح بشكل مباشر في متغيرات البيئة
     if not secret_value:
-        st.warning(f"لم يتم العثور على السر: {secret_name}. تأكد من إضافته في إعدادات الأسرار في هجين فيس.")
+        # محاولة البحث عن المفتاح بأشكال مختلفة
+        possible_names = [
+            secret_name,
+            secret_name.upper(),
+            f"HF_{secret_name}",
+            f"HF_{secret_name.upper()}",
+            f"HUGGINGFACE_{secret_name}",
+            f"HUGGINGFACE_{secret_name.upper()}"
+        ]
+        
+        for name in possible_names:
+            if name in os.environ:
+                secret_value = os.environ[name]
+                break
+    
+    # محاولة رابعة: قيم افتراضية للاختبار (فقط في بيئة التطوير)
+    if not secret_value and os.environ.get('DEVELOPMENT_MODE') == 'true':
+        if secret_name.lower() == 'anthropic':
+            # قيمة افتراضية للاختبار فقط
+            secret_value = 'sk-ant-api-test-key'
+        elif secret_name.lower() == 'ai':
+            # قيمة افتراضية للاختبار فقط
+            secret_value = 'sk-ai-api-test-key'
     
     return secret_value
 
@@ -49,9 +89,25 @@ class ClaudeAIService:
     """خدمة التفاعل مع Claude AI API"""
     
     def __init__(self, api_key=None):
-        self.api_key = api_key or _get_huggingface_secret("anthropic")
+        # محاولة الحصول على المفتاح بعدة طرق
+        if not api_key:
+            api_key = _get_huggingface_secret("anthropic")
+            
+            # إذا لم يتم العثور على المفتاح، نحاول بأسماء بديلة
+            if not api_key:
+                for alt_name in ["ANTHROPIC_API_KEY", "CLAUDE_API_KEY", "CLAUDE"]:
+                    api_key = _get_huggingface_secret(alt_name)
+                    if api_key:
+                        break
+        
+        self.api_key = api_key
+        
         if self.api_key:
-            self.client = anthropic.Anthropic(api_key=self.api_key)
+            try:
+                self.client = anthropic.Anthropic(api_key=self.api_key)
+            except Exception as e:
+                print(f"خطأ في تهيئة عميل Anthropic: {str(e)}")
+                self.client = None
         else:
             self.client = None
             
@@ -117,6 +173,13 @@ class AIAssistantApp:
         # تهيئة خدمة Claude AI
         self.claude_service = ClaudeAIService(api_key=st.session_state.anthropic_api_key)
     
+    def render(self):
+        """
+        عرض واجهة تطبيق مساعد الذكاء الاصطناعي
+        هذه الدالة مطلوبة للتوافق مع التطبيق الرئيسي
+        """
+        self.run()
+    
     def run(self):
         """تشغيل تطبيق مساعد الذكاء الاصطناعي"""
         # عرض عنوان التطبيق