Update app.py

app.py

@@ -2,10 +2,9 @@ import gradio as gr
 from video_processing import process_video, download_video, find_scenes, analyze_scenes, extract_best_scene, cleanup_temp_files
 from gradio.themes.base import Base
 from gradio.themes.utils import colors, fonts, sizes
+from typing import Iterable
 import uuid
 import os
-from typing import Iterable
-
 
 class CustomTheme(Base):
     def __init__(
@@ -57,43 +56,31 @@ def save_uploaded_file(uploaded_file):
     with open(file_path, "wb") as f:
         f.write(uploaded_file)
     return file_path
-
+    
 def display_results(video_url, video_file, description):
-    output_message = None
     if video_url:
         video_path = download_video(video_url)
     elif video_file:
         video_path = save_uploaded_file(video_file)
     else:
-        output_message = "No video provided"
-
-    if not output_message:
-        scenes = find_scenes(video_path)
-        if not scenes:
-            output_message = "No scenes detected"
-
-    if not output_message:
-        best_scene_info = analyze_scenes(video_path, scenes, description)
-        if best_scene_info:
-            best_scene = best_scene_info[0] if isinstance(best_scene_info, tuple) else None
-            sentiment_distribution = best_scene_info[-1] if isinstance(best_scene_info, tuple) else None
-            final_clip = extract_best_scene(video_path, best_scene) if best_scene else None
-            if final_clip:
-                output_dir = "output"
-                os.makedirs(output_dir, exist_ok=True)
-                final_clip_path = os.path.join(output_dir, f"{uuid.uuid4()}_final_clip.mp4")
-                final_clip.write_videofile(final_clip_path, codec='libx264', audio_codec='aac')
-                cleanup_temp_files()
+        return "No video provided", None
 
-                plot = create_radial_plot(sentiment_distribution) if sentiment_distribution else "No sentiment data available"
-                return final_clip_path, plot
-            else:
-                output_message = "No matching scene found"
-        else:
-            output_message = "Analysis failed or no suitable scenes found"
-
-    return None, output_message
+    scenes = find_scenes(video_path)
+    if not scenes:
+        return "No scenes detected", None
 
+    best_scene = analyze_scenes(video_path, scenes, description)
+    final_clip = extract_best_scene(video_path, best_scene)
+    if final_clip:
+        output_dir = "output"
+        os.makedirs(output_dir, exist_ok=True)
+        final_clip_path = os.path.join(output_dir, f"{uuid.uuid4()}_final_clip.mp4")
+        final_clip.write_videofile(final_clip_path, codec='libx264', audio_codec='aac')
+        cleanup_temp_files()
+        return final_clip_path, final_clip_path
+    else:
+        return "No matching scene found", None
+        
 
 # Custom CSS for additional styling
 css = """
@@ -124,7 +111,10 @@ body {
     color: #ffffff;
     border: 2px solid #ffffff;
 }
-label[for="video_url"], label[for="description"] {
+label[for="video_url"] {
+    color: #eb5726 !important;
+}
+label[for="description"] {
     color: #eb5726 !important;
 }
 h3 {
@@ -141,19 +131,26 @@ h3 {
     text-transform: uppercase;
 }
 """
+
+def save_uploaded_file(uploaded_file):
+    upload_dir = "uploaded_videos"
+    os.makedirs(upload_dir, exist_ok=True)
+    file_path = os.path.join(upload_dir, f"{uuid.uuid4()}.mp4")
+    with open(file_path, "wb") as f:
+        f.write(uploaded_file)
+    return file_path
+
 with gr.Blocks(theme=custom_theme, css=css) as demo:
     with gr.Column():
-        video_url_input = gr.Textbox(label="Video URL:", elem_id="video_url")
-        video_file_input = gr.File(label="Upload Video File:", interactive=True, file_types=["video"], type="binary")
-        description_input = gr.Textbox(label="Describe your clip:", elem_id="description")
+        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 video editing skills. Upload your video, describe 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](https://strongholdlabs.io/)**", elem_classes="centered-markdown")
+        video_url = gr.Textbox(label="Video URL:", elem_id="video_url")
+        video_file = gr.File(label="Upload Video File:", elem_id="video_file", interactive=True, file_types=["video"], type="binary")
+        description = gr.Textbox(label="Describe your 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")
-        sentiment_plot_output = gr.Plot(label="Sentiment Distribution", elem_id="sentiment_plot")
-
-        submit_button.click(
-            fn=display_results,
-            inputs=[video_url_input, video_file_input, description_input],
-            outputs=[video_output, sentiment_plot_output]
-        )
+        download_output = gr.File(label="Download Processed Video", elem_id="download_output")
+        submit_button.click(fn=display_results, inputs=[video_url, video_file, description], outputs=[video_output, download_output])
 
-demo.launch()
+demo.launch()