history_log

app.py

@@ -1,15 +1,22 @@
 import cv2
 import gradio as gr
 import os
+from datetime import datetime
+import time
+import psutil
 
-video_history = []
+def convert_to_grayscale(video_file, history):
+    start_time = time.time()
+    file_name = os.path.splitext(os.path.basename(video_file))[0]
+    output_file = f"{file_name}_output.mp4"
 
-def convert_to_grayscale(video_file):
     cap = cv2.VideoCapture(video_file)
-    output_file = "output.mp4"
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     fps = int(cap.get(cv2.CAP_PROP_FPS))
+    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    video_duration = frame_count / fps
+
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(output_file, fourcc, fps, (width, height), isColor=False)
 
@@ -24,28 +31,40 @@ def convert_to_grayscale(video_file):
     cap.release()
     out.release()
 
-    return output_file
+    end_time = time.time()
+    process_time = end_time - start_time
+    current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    memory_usage = psutil.Process(os.getpid()).memory_info().rss / 1024 / 1024  # in MB
+
+    history.append(f"#{len(history)+1} File Name: {file_name}, Collected on: {current_time}, Video Duration: {video_duration:.2f} seconds, Processing Time: {process_time:.2f} seconds, Memory Usage: {memory_usage:.2f} MB")
+
+    return output_file, history
 
-# function to get history (coming soon)
-def get_history():
-    history_html = ""
+def get_history(history):
+    history_html = "<br>".join(history)
     return history_html
 
+with gr.Blocks() as demo:
+    history_state = gr.State([])
+
+    with gr.Tab("Convert Video"):
+        with gr.Row(""):
+            video_input = gr.Video(label="Upload Video", height=500, width=500)
+            video_output = gr.Video(label="Grayscale Video", height=500, width=500)
+        convert_button = gr.Button("Convert")
+
+        convert_button.click(
+            fn=convert_to_grayscale,
+            inputs=[video_input, history_state],
+            outputs=[video_output, history_state]
+        )
+
+    with gr.Tab("History") as history_tab:
+        history_output = gr.HTML(label="History")
 
-demo = gr.Interface(
-    fn=convert_to_grayscale,
-    title="Video Upload and Display",
-    inputs=gr.Video(label="Upload Video", height=500, width=500),
-    outputs=gr.Video(label="Grayscale Video", height=500, width=500),
-    allow_flagging="manual",
-)
+        def update_history(history):
+            return get_history(history)
 
-history_demo = gr.Interface(
-    fn=get_history,
-    title="Processed Video History",
-    inputs=[],
-    outputs=gr.Textbox(label="History"),
-    allow_flagging="never"
-)
+        convert_button.click(fn=update_history, inputs=history_state, outputs=history_output)
 
-gr.TabbedInterface([demo, history_demo], ["Convert Video", "History"]).launch()
+demo.launch()