Update app.py

app.py

@@ -6,6 +6,9 @@ from curl_cffi import requests
 from lxml import html
 import pycountry
 
+# إضافة مكتبة TikTokApi لجلب قائمة المتابعين الذين يتابعهم المستخدم
+from TikTokApi import TikTokApi
+
 
 class TikTokUser:
     def __init__(self, data) -> None:
@@ -27,11 +30,9 @@ class TikTokUser:
         self.video_count = stats.get("videoCount", 0)
         self.friends = stats.get("friendCount", 0)
 
-        # تحويل الطابع الزمني إلى صيغة قابلة للقراءة
         if self.created:
             self.created = datetime.datetime.fromtimestamp(self.created).strftime("%Y-%m-%d %H:%M:%S")
 
-        # تحويل رمز البلد واللغة إلى الأسماء
         if self.country:
             c = pycountry.countries.get(alpha_2=self.country.upper())
             self.country = c.name if c else self.country
@@ -40,7 +41,6 @@ class TikTokUser:
             l = pycountry.languages.get(alpha_2=self.language.lower())
             self.language = l.name if l else self.language
 
-        # إضافة رابط bio إذا وُجد
         if bio_link := user.get("bioLink", ""):
             self.bio += "\n" + bio_link.get("link", "")
 
@@ -84,7 +84,7 @@ def classify_bio(bio: str) -> str:
     يصنف السيرة الذاتية إلى فئات مبسطة بناءً على كلمات مفتاحية.
     """
     lower_bio = bio.lower()
-    if any(kw in lower_bio for kw in ["tech", "ai", "developer", "software", "python"]):
+    if any(kw in lower_bio for kw in ["tech", "ai", "developer", "python", "software"]):
         return "تكنولوجيا"
     if any(kw in lower_bio for kw in ["sports", "football", "fitness", "gym", "athlete"]):
         return "رياضة"
@@ -95,23 +95,25 @@ def classify_bio(bio: str) -> str:
     return "عام"
 
 
-def extract_following(page_html: str, limit: int = 10) -> list[str]:
+def get_following_list(username: str, limit: int = 10) -> list[str]:
     """
-    يحاول استخراج قائمة الأسماء (uniqueId) للحسابات التي يتابعها المستخدم،
-    بأخذ أول N = limit اسمًا يظهر في HTML. هذه طريقة تقديرية وليست رسمية.
+    يستخدم TikTokApi لجلب قائمة أسماء المستخدمين التي يتابعها الحساب،
+    وينتهي عند أول `limit` مستخدم.
     """
-    # نبحث عن كل مواضع "uniqueId":"someUser"
-    matches = re.findall(r'"uniqueId":"(.*?)"', page_html)
-    # نمسح duplicates ونقتصر على first 'limit'
-    seen = set()
-    following = []
-    for uid in matches:
-        if uid not in seen:
-            seen.add(uid)
-            following.append(uid)
-            if len(following) >= limit:
-                break
-    return following
+    following_usernames = []
+    try:
+        with TikTokApi() as api:
+            user = api.user(username=username)
+            count = 0
+            for f in user.following(count=limit):
+                following_usernames.append(f.as_dict.get("uniqueId", ""))
+                count += 1
+                if count >= limit:
+                    break
+    except Exception:
+        # إذا فشل TikTokApi لأي سبب (مثل تغيير API)، نعطي قائمة فارغة
+        return []
+    return following_usernames
 
 
 def lookup_tiktok(username: str):
@@ -123,19 +125,16 @@ def lookup_tiktok(username: str):
     # تصنيف السيرة الذاتية
     category = classify_bio(user.bio)
 
-    # استخراج أول 10 أسماء يتابعها
-    following_list = extract_following(page_html, limit=10)
-
-    # تنسيق قائمة المتابعين الذين يتابعهم
+    # جلب حتى 10 أشخاص يتابعهم المستخدم
+    following_list = get_following_list(username, limit=10)
     if following_list:
         following_md = "\n".join(f"- @{uid}" for uid in following_list)
     else:
-        following_md = "لا توجد بيانات متاحة أو الحساب لا يعرض من يتابعهم."
+        following_md = "لا توجد بيانات متاحة أو تعذَّر جلب القائمة."
 
-    # تركيب النص النهائي
+    # إعداد النص النهائي المنسق
     info_md = f"""
-**الاسم (Name):** {user.name}  
-**اسم المستخدم:** @{user.username}  
+**(Name):** {user.name}  
 **البلد (Country):** {user.country}  
 **اللغة (Language):** {user.language}  
 **تاريخ الإنشاء (Created):** {user.created}