SickstadiumAI / app.py
jschwab21's picture
Update app.py
4b4dd62 verified
raw
history blame
1.4 kB
import gradio as gr
from video_processing import process_video
def display_results(video_url, description):
final_clip_path = process_video(video_url, description)
if final_clip_path:
return final_clip_path, final_clip_path
return "No matching scene found", None
# Custom CSS
css = """
body {
background-color: #eb5726;
color: #ecf0f1;
font-family: 'Arial', sans-serif;
}
h1, h2, h3, h4, h5, h6 {
color: #0b0d1a;
}
#video_url, #description {
background-color: #eb5726;
color: #ecf0f1;
border: 1px solid #ecf0f1;
}
#submit_button {
background-color: #eb5726;
color: #ecf0f1;
border: 1px solid #ecf0f1;
}
#submit_button:hover {
background-color: #c0392b;
}
#video_output, #download_output {
border: 1px solid #eb5726;
}
"""
with gr.Blocks(css=css) as demo:
gr.Markdown("# Sickstadium AI")
video_url = gr.Textbox(label="Video URL or Filepath", elem_id="video_url")
description = gr.Textbox(label="Description of desired clip", elem_id="description")
video_output = gr.Video(label="Processed Video", elem_id="video_output")
download_output = gr.File(label="Download Processed Video", elem_id="download_output")
submit_button = gr.Button("Process Video", elem_id="submit_button")
submit_button.click(fn=display_results, inputs=[video_url, description], outputs=[video_output, download_output])
demo.launch()