app.py

import gradio as gr
import cv2
from Yolov5_Deepsort.demo import app_main
import sys
import os
sys.path.append(os.path.dirname(os.path.abspath(__file__)))


css = """
.video-container {
    display: flex;
    justify-content: center;
    align-items: center;

}
"""

def show_video():
    return "DDMDeepsort1.mp4"

def show_result_video():
    return "result.mp4"

def process_video(video):
    cap = cv2.VideoCapture(video)
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    out = cv2.VideoWriter('output.mp4', fourcc, 20.0, (int(cap.get(3)), int(cap.get(4))))

    while cap.isOpened():
        ret, frame = cap.read()
        if not ret:
            break
        gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        out.write(cv2.cvtColor(gray_frame, cv2.COLOR_GRAY2BGR))
    
    cap.release()
    out.release()
    return 'output.mp4'

title = "Welcome to DDM DeepSort"
description = "Upload a video to process it using DDM and DeepSORT."

with gr.Blocks(css=css) as demo:

    gr.Interface(
        fn=app_main,
        inputs="video",
        outputs="video",
        title=title,
        description=description
    )
    gr.HTML("""
<h1>Welcome to My Neuroscience Project</h1>
<p>The author is a third-year undergraduate student at the School of Intelligent Science and Technology, Nanjing University, Suzhou Campus.</p>
<h2>Note</h2>
<div style="border: 2px solid rgba(0, 0, 0, 0.1); border-radius: 10px; background-color: rgba(255, 255, 255, 0.8); padding: 20px;">
<p>Since this project uses Hugging Face's free CPU, the processing speed is very slow. In the worst case, even a video with a dozen frames can take several minutes to process. Therefore, if possible, it is recommended to deploy on a device with a better GPU.</p>
<p>Although the YOLOv5 model supports up to 80 classes, my project is primarily focused on autonomous driving. Therefore, objects other than people and cars will be excluded after object detection.</p>
</div>
<div style="border: 2px solid rgba(0, 0, 0, 0.1); border-radius: 10px; background-color: rgba(255, 255, 255, 0.8); padding: 20px;">
<h3>Tips for First-Time Users:</h3>
<ul>
<li>Ensure that the video includes at least people and cars.</li>
<li>It's recommended that the video is not too long, ideally within 10 seconds.</li>
</ul>
</div>
<div style="border: 2px solid rgba(0, 0, 0, 0.1); border-radius: 10px; background-color: rgba(255, 255, 255, 0.8); padding: 20px;">
The following video is a short animation created by the author using Manim to explain the general process.
</div>
""")


    with gr.Row():
        gr.HTML('<div class="video-container">')
        gr.Video(show_video(), label="animation")
        gr.HTML('</div>')
    with gr.Row():
        gr.HTML('<div class="video-container">')
        gr.Video('3d_ddm_deepsort.mp4', label="3d")
        gr.HTML('</div>')
    with gr.Row():
        gr.HTML('<div class="video-container">')
        gr.Video(show_result_video(), label="result")
        gr.HTML('</div>')



demo.launch()