Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import cv2
 import numpy as np
 import gradio as gr
+import gradio.utils as gr_utils
 from ultralytics import YOLO
 from PIL import Image, ImageDraw, ImageFont
 import arabic_reshaper
@@ -13,19 +14,29 @@ import google.generativeai as genai
 import torch
 
 # ==========================
-# ⚠️ هنا تحط مفتاحك مباشرة
+# ⚠️ Gemini API Key
 # ==========================
 
-GEMINI_API_KEY = "YOUR_GEMINI_API_KEY_HERE"  # 👈 استبدله بمفتاحك
-
+GEMINI_API_KEY = "YOUR_GEMINI_API_KEY_HERE"  # ← حط المفتاح هنا
 genai.configure(api_key=GEMINI_API_KEY)
 
+# ==========================
+# Patch لمشكلة Spaces hot-reload
+# ==========================
+
+# داخل Hugging Face Spaces، SPACE_ID يكون موجود في الـ env
+if os.getenv("SPACE_ID"):
+    def _no_watchfn_spaces(*args, **kwargs):
+        # نطفي الـ hot-reload اللي يسبب RuntimeError
+        return
+    # نكتب فوق الدالة الأصلية
+    gr_utils.watchfn_spaces = _no_watchfn_spaces
+
 # ==========================
 # إعدادات أداء PyTorch / GPU
 # ==========================
 
-# تفعيل تسريع CUDNN
-torch.backends.cudnn.benchmark = True
+torch.backends.cudnn.benchmark = True  # تسريع الـ conv على GPU
 
 DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
 USE_HALF = DEVICE.startswith("cuda")
@@ -53,10 +64,6 @@ arabic_map = {
     "toot": "ة", "al": "ال"
 }
 
-# ==========================
-# نص توجيهات Gemini
-# ==========================
-
 SYSTEM_PROMPT = (
     "أنت مساعد ذكي يستقبل كلمات أو جمل قصيرة قادمة من مترجم لغة "
     "الإشارة العربية، ودورك أن تعيد صياغتها كنص عربي واضح ومفهوم، "
@@ -64,7 +71,7 @@ SYSTEM_PROMPT = (
 )
 
 # ==========================
-# إعداد خط عربي مرة وحدة (بدون تكرار)
+# إعداد خط عربي مرة وحدة
 # ==========================
 
 DEFAULT_FONT_SIZE = 24
@@ -77,21 +84,15 @@ except Exception:
 
 
 def prepare_arabic(text: str) -> str:
-    """تجهيز النص العربي (reshaper + bidi) بدون رسم."""
     reshaped = arabic_reshaper.reshape(text)
     bidi_text = get_display(reshaped)
     return bidi_text
 
-
 # ==========================
-# رسم الديتكشن بشكل أسرع
+# رسم الديتكشن بشكل سريع
 # ==========================
 
 def draw_detections(result, frame, names):
-    """
-    نرسم البوكسات بـ OpenCV،
-    وبعدين نحول الصورة إلى PIL مرة واحدة ونرسم كل النصوص.
-    """
     boxes = result.boxes
     detected_labels = []
 
@@ -112,7 +113,7 @@ def draw_detections(result, frame, names):
         ar_label = arabic_map.get(eng_label, eng_label)
         detected_labels.append(ar_label)
 
-        # رسم البوكس
+        # box
         cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
 
         label_bg_y1 = max(0, y1 - 35)
@@ -127,13 +128,12 @@ def draw_detections(result, frame, names):
 
         label_infos.append(
             (
-                prepare_arabic(ar_label),  # نص عربي جاهز للرسم
+                prepare_arabic(ar_label),
                 x1 + 5,
                 label_bg_y1 + 5,
             )
         )
 
-    # نحول لمرة وحدة إلى PIL ونرسم كل النصوص
     img_pil = Image.fromarray(frame)
     draw = ImageDraw.Draw(img_pil)
 
@@ -142,7 +142,6 @@ def draw_detections(result, frame, names):
 
     return np.array(img_pil), detected_labels
 
-
 # ==========================
 # تحميل YOLO على GPU + half
 # ==========================
@@ -151,7 +150,6 @@ print("🔹 Loading YOLO model...")
 model = YOLO(WEIGHTS_PATH)
 model.to(DEVICE)
 
-# تفعيل half precision لو معنا GPU
 if USE_HALF:
     try:
         model.model.half()
@@ -161,7 +159,6 @@ if USE_HALF:
 
 print("📚 Classes:", model.names)
 
-
 # ==========================
 # Gemini API Call
 # ==========================
@@ -171,7 +168,6 @@ def call_gemini_on_word(word: str) -> str:
         return ""
 
     try:
-        # flash سريع أصلا، نخليه كما هو
         model_g = genai.GenerativeModel("gemini-1.5-flash")
 
         prompt = (
@@ -185,7 +181,6 @@ def call_gemini_on_word(word: str) -> str:
     except Exception as e:
         return f"خطأ Gemini: {e}"
 
-
 # ==========================
 # معالجة الفريم
 # ==========================
@@ -201,12 +196,9 @@ def process_frame(
     if chat_history is None:
         chat_history = []
 
-    # Gradio يعطينا RGB، نحوله BGR لـ OpenCV
     frame_bgr = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
-    # نعكس الصورة أفقياً
     frame_bgr = cv2.flip(frame_bgr, 1)
 
-    # YOLO على GPU + نصف دقة لو متوفر
     results = model.predict(
         frame_bgr,
         conf=CONF_THRESHOLD,
@@ -269,7 +261,6 @@ def process_frame(
         chat_history,
     )
 
-
 # ==========================
 # واجه�� Gradio
 # ==========================
@@ -317,8 +308,6 @@ with gr.Blocks() as demo:
         ],
     )
 
-
+# في Spaces ما نحتاج نحدد port غالباً
 if __name__ == "__main__":
-    # لو أنت على Hugging Face Spaces وطلع لك موضوع الـ hot-reload،
-    # نقدر نضيف باتش بسيط هنا، بس حالياً نخليها عادية:
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch()