Update app.py

app.py

@@ -1,161 +1,24 @@
-import os
 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("Sora"),
-            "ui-sans-serif",
-            "sans-serif",
-        ),
-        font_mono: fonts.Font | str | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("Sora"),
-            "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, 1) 0%, rgba(235, 87, 38, 0) 70%), radial-gradient(#eb5726 1px, transparent 1px)",
-            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_text_color="#ffffff",
-        )
-
-custom_theme = CustomTheme()
-
-from datetime import datetime
-
-def save_uploaded_file(uploaded_bytes):
-    upload_dir = "uploaded_videos"
-    os.makedirs(upload_dir, exist_ok=True)
-    # Generate a unique filename using timestamp to avoid overwriting files
-    file_name = f"uploaded_video_{datetime.now().strftime('%Y%m%d%H%M%S')}.mp4"
-    file_path = os.path.join(upload_dir, file_name)
-    with open(file_path, "wb") as f:
-        f.write(uploaded_bytes)
-    return file_path
-
-def display_results(video_file):
-    if video_file:
-        video_file_path = save_uploaded_file(video_file)  # Assume this function returns the saved file path
-        scenes = find_scenes(video_file_path)
-        if scenes:
-            return f"Detected scenes: {scenes}"
-        else:
-            return "No scenes detected."
-    else:
-        return "No file uploaded."
-
-
-css = """
-body {
-    background-color: #ffffff;
-    background-image: radial-gradient(#eb5726 1px, transparent 1px);
-    background-size: 10px 10px;
-    background-repeat: repeat;
-    background-attachment: fixed;
-}
-#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;
-}
-#submit_button:hover {
-    background-color: #f5986e;
-    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: 3em !important;
-    font-weight: bold;
-    text-transform: uppercase;
-}
-"""
-
-def interface_function(video_file):
-    if video_file is not None:
-        file_path = save_uploaded_file(video_file)
-        return f"File saved at {file_path}"
-    return "No file uploaded."
-
-def test_upload(uploaded_file):
-    if uploaded_file is None:
-        return "No file uploaded."
+def display_video(video_file):
+    if video_file is None:
+        return None, "No video uploaded."
     try:
-        # Save the uploaded file and process it
-        file_path = save_uploaded_file(uploaded_file)
-        if not file_path:
-            return "Failed to save file."
-
-        # Process the video using the path
-        result_path = process_video(file_path, description="Describe your clip here", is_url=False)
-        if not result_path:
-            return "Failed to process video or no scenes found."
-
-        return f"Video processed and saved to: {result_path}"
+        # Directly return the path to the uploaded video file for displaying
+        return video_file, None
     except Exception as e:
-        # Catch any exceptions that occur and return them for debugging
-        return f"An error occurred: {str(e)}"
+        return None, f"An error occurred: {str(e)}"
 
 with gr.Blocks() as demo:
     with gr.Column():
         video_file = gr.UploadButton("Upload Video File", type="binary", file_types=["video"])
-        output = gr.Textbox(label="Output")
-        submit_button = gr.Button("Process Video")
+        output_video = gr.Video()
+        output_message = gr.Textbox(label="Output Message")
+        submit_button = gr.Button("Display Video")
         submit_button.click(
-            fn=test_upload, 
-            inputs=[video_file], 
-            outputs=[output]
+            fn=display_video, 
+            inputs=video_file, 
+            outputs=[output_video, output_message]
         )
 
-demo.launch()
+demo.launch()