Update app.py

app.py

@@ -31,21 +31,19 @@ DEEPSEEK_EMAIL = os.getenv("DEEPSEEK_EMAIL", "").strip()
 DEEPSEEK_PASSWORD = os.getenv("DEEPSEEK_PASSWORD", "").strip()
 DEEPSEEK_EMAIL_2 = os.getenv("DEEPSEEK_EMAIL_2", "").strip()
 DEEPSEEK_PASSWORD_2 = os.getenv("DEEPSEEK_PASSWORD_2", "").strip()
-DEEPSEEK_TOKEN = os.getenv("DEEPSEEK_TOKEN", "").strip()
-DEEPSEEK_TOKEN_2 = os.getenv("DEEPSEEK_TOKEN_2", "").strip()
 
 ACCOUNTS = []
 if DEEPSEEK_EMAIL and DEEPSEEK_PASSWORD:
     ACCOUNTS.append({
         "email": DEEPSEEK_EMAIL,
         "password": DEEPSEEK_PASSWORD,
-        "token": DEEPSEEK_TOKEN or ""
+        "token": ""
     })
 if DEEPSEEK_EMAIL_2 and DEEPSEEK_PASSWORD_2:
     ACCOUNTS.append({
         "email": DEEPSEEK_EMAIL_2,
         "password": DEEPSEEK_PASSWORD_2,
-        "token": DEEPSEEK_TOKEN_2 or ""
+        "token": ""
     })
 
 WASM_PATH = os.getenv("WASM_PATH", "./sha3_wasm_bg.7b9ca65ddd.wasm")
@@ -54,7 +52,6 @@ WASM_URL = "https://raw.githubusercontent.com/iidamie/deepseek2api/main/sha3_was
 KEEP_ALIVE_TIMEOUT = 5
 CLAUDE_DEFAULT_MODEL = "claude-sonnet-4-20250514"
 
-
 # ============================================================
 # Logging
 # ============================================================
@@ -91,16 +88,13 @@ templates = Jinja2Templates(directory="templates")
 # ============================================================
 
 def download_wasm_file():
-    """تحميل ملف WASM من GitHub إذا كان مفقوداً"""
     try:
         logger.info(f"محاولة تحميل WASM من: {WASM_URL}")
         urllib.request.urlretrieve(WASM_URL, WASM_PATH)
         if os.path.isfile(WASM_PATH):
             logger.info(f"تم تحميل WASM بنجاح إلى: {os.path.abspath(WASM_PATH)}")
             return True
-        else:
-            logger.error("فشل تحميل WASM: الملف لم يتم إنشاؤه")
-            return False
+        return False
     except Exception as e:
         logger.error(f"فشل تحميل WASM: {e}")
         return False
@@ -132,7 +126,7 @@ BASE_HEADERS = {
 
 
 # ============================================================
-# Startup event with WASM auto-download
+# Startup event
 # ============================================================
 
 @app.on_event("startup")
@@ -147,40 +141,36 @@ async def startup_event():
         for acc in ACCOUNTS:
             logger.info("  - %s", acc.get("email"))
     else:
-        logger.warning("No accounts configured. Please set DEEPSEEK_EMAIL and DEEPSEEK_PASSWORD.")
+        logger.warning("No accounts configured.")
     
-    # التحقق من وجود ملف WASM وتحميله تلقائياً
     if not os.path.isfile(WASM_PATH):
-        logger.warning(f"ملف WASM غير موجود: {os.path.abspath(WASM_PATH)}")
-        logger.info("محاولة تحميل الملف تلقائياً...")
+        logger.warning("WASM file missing, attempting download...")
         if download_wasm_file():
-            logger.info("تم تحميل WASM بنجاح")
+            logger.info("WASM downloaded successfully")
         else:
-            logger.critical("فشل تحميل WASM. سيعمل التطبيق بدون PoW (غير موصى به)")
+            logger.critical("Failed to download WASM file")
     else:
-        logger.info("WASM file found successfully.")
+        logger.info("WASM file found.")
     
     logger.info("=" * 60)
 
 
 # ============================================================
-# Authentication with token refresh
+# Authentication
 # ============================================================
 
 account_queue = ACCOUNTS.copy()
 random.shuffle(account_queue)
 
+# تخزين التوكنات مؤقتاً
+token_cache = {}
+
 def login_deepseek_via_account(account):
-    """تسجيل الدخول مع تسجيل تفصيلي للاستجابة"""
     email = account.get("email", "").strip()
     password = account.get("password", "").strip()
     
     if not email or not password:
-        logger.error("الحساب يفتقر إلى البريد الإلكتروني أو كلمة المرور")
-        raise HTTPException(
-            status_code=400,
-            detail="الحساب يجب أن يحتوي على email و password"
-        )
+        raise HTTPException(status_code=400, detail="الحساب يجب أن يحتوي على email و password")
     
     payload = {
         "email": email,
@@ -189,7 +179,7 @@ def login_deepseek_via_account(account):
         "os": "android",
     }
     
-    logger.info(f"محاولة تسجيل الدخول للحساب: {email}")
+    logger.info(f"تسجيل الدخول للحساب: {email}")
     
     try:
         resp = requests.post(
@@ -203,25 +193,23 @@ def login_deepseek_via_account(account):
         try:
             data = resp.json()
         except Exception as e:
-            logger.error(f"فشل تحليل استجابة JSON: {e}")
-            logger.error(f"نص الاستجابة: {resp.text[:500]}")
-            raise HTTPException(status_code=500, detail="استجابة غير صالحة من الخادم")
+            logger.error(f"فشل تحليل الاستجابة: {e}")
+            raise HTTPException(status_code=500, detail="استجابة غير صالحة")
         
         logger.info(f"استجابة تسجيل الدخول: code={data.get('code')}, msg={data.get('msg', 'N/A')}")
         
         if resp.status_code != 200 or data.get("code") != 0:
-            error_msg = data.get("msg", "خطأ غير معروف في تسجيل الدخول")
-            logger.error(f"فشل تسجيل الدخول للحساب {email}: {error_msg}")
+            error_msg = data.get("msg", "خطأ غير معروف")
+            logger.error(f"فشل تسجيل الدخول: {error_msg}")
             raise HTTPException(status_code=401, detail=f"فشل تسجيل الدخول: {error_msg}")
         
         user_data = data.get("data", {}).get("biz_data", {}).get("user")
         if not user_data:
-            logger.error(f"استجابة تسجيل الدخول غير مكتملة: {data}")
-            raise HTTPException(status_code=500, detail="استجابة تسجيل الدخول غير مكتملة")
+            logger.error(f"استجابة غير مكتملة: {data}")
+            raise HTTPException(status_code=500, detail="استجابة غير مكتملة")
         
         new_token = user_data.get("token")
         if not new_token:
-            logger.error(f"لم يتم العثور على token في الاستجابة: {user_data}")
             raise HTTPException(status_code=500, detail="لم يتم الحصول على token")
         
         account["token"] = new_token
@@ -230,47 +218,39 @@ def login_deepseek_via_account(account):
         
     except HTTPException:
         raise
-    except requests.RequestException as e:
-        logger.error(f"خطأ في طلب تسجيل الدخول للحساب {email}: {e}")
-        raise HTTPException(status_code=500, detail=f"خطأ في الاتصال: {str(e)}")
     except Exception as e:
-        logger.error(f"خطأ غير متوقع في تسجيل الدخول للحساب {email}: {e}")
-        raise HTTPException(status_code=500, detail=f"خطأ داخلي: {str(e)}")
+        logger.error(f"خطأ في تسجيل الدخول: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
 
-def refresh_token_for_account(account):
-    """تحديث التوكن لحساب معين"""
-    try:
-        logger.info(f"تحديث التوكن للحساب: {account.get('email')}")
-        return login_deepseek_via_account(account)
-    except Exception as e:
-        logger.error(f"فشل تحديث التوكن للحساب {account.get('email')}: {e}")
-        raise
 
 def get_valid_token():
-    """الحصول على token صالح مع إعادة المحاولة"""
     if not account_queue:
-        logger.error("لا توجد حسابات متاحة")
-        raise HTTPException(
-            status_code=500,
-            detail="لا توجد حسابات متاحة. تأكد من تعيين DEEPSEEK_EMAIL و DEEPSEEK_PASSWORD."
-        )
+        raise HTTPException(status_code=500, detail="لا توجد حسابات متاحة")
     
+    # نأخذ حساب من القائمة
     account = account_queue.pop(0)
     email = account.get("email", "unknown")
     
-    # محاولة تسجيل الدخول للحصول على token جديد دائماً
+    # التحقق من التخزين المؤقت
+    if email in token_cache and token_cache[email].get("token"):
+        logger.info(f"استخدام token مخزن للحساب: {email}")
+        token = token_cache[email]["token"]
+        account["token"] = token
+        account_queue.append(account)
+        return token, account
+    
+    # تسجيل الدخول
     try:
-        logger.info(f"تسجيل الدخول للحصول على token جديد: {email}")
         token = login_deepseek_via_account(account)
+        token_cache[email] = {"token": token, "timestamp": time.time()}
         account_queue.append(account)
         return token, account
     except Exception as e:
-        logger.error(f"فشل الحصول على token من الحساب {email}: {e}")
+        logger.error(f"فشل تسجيل الدخول للحساب {email}: {e}")
         if account_queue:
-            logger.info("محاولة استخدام الحساب التالي...")
             return get_valid_token()
-        else:
-            raise HTTPException(status_code=500, detail="فشل تسجيل الدخول لجميع الحسابات")
+        raise
+
 
 def get_deepseek_token():
     try:
@@ -278,30 +258,28 @@ def get_deepseek_token():
     except HTTPException:
         raise
     except Exception as e:
-        logger.error(f"خطأ غير متوقع في get_deepseek_token: {e}")
-        raise HTTPException(status_code=500, detail="خطأ داخلي في المصادقة")
+        logger.error(f"خطأ في get_deepseek_token: {e}")
+        raise HTTPException(status_code=500, detail="خطأ في المصادقة")
+
 
 def determine_mode_and_token(request: Request):
     try:
         token, account = get_deepseek_token()
         request.state.deepseek_token = token
         request.state.account = account
-        request.state.use_config_token = False
-        logger.info(f"تم تعيين التوكن للطلب بنجاح")
-    except HTTPException as e:
-        logger.error(f"فشل تعيين التوكن: {e.detail}")
-        raise
+        logger.info(f"تم تعيين التوكن للطلب")
     except Exception as e:
-        logger.error(f"خطأ غير متوقع في determine_mode_and_token: {e}")
-        raise HTTPException(status_code=500, detail="خطأ في المصادقة")
+        logger.error(f"فشل تعيين التوكن: {e}")
+        raise
+
 
 def determine_claude_mode_and_token(request: Request):
     determine_mode_and_token(request)
 
+
 def get_auth_headers(request: Request):
     token = getattr(request.state, "deepseek_token", None)
     if not token:
-        logger.warning("لم يتم العثور على token في الطلب، محاولة الحصول على واحد جديد")
         token, _ = get_deepseek_token()
         request.state.deepseek_token = token
     return {
@@ -311,7 +289,7 @@ def get_auth_headers(request: Request):
 
 
 # ============================================================
-# Utility functions (unchanged)
+# Utility functions
 # ============================================================
 
 def safe_json_response_error(status_code: int, message: str):
@@ -515,7 +493,7 @@ Do not write explanations before or after the JSON.
 
 
 # ============================================================
-# DeepSeek completion helpers
+# DeepSeek completion helpers - FIXED session_id extraction
 # ============================================================
 
 def call_completion_endpoint(payload, headers, max_attempts=3):
@@ -539,9 +517,13 @@ def call_completion_endpoint(payload, headers, max_attempts=3):
             time.sleep(1)
     return None
 
-def create_session(request: Request, max_attempts=3):
+def create_session(request: Request, max_attempts=2):
+    """
+    إنشاء جلسة DeepSeek مع إصلاح قراءة session_id
+    """
     for attempt in range(max_attempts):
         headers = get_auth_headers(request)
+        
         try:
             response = requests.post(
                 DEEPSEEK_CREATE_SESSION_URL,
@@ -550,84 +532,71 @@ def create_session(request: Request, max_attempts=3):
                 impersonate="safari15_3",
                 timeout=30
             )
+            
             try:
                 data = response.json()
-                logger.info(f"Create session response: code={data.get('code')}, msg={data.get('msg', 'N/A')}")
-                
-                # طباعة الاستجابة كاملة للتصحيح
-                if data.get('code') == 0:
-                    biz_data = data.get('data', {}).get('biz_data', {})
-                    logger.info(f"biz_data: {biz_data}")
+                logger.info(f"Create session response: code={data.get('code')}")
             except Exception as e:
-                logger.error(f"فشل تحليل استجابة إنشاء الجلسة: {e}")
+                logger.error(f"فشل تحليل الاستجابة: {e}")
                 data = {}
             
             if response.status_code == 200 and data.get("code") == 0:
-                # محاولة الحصول على session_id من عدة أماكن
-                session_id = (
-                    data.get("data", {}).get("biz_data", {}).get("id") or
-                    data.get("data", {}).get("id") or
-                    data.get("session_id")
-                )
-                response.close()
+                # ============================================================
+                # إصلاح قراءة session_id من المكان الصحيح
+                # ============================================================
+                biz_data = data.get("data", {}).get("biz_data", {})
+                chat_session = biz_data.get("chat_session", {})
+                session_id = chat_session.get("id")
+                
                 if session_id:
+                    response.close()
                     logger.info(f"تم إنشاء الجلسة: {session_id}")
                     return session_id
                 else:
-                    logger.warning(f"Create session: session_id missing in response. Data: {data}")
-                    # محاولة إعادة تسجيل الدخول للحصول على token جديد
+                    logger.warning(f"session_id غير موجود في الاستجابة")
+                    # نحاول مرة واحدة فقط إعادة تسجيل الدخول
+                    if attempt == 0:
+                        logger.info("محاولة تسجيل الدخول مرة أخرى...")
+                        account = getattr(request.state, "account", None)
+                        if account:
+                            try:
+                                # تحديث التوكن
+                                token = login_deepseek_via_account(account)
+                                request.state.deepseek_token = token
+                                continue
+                            except Exception as e:
+                                logger.error(f"فشل تحديث التوكن: {e}")
+                    continue
+            else:
+                error_msg = data.get("msg", f"HTTP {response.status_code}")
+                logger.warning(f"Create session failed: {error_msg}")
+                response.close()
+                
+                # في حالة 401، نحاول تحديث التوكن
+                if response.status_code == 401 and attempt == 0:
                     account = getattr(request.state, "account", None)
                     if account:
                         try:
-                            logger.info("محاولة تحديث التوكن...")
-                            new_token = refresh_token_for_account(account)
-                            request.state.deepseek_token = new_token
+                            token = login_deepseek_via_account(account)
+                            request.state.deepseek_token = token
                             continue
                         except Exception as e:
                             logger.error(f"فشل تحديث التوكن: {e}")
-            elif response.status_code == 401:
-                logger.warning("Create session: 401 Unauthorized, refreshing token...")
-                account = getattr(request.state, "account", None)
-                if account:
-                    try:
-                        new_token = refresh_token_for_account(account)
-                        request.state.deepseek_token = new_token
-                        continue
-                    except Exception as e:
-                        logger.error(f"فشل تحديث التوكن: {e}")
-                        try:
-                            token, new_account = get_deepseek_token()
-                            request.state.deepseek_token = token
-                            request.state.account = new_account
-                            continue
-                        except Exception as e2:
-                            logger.error(f"فشل الحصول على حساب آخر: {e2}")
-                else:
-                    try:
-                        token, new_account = get_deepseek_token()
-                        request.state.deepseek_token = token
-                        request.state.account = new_account
-                        continue
-                    except Exception as e:
-                        logger.error(f"فشل الحصول على حساب: {e}")
-            else:
-                error_msg = data.get("msg", f"HTTP {response.status_code}")
-                logger.warning(f"Create session failed: {error_msg}")
-                response.close()
+                
         except Exception as exc:
             logger.warning(f"Create session error: {exc}")
+        
         if attempt < max_attempts - 1:
             time.sleep(1)
+    
     return None
 
 def compute_pow_answer(algorithm, challenge_str, salt, difficulty, expire_at, signature, target_path, wasm_path):
     if algorithm != "DeepSeekHashV1":
         raise ValueError(f"Unsupported PoW algorithm: {algorithm}")
     if not os.path.isfile(wasm_path):
-        logger.error(f"WASM file not found: {wasm_path}")
-        # محاولة تحميل الملف
         if download_wasm_file():
-            logger.info("تم تحميل WASM بنجاح")
+            logger.info("WASM downloaded successfully")
         else:
             raise FileNotFoundError(f"WASM file not found: {wasm_path}")
     
@@ -677,7 +646,7 @@ def compute_pow_answer(algorithm, challenge_str, salt, difficulty, expire_at, si
     finally:
         add_to_stack(store, 16)
 
-def get_pow_response(request: Request, max_attempts=3):
+def get_pow_response(request: Request, max_attempts=2):
     for attempt in range(max_attempts):
         headers = get_auth_headers(request)
         try:
@@ -724,34 +693,18 @@ def get_pow_response(request: Request, max_attempts=3):
                 response.close()
                 logger.info("تم الحصول على PoW بنجاح")
                 return encoded
-            elif response.status_code == 401:
+            elif response.status_code == 401 and attempt == 0:
                 logger.warning("PoW: 401 Unauthorized, refreshing token...")
                 account = getattr(request.state, "account", None)
                 if account:
                     try:
-                        new_token = refresh_token_for_account(account)
-                        request.state.deepseek_token = new_token
-                        continue
-                    except Exception as e:
-                        logger.error(f"فشل تحديث التوكن: {e}")
-                        try:
-                            token, new_account = get_deepseek_token()
-                            request.state.deepseek_token = token
-                            request.state.account = new_account
-                            continue
-                        except Exception as e2:
-                            logger.error(f"فشل الحصول على حساب آخر: {e2}")
-                else:
-                    try:
-                        token, new_account = get_deepseek_token()
+                        token = login_deepseek_via_account(account)
                         request.state.deepseek_token = token
-                        request.state.account = new_account
                         continue
                     except Exception as e:
-                        logger.error(f"فشل الحصول على حساب: {e}")
+                        logger.error(f"فشل تحديث التوكن: {e}")
             else:
-                logger.warning("PoW request failed: HTTP=%s code=%s msg=%s",
-                               response.status_code, data.get("code"), data.get("msg"))
+                logger.warning("PoW request failed: HTTP=%s code=%s", response.status_code, data.get("code"))
                 response.close()
         except Exception as exc:
             logger.error("PoW error: %s", exc)
@@ -771,9 +724,7 @@ def delete_session(request: Request, session_id: str):
             timeout=5
         )
         if response.status_code == 200:
-            logger.info("Deleted DeepSeek session: %s", session_id)
-        else:
-            logger.warning("Delete session failed: HTTP %s", response.status_code)
+            logger.info("Deleted session: %s", session_id)
         response.close()
     except Exception as exc:
         logger.warning("Delete session error: %s", exc)
@@ -831,7 +782,7 @@ def parse_deepseek_stream(response, on_text=None, on_thinking=None):
 
 
 # ============================================================
-# Rest of the endpoints (same as before)
+# Endpoints (unchanged)
 # ============================================================
 
 @app.get("/v1/models")
@@ -844,6 +795,7 @@ async def list_models():
     ]
     return {"object": "list", "data": models}
 
+
 @app.post("/v1/chat/completions")
 async def chat_completions(request: Request):
     try:
@@ -873,10 +825,12 @@ async def chat_completions(request: Request):
         
         final_prompt = messages_prepare(messages)
         
+        # إنشاء الجلسة
         session_id = create_session(request)
         if not session_id:
             return safe_json_response_error(401, "Unable to create DeepSeek session.")
         
+        # الحصول على PoW
         pow_response = get_pow_response(request)
         if not pow_response:
             delete_session(request, session_id)
@@ -903,6 +857,7 @@ async def chat_completions(request: Request):
         created = int(time.time())
         completion_id = f"chatcmpl-{session_id}"
         
+        # Streaming
         if req_data.get("stream", False):
             def stream_generator():
                 final_text = []
@@ -1022,6 +977,7 @@ async def chat_completions(request: Request):
                 headers={"Cache-Control": "no-cache", "Connection": "keep-alive", "X-Accel-Buffering": "no"}
             )
         
+        # Non-streaming
         final_text = []
         final_thinking = []
         parse_deepseek_stream(deepseek_response, on_text=final_text.append, on_thinking=final_thinking.append)
@@ -1061,7 +1017,7 @@ async def chat_completions(request: Request):
 
 
 # ============================================================
-# Claude endpoints (shortened - same as before)
+# Claude endpoints (shortened)
 # ============================================================
 
 @app.get("/anthropic/v1/models")