Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from ffmpy import FFmpeg
+from ffmpy import FFmpeg, FFprobe
 import gradio as gr
 import subprocess
 import shortuuid
@@ -35,23 +35,48 @@ def convert(file: _TemporaryFileWrapper, options: str):
         stdout += f"{stderr}"
     return [stdout, output_file]
 
+# Check Video Codec
+def chk_cod(a):
+    command = f"ffprobe {a} 2>&1 >/dev/null"
+    output = subprocess.check_output(command, shell=True).decode("utf-8")
+    match = re.search(r"Stream.*Video.*", output)
+    if match:
+        video_info = match.group()
+        codec = re.sub(r".*Video: ([^, ]+).*", r"\1", video_info)
+        return codec
+    
 # Command Builder: Smooth Interpolation
-def cmdb_si(a, b, c):
-    tuning = c.split(" –")[0]
+def cmdb_si(a, b, c, d):
+    # Check Input Video Codec
+    cod = chk_cod(d)
+    if cod == "h264":
+        tuning = f"-tune {c.split(' –')[0]}"
+    else:
+        tuning = ""
     # print(tuning)
-    return f"-filter:v \"minterpolate='mi_mode=mci:mc_mode=aobmc:me_mode=bidir:vsbmc=1:fps={a}'\" -r {a} -preset {b} -tune {tuning}"
+    return f"-filter:v \"minterpolate='mi_mode=mci:mc_mode=aobmc:me_mode=bidir:vsbmc=1:fps={a}'\" -r {a} -preset {b} {tuning}"
 
 # Command Builder: Frame Blending
-def cmdb_fb(a, b, c):
-    tuning = c.split(" –")[0]
+def cmdb_fb(a, b, c, d):
+    # Check Input Video Codec
+    cod = chk_cod(d)
+    if cod == "h264":
+        tuning = f"-tune {c.split(' –')[0]}"
+    else:
+        tuning = ""
     # print(tuning)
-    return f"-filter:v \"tblend\" -r {a} -preset {b} -tune {tuning}"
+    return f"-filter:v \"tblend\" -r {a} -preset {b} {tuning}"
 
 # Command Builder: Advanced
-def cmdb_adv(a, b):
-    tuning = b.split(" –")[0]
+def cmdb_adv(a, b, c):
+    # Check Input Video Codec
+    cod = chk_cod(c)
+    if cod == "h264":
+        tuning = f"-tune {b.split(' –')[0]}"
+    else:
+        tuning = ""
     #gr.Textbox.update(value=f"-preset {a} -tune {b}")
-    return f"-preset {a} -tune {tuning}"
+    return f"-preset {a} {tuning}"
 
 with gr.Blocks(title="FFmo - FFmpeg Online", theme=gr.themes.Soft()) as main:
     with gr.Tabs():
@@ -63,7 +88,8 @@ with gr.Blocks(title="FFmo - FFmpeg Online", theme=gr.themes.Soft()) as main:
                     input_tune = gr.Radio(["film – use for high quality movie content; lowers deblocking", "animation – good for cartoons; uses higher deblocking and more reference frames", "grain – preserves the grain structure in old, grainy film material", "stillimage – good for slideshow-like content", "fastdecode – allows faster decoding by disabling certain filters", "zerolatency – good for fast encoding and low-latency streaming", "psnr – ignore this as it is only used for codec development", "ssim – ignore this as it is only used for codec development"], value=["film – use for high quality movie content; lowers deblocking"], label="Tune (Required)", info="Tuning Setting")
                     input_video = gr.Video(label="Input Video")
                     input_textbox = gr.Textbox(label="FFMPEG Command")
-                    buildcmd = gr.Button("Build FFMPEG Command", variant="primary").click(fn=cmdb_si, inputs=[input_fps,input_preset,input_tune], outputs=[input_textbox])
+                    buildcmd = gr.Button("Build FFMPEG Command", variant="primary").click(fn=cmdb_si, inputs=[input_fps,input_preset,input_tune,input_video], outputs=[input_textbox])
+                    # input_video.change()
 
                 with gr.Column() as out_si:
                     output_textbox = gr.Textbox(label="Output Logs", interactive=False)