Spaces:
Sleeping
Sleeping
File size: 4,962 Bytes
dcb53fc 0268b89 79da273 0268b89 3af1c8c 0268b89 79da273 0268b89 79da273 0d88658 beacb7e 0268b89 0d88658 beacb7e 0268b89 d0d0ed9 3af1c8c 0268b89 12d9720 81edc2c 79da273 0268b89 3af1c8c dcb53fc f774200 e579fcc f774200 4e02215 82df05c d0d0ed9 82df05c 4e02215 df55e72 e755c35 4e02215 de9023a 4e02215 5d5059e 6bf06f4 d0d0ed9 cb240bb 4e02215 5d5059e dcb53fc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
import gradio as gr
from video_processing import process_video
from gradio.themes.base import Base
from gradio.themes.utils import colors, fonts, sizes
from typing import Iterable
class CustomTheme(Base):
def __init__(
self,
*,
primary_hue: colors.Color | str = colors.orange,
secondary_hue: colors.Color | str = colors.orange,
neutral_hue: colors.Color | str = colors.gray,
spacing_size: sizes.Size | str = sizes.spacing_md,
radius_size: sizes.Size | str = sizes.radius_md,
text_size: sizes.Size | str = sizes.text_md,
font: fonts.Font | str | Iterable[fonts.Font | str] = (
fonts.GoogleFont("Roboto"),
"ui-sans-serif",
"sans-serif",
),
font_mono: fonts.Font | str | Iterable[fonts.Font | str] = (
fonts.GoogleFont("Roboto Mono"),
"ui-monospace",
"monospace",
),
):
super().__init__(
primary_hue=primary_hue,
secondary_hue=secondary_hue,
neutral_hue=neutral_hue,
spacing_size=spacing_size,
radius_size=radius_size,
text_size=text_size,
font=font,
font_mono=font_mono,
)
super().set(
body_background_fill="radial-gradient(circle at center, rgba(235, 87, 38, 0.3), rgba(235, 87, 38, 0.3) 20%, rgba(255, 255, 255, 0) 21%), radial-gradient(circle at center, rgba(235, 87, 38, 0.2), rgba(235, 87, 38, 0.2) 40%, rgba(255, 255, 255, 0) 41%), radial-gradient(circle at center, rgba(235, 87, 38, 0.1), rgba(235, 87, 38, 0.1) 60%, rgba(255, 255, 255, 0) 61%), radial-gradient(circle at center, rgba(235, 87, 38, 0.05), rgba(235, 87, 38, 0.05) 80%, rgba(255, 255, 255, 0) 81%), #ffffff",
body_text_color="#282828",
block_background_fill="#ffffff",
block_title_text_color="#eb5726",
block_label_text_color="#eb5726",
button_primary_background_fill="#eb5726",
button_primary_background_fill_hover="#ffffff",
button_primary_text_color="#ffffff",
button_primary_text_color_hover="#eb5726",
)
custom_theme = CustomTheme()
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 for additional styling
css = """
body {
background-color: #ffffff;
background-image: radial-gradient(circle at center, rgba(235, 87, 38, 0.3), rgba(235, 87, 38, 0.3) 20%, rgba(255, 255, 255, 0) 21%), radial-gradient(circle at center, rgba(235, 87, 38, 0.2), rgba(235, 87, 38, 0.2) 40%, rgba(255, 255, 255, 0) 41%), radial-gradient(circle at center, rgba(235, 87, 38, 0.1), rgba(235, 87, 38, 0.1) 60%, rgba(255, 255, 255, 0) 61%), radial-gradient(circle at center, rgba(235, 87, 38, 0.05), rgba(235, 87, 38, 0.05) 80%, rgba(255, 255, 255, 0) 81%);
}
#video_url {
background-color: #ffffff;
color: #282828;
border: 2px solid #eb5726;
}
#description {
background-color: #ffffff;
color: #282828;
border: 2px solid #eb5726;
}
#submit_button {
background-color: #eb5726;
color: #ffffff;
border: 2px solid #ffffff;
}
label[for="video_url"] {
color: #eb5726 !important;
}
label[for="description"] {
color: #eb5726 !important;
}
h3 {
color: #eb5726;
}
.centered-markdown {
text-align: center;
background-color: #ffffff;
padding: 10px;
}
#sickstadium-title {
font-size: 2em !important;
font-weight: bold;
}
"""
with gr.Blocks(theme=custom_theme, css=css) as demo:
with gr.Column():
gr.Markdown("# **Sickstadium AI**", elem_classes="centered-markdown", elem_id="sickstadium-title")
gr.Markdown("### **Upload your videos. Find sick clips. Tell your truth.**", elem_classes="centered-markdown")
gr.Markdown("**Welcome to Sickstadium AI. Our goal is to empower content creators with the ability to tell their stories without the friction of traditional video editing software. Skip the timeline, and don't worry about your experience in video editing. Upload your video, explain the clip you want, and let our AI video editor do the work for you. Get more info about the Sickstadium project at Strongholdlabs.io**", elem_classes="centered-markdown")
video_url = gr.Textbox(label="Video URL or Filepath", elem_id="video_url")
description = gr.Textbox(label="Description of desired clip", elem_id="description")
submit_button = gr.Button("Process Video", elem_id="submit_button")
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.click(fn=display_results, inputs=[video_url, description], outputs=[video_output, download_output])
demo.launch()
|