added another interface

Dockerfile

@@ -2,10 +2,13 @@ FROM python:3.9-slim-buster
 
 WORKDIR /app
 
+RUN mkdir /app/results_t
+
+RUN mkdir /app/results_c
+
 COPY requirements.txt .
 
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
 RUN apt-get update && apt-get install -y ffmpeg
 

app.py

@@ -1,92 +1,68 @@
+import os
+
 import gradio as gr
-import subprocess
-from pathlib import Path
-from utlis import HHMMSS_to_sec
 
-def trim_media(media_in, media_out, start, end): 
-    
-    # options for writing out audio if converting
-    WAV_CHANNELS = 1
-    WAV_SAMPLE_RATE = 16000
-    
-    media_type = Path(media_in).suffix
-    ext = Path(media_out).suffix
+from pathlib import Path
 
-    if isinstance(start, str):
-        start_sec = HHMMSS_to_sec(start)
-    else: 
-        start_sec = float(start)
-    if isinstance(end, str):
-        end_sec = HHMMSS_to_sec(end)
-    else:
-        end_sec = float(end)
+from utlis import HHMMSS_to_sec, convert_video_format, trim_media
 
-    if ext == '.wav':
-        # convert to wav with standard format for audio models
-        print(f'...Using ffmpeg to trim video from {start} to {end} \n   and convert to {WAV_SAMPLE_RATE}Hz WAV with {WAV_CHANNELS} channels...')            
-        print(f'...generating {media_out}...')            
 
-        subprocess.call(['ffmpeg',
-            '-y',
-            '-i',
-            media_in,
-            '-ss',
-            f'{start_sec}',
-            '-to',
-            f'{end_sec}',
-            '-acodec',
-            'pcm_s16le',
-            '-ac',
-            WAV_CHANNELS,
-            '-ar',
-            WAV_SAMPLE_RATE,
-            media_out,
-            '-hide_banner', 
-            '-loglevel', 
-            'warning'        
-        ], shell=False)   
-    
-    else: 
-        print(f'...Using ffmpeg to trim video from {start_sec} to {end_sec}...')      
-        print(f'...generating {media_out}...')            
-      
-        subprocess.call(['ffmpeg',
-            '-y',
-            '-i',
-            media_in,
-            '-ss',
-            f'{start_sec}',
-            '-to',
-            f'{end_sec}',
-            '-c',
-            'copy',
-            media_out,
-            '-hide_banner', 
-            '-loglevel', 
-            'warning'        
-        ], shell=False) 
-        
-def convert_video(input_file, output_format, start_time, end_time):
+def trim_video(input_file, output_format, start_time, end_time):
     try:
         # Set output file name and extension
-        output_file = f"{input_file.name.split('.')[0]}.{output_format}"
-        
+        output_folder = f"{os.getcwd()}/results_vt/"
+        if not os.path.exists(output_folder):
+            os.makedirs(output_folder)
+        file_name = f"{Path(input_file.name).stem.partition('.')[0]}_trimmed.{output_format}"
+        output_file = os.path.join(output_folder, file_name)
+        print(f"Output folder: {output_folder}")
+        print(f"file_name: {file_name}")
+        print(f"Output file: {output_file}")
         if start_time == "":
-            start_time = 0
+            start_time = "00:00:00"
         if end_time == "":
-            end_time = 0
+            end_time = "00:05:00"
         # Trim the video
-        trim_media(input_file.name, output_file, start_time, end_time)
+        print("start trimming")
+        output_file = trim_media(input_file.name, output_file, start_time, end_time)
+        
+        return output_file
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+def convert_video(input_file, output_format):
+    try:
+        # Set output file name and extension
+        output_folder = f"{os.getcwd()}/results_ct/"
+        if not os.path.exists(output_folder):
+            os.makedirs(output_folder)
+        output_file = os.path.join(output_folder, f"{input_file.name.split('.')[0]}.{output_format}")
+        print(f"Output folder: {output_folder}")
+        print(f"Output file: {output_file}")
+        # Convert the video
+        print("start converting")
+        convert_video_format(input_file.name, output_file)
         
         return output_file
     except Exception as e:
         return f"Error: {str(e)}"
 
+# gr components for video trimmer
 input_file = gr.File(label="Select video file")
 output_format = gr.Dropdown(choices=["mkv", "MOV", "mp4", "wav"], label="Select output format")
 start_time = gr.Textbox(label="Start time (in seconds or HH:MM:SS). Leave blank to start at beginning.")
-end_time = gr.Textbox(label="End time (in seconds or HH:MM:SS). Leave blank to end at end.")
+end_time = gr.Textbox(label="End time (in seconds or HH:MM:SS). Leave blank to trim a  5-minute interval since start.")
 output_file = gr.File(label="Download converted file")
 
-interface = gr.Interface(fn=convert_video, inputs=[input_file, output_format, start_time, end_time], outputs=output_file, title="Video Trimmer")
-interface.launch(server_name="0.0.0.0", server_port=7860)
+interface = gr.Interface(fn=trim_video, inputs=[input_file, output_format, start_time, end_time], outputs=output_file, title="Video Trimmer")
+
+# gr components for video converter
+input_file_c = gr.File(label="Select video file")
+output_format_c = gr.Dropdown(choices=["mkv", "MOV", "mp4", "wav"], label="Select output format")
+output_file_c = gr.File(label="Download converted file")
+
+interface_c = gr.Interface(fn=convert_video, inputs=[input_file_c, output_format_c], outputs=output_file_c, title="Video Converter")
+
+demo = gr.TabbedInterface([interface, interface_c], ["Video Trimmer", "Video Converter"])
+
+demo.launch(server_name="0.0.0.0", server_port=7860)

utlis.py

@@ -1,9 +1,92 @@
-import os
 import csv
+import os
 import re
-import pandas as pd
+import subprocess
 from pathlib import Path
 
+import pandas as pd
+
+def convert_video_format(media_in, media_out):
+    WAV_CHANNELS = 1
+    WAV_SAMPLE_RATE = 16000
+    
+    if not isinstance(media_in, (str, Path)):
+        raise TypeError("media_in must be a string or a PathLike object")
+    
+    if not isinstance(media_out, (str, Path)):
+        raise TypeError("media_out must be a string or a PathLike object")
+    ext = Path(media_out).suffix
+    if ext == '.wav':
+        # convert to wav with standard format for audio models         
+        print(f'...generating {media_out}...')            
+        subprocess.run(f'ffmpeg -y -i "{media_in}" -acodec pcm_s16le -ac {WAV_CHANNELS} -ar {WAV_SAMPLE_RATE} "{media_out}" -hide_banner -loglevel warning', check=True, shell=True)
+
+
+    else:
+        print(f'...Using ffmpeg to convert {media_in} to {media_out}...')      
+        print(f'...generating {media_out}...')            
+      
+        subprocess.run(['ffmpeg',
+                         '-y',
+                         '-i',
+                         media_in,
+                         '-c',
+                         'copy',
+                         media_out,
+                         '-hide_banner', 
+                         '-loglevel', 
+                         'warning'
+                        ], check=True)
+
+def trim_media(media_in, media_out, start, end): 
+    
+    # options for writing out audio if converting
+    WAV_CHANNELS = 1
+    WAV_SAMPLE_RATE = 16000
+    
+    media_type = Path(media_in).suffix
+    ext = Path(media_out).suffix
+    print("EXT", ext)
+    if isinstance(start, str):
+        start_sec = HHMMSS_to_sec(start)
+    else: 
+        start_sec = float(start)
+    if isinstance(end, str):
+        end_sec = HHMMSS_to_sec(end)
+    else:
+        end_sec = float(end)
+
+    if ext == '.wav':
+        # convert to wav with standard format for audio models
+        print(f'...Using ffmpeg to trim video from {start} to {end} \n   and convert to {WAV_SAMPLE_RATE}Hz WAV with {WAV_CHANNELS} channels...')            
+        print(f'...generating {media_out}...')            
+        
+        subprocess.run(f'ffmpeg -y -i "{media_in}" -ss {start_sec} -to {end_sec} -acodec pcm_s16le -ac {WAV_CHANNELS} -ar {WAV_SAMPLE_RATE} "{media_out}" -hide_banner -loglevel warning', check=True, shell=True)
+
+        print(f'...done trimming and converting to {media_out}...')
+        return media_out
+    
+    else: 
+        print(f'...Using ffmpeg to trim video from {start_sec} to {end_sec}...')      
+        print(f'...generating {media_out}...')            
+      
+        subprocess.run(['ffmpeg',
+                         '-y',
+                         '-i',
+                         media_in,
+                         '-ss',
+                         f'{start_sec}',
+                         '-to',
+                         f'{end_sec}',
+                         '-c',
+                         'copy',
+                         media_out,
+                         '-hide_banner', 
+                         '-loglevel', 
+                         'warning'
+                        ], check=True)
+        return media_out
+
 def HHMMSS_to_sec(time_str):
     """Get Seconds from timestamp string with milliseconds."""
     if not time_str: