Update main.py

main.py

@@ -3,7 +3,8 @@ from fastapi import FastAPI, HTTPException, Query
 from pydantic import BaseModel
 from concurrent.futures import ThreadPoolExecutor
 import numpy as np
-from embeddingonnx import text_to_embedding, query_to_embedding  # دوال التحويل موجودة مسبقًا
+from embeddingonnx import text_to_embedding, query_to_embedding
+import threading
 
 # ==============================
 # إنشاء تطبيق FastAPI
@@ -11,7 +12,7 @@ from embeddingonnx import text_to_embedding, query_to_embedding  # دوال ال
 app = FastAPI(title="Arabic Text Embedding API")
 
 # ==============================
-# إنشاء ThreadPoolExecutor
+# ThreadPoolExecutor لمعالجة الدوال الثقيلة
 # ==============================
 executor = ThreadPoolExecutor(max_workers=2)
 
@@ -41,7 +42,6 @@ def embed_endpoint(request: TextRequest):
     if not text:
         raise HTTPException(status_code=400, detail="النص فارغ.")
     try:
-        # تنفيذ في ThreadPool
         future = executor.submit(text_to_embedding, text, True)
         vector = future.result()
         if vector is None:
@@ -56,7 +56,6 @@ def query_endpoint(request: TextRequest):
     if not query_text:
         raise HTTPException(status_code=400, detail="النص فارغ.")
     try:
-        # تنفيذ في ThreadPool
         future = executor.submit(query_to_embedding, query_text, True)
         vector = future.result()
         if vector is None:
@@ -97,13 +96,24 @@ def query_get(text: str = Query(..., description="النص المراد تحوي
         raise HTTPException(status_code=500, detail=f"خطأ أثناء إنشاء embedding للاستعلام: {str(e)}")
 
 # ==============================
-# تشغيل السيرفر مع warm-up اختياري
+# Warm-up تلقائي للنموذج عند بدء التشغيل
 # ==============================
-if __name__ == "__main__":
-    # Warm-up للنموذج لتقليل احتمال فشل أول طلب
-    dummy_text = "هذا نص للتجربة"
-    text_to_embedding(dummy_text)
-    query_to_embedding(dummy_text)
+def warmup_models():
+    try:
+        dummy_text = "هذا نص للتجربة"
+        # تنفيذ في ThreadPool لتجنب blocking السيرفر
+        executor.submit(text_to_embedding, dummy_text, True)
+        executor.submit(query_to_embedding, dummy_text, True)
+        print("✅ Warm-up للنموذج اكتمل.")
+    except Exception as e:
+        print(f"⚠️ خطأ أثناء warm-up: {e}")
+
+# تشغيل Warm-up في thread منفصل
+threading.Thread(target=warmup_models, daemon=True).start()
 
+# ==============================
+# تشغيل السيرفر
+# ==============================
+if __name__ == "__main__":
     import uvicorn
     uvicorn.run("main:app", host="0.0.0.0", port=8000)