Update app.py

app.py

@@ -1,16 +1,31 @@
-import gradio as gr
-from PIL import Image
+"""
+Реальное время: распознавание объектов с веб-камеры через Gradio и YOLOv8.
+Работает с Gradio 3.49.3
+"""
+
 import numpy as np
-from ultralytics import YOLO
+from PIL import Image
+import gradio as gr
+
+try:
+    from ultralytics import YOLO
+except Exception as e:
+    YOLO = None
+    print("Ultralytics import failed:", e)
 
+# Кэширование модели
 _MODEL = {"obj": None}
 
-def load_model(model_name="yolov8n.pt"):
+def load_model(model_name: str = "yolov8n.pt"):
+    """Загружает YOLO-модель"""
+    if YOLO is None:
+        raise RuntimeError("Пакет ultralytics не найден. Проверь requirements.txt.")
     if _MODEL["obj"] is None:
         _MODEL["obj"] = YOLO(model_name)
     return _MODEL["obj"]
 
-def annotate_frame(frame, conf=0.25):
+def annotate_frame(frame: np.ndarray, conf: float = 0.25):
+    """Выполняет детекцию и возвращает кадр с аннотациями"""
     if frame is None:
         return None
     model = load_model()
@@ -18,33 +33,45 @@ def annotate_frame(frame, conf=0.25):
     annotated = results[0].plot()
     return np.array(annotated)
 
-def detect_live(frame, conf):
-    if frame is None:
+def detect_live(image, conf):
+    """Обработчик Gradio для стриминга с веб-камеры"""
+    if image is None:
         return None
-    annotated = annotate_frame(frame, conf)
+    annotated = annotate_frame(image, conf)
     return Image.fromarray(annotated)
 
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎥 Реальное время: детекция объектов (YOLOv8)")
-    with gr.Row():
-        with gr.Column(scale=2):
-            # Захват с камеры через tool="camera"
-            webcam = gr.Video(label="Веб-камера", tool="camera", interactive=True)
-            conf = gr.Slider(0, 1, 0.25, 0.01, label="Confidence")
-            model_path = gr.Textbox(value="yolov8n.pt", visible=False)
-        with gr.Column(scale=1):
-            output = gr.Image(label="Результат детекции")
-            start_btn = gr.Button("Load model")
-            status = gr.Textbox(value="Модель не загружена", interactive=False)
-
-    def load_event(spec):
-        try:
-            load_model(spec)
-            return f"Модель загружена: {spec}"
-        except Exception as e:
-            return f"Ошибка: {e}"
-
-    start_btn.click(load_event, inputs=[model_path], outputs=[status])
-    webcam.change(detect_live, inputs=[webcam, conf], outputs=[output])
-
-demo.launch()
+def main():
+    title = "🎥 Реальное время: распознавание объектов (YOLOv8)"
+    description = "Потоковое детектирование объектов с веб-камеры. Используется модель `yolov8n.pt`."
+
+    with gr.Blocks() as demo:
+        gr.Markdown(f"# {title}\n\n{description}")
+
+        with gr.Row():
+            with gr.Column(scale=2):
+                # Для Gradio 3.49.3 — рабочий способ захвата веб-камеры
+                webcam = gr.Image(source="webcam", streaming=True, label="Веб-камера")
+                conf = gr.Slider(0.0, 1.0, value=0.25, step=0.01, label="Порог достоверности")
+                model_path = gr.Textbox(value="yolov8n.pt", label="Модель", visible=False)
+
+            with gr.Column(scale=1):
+                output = gr.Image(label="Результат детекции", interactive=False)
+                start_btn = gr.Button("Загрузить модель")
+                status = gr.Textbox(value="Модель не загружена", label="Статус", interactive=False)
+
+        # Кнопка загрузки модели
+        def load_event(spec):
+            try:
+                load_model(spec)
+                return f"Модель загружена: {spec}"
+            except Exception as e:
+                return f"Ошибка: {e}"
+
+        start_btn.click(load_event, inputs=[model_path], outputs=[status])
+        webcam.change(detect_live, inputs=[webcam, conf], outputs=[output])
+
+        demo.launch()
+
+if __name__ == "__main__":
+    main()
+