Upload app.py

app.py

@@ -37,11 +37,24 @@ def load_detector(model_name):
 @spaces.GPU
 def process_webcam(image, model_choice, confidence_threshold):
     """Traite l'image de la webcam"""
-    print(f"🎥 Frame reçue - Type: {type(image)}")
+    print(f"🎥 Frame reçue - Type: {type(image)}, Shape: {getattr(image, 'size', 'N/A')}")
     
     if image is None:
-        print("❌ Image None reçue")
-        return None
+        print("❌ Image None reçue - webcam pas encore initialisée")
+        # Retourner une image d'attente
+        waiting_img = Image.new('RGB', (640, 480), color='#1f2937')
+        draw = ImageDraw.Draw(waiting_img)
+        try:
+            font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 32)
+        except:
+            font = ImageFont.load_default()
+        
+        text = "🎥 En attente de la webcam..."
+        bbox = draw.textbbox((0, 0), text, font=font)
+        x = (640 - bbox[2]) // 2
+        y = (480 - bbox[3]) // 2
+        draw.text((x, y), text, fill='white', font=font)
+        return waiting_img
     
     try:
         # S'assurer qu'on a une image PIL
@@ -138,23 +151,30 @@ def draw_detections(image, detections):
     
     return img_copy
 
-# Interface simplifiée au maximum
+# Interface avec configuration webcam améliorée
 demo = gr.Interface(
     fn=process_webcam,
     inputs=[
-        gr.Image(sources=["webcam"], streaming=True, type="pil"),
+        gr.Image(
+            sources=["webcam"], 
+            streaming=True, 
+            type="pil",
+            mirror_webcam=False,
+            show_download_button=False
+        ),
         gr.Dropdown(
             choices=list(REALTIME_MODELS.keys()),
             value="YOLOS Tiny (ultra-rapide)",
             label="Modèle"
         ),
-        gr.Slider(0.1, 1.0, 0.3, step=0.1, label="Confiance")
+        gr.Slider(0.1, 1.0, 0.1, step=0.1, label="Confiance")
     ],
-    outputs=gr.Image(streaming=True, type="pil"),
+    outputs=gr.Image(streaming=True, type="pil", show_download_button=False),
     live=True,
     title="🎥 Détection Live",
-    description="Autorisez la webcam pour voir la détection d'objets en temps réel",
-    allow_flagging="never"
+    description="⚠️ Cliquez sur l'icône webcam et autorisez l'accès pour commencer",
+    allow_flagging="never",
+    flagging_mode="never"
 )
 
 if __name__ == "__main__":