Update app.py

app.py

@@ -40,14 +40,20 @@ def detect_objects_video(video):
     return temp_output.name
 
 
-demo = gr.Blocks()
-
-#Interface Gradio
-image_input = gr.Image(type="numpy", label="Image à analyser")
-
-image_output = gr.Image(type="numpy", label="Image annotée")
-
-
-interface = gr.Interface(fn=detect_objects_image, inputs=image_input, outputs=image_output, title="Détection sur Image")
-
-interface.launch()
+# Lancer l'interface avec deux onglets
+with gr.Blocks(theme='shivi/calm_seafoam') as demo:
+    gr.Markdown("## 📸 Détection d’objets avec YOLOv8 - Image et Vidéo")
+
+    with gr.Tab("🖼️ Image"):
+        image_input = gr.Image(type="numpy", label="Importer une image")
+        image_output = gr.Image(type="numpy", label="Image avec détection")
+        image_btn = gr.Button("Analyser l'image")
+        image_btn.click(fn=detect_objects_image, inputs=image_input, outputs=image_output)
+
+    with gr.Tab("🎞️ Vidéo"):
+        video_input = gr.Video(label="Importer une vidéo (.mp4)")
+        video_output = gr.Video(label="Vidéo traitée")
+        video_btn = gr.Button("Analyser la vidéo")
+        video_btn.click(fn=detect_objects_video, inputs=video_input, outputs=video_output)
+
+demo.launch()