Update app.py

app.py

@@ -63,23 +63,17 @@ def display_results(video_url, video_file, description):
     elif video_file:
         video_path = save_uploaded_file(video_file)
     else:
-        return "No video provided", None
+        return "No video provided", None, None
 
     scenes = find_scenes(video_path)
     if not scenes:
-        return "No scenes detected", None
+        return "No scenes detected", None, 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
+    final_clip_path, sentiments = analyze_scenes(video_path, scenes, description)
+    if final_clip_path:
+        return final_clip_path, sentiments
     else:
-        return "No matching scene found", None
+        return "No matching scene found", None, None
         
 
 # Custom CSS for additional styling
@@ -140,17 +134,25 @@ def save_uploaded_file(uploaded_file):
         f.write(uploaded_file)
     return file_path
 
+import matplotlib.pyplot as plt
+
+def create_plot(sentiments):
+    categories = ["Joy", "Trust", "Fear", "Surprise", "Sadness", "Disgust", "Anger", "Anticipation"]
+    fig, ax = plt.subplots()
+    ax.bar(categories, sentiments)
+    ax.set_ylabel('Probability')
+    ax.set_title('Sentiment Distribution')
+    plt.setp(ax.get_xticklabels(), rotation=45, horizontalalignment='right')
+    return fig
+
 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 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")
-        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])
-
+        sentiment_plot = gr.Plot(label="Sentiment Analysis", elem_id="sentiment_plot")
+        submit_button.click(fn=display_results, inputs=[video_url, video_file, description], outputs=[video_output, sentiment_plot])
 demo.launch()