Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 import zipfile
 import subprocess
-from gradio import Progress
+import shutil
 
 
 # --------- AUTO EXTRACT train_log.zip ----------
@@ -37,8 +37,6 @@ def ensure_train_log():
         return
 
     os.rename(found_path, target_folder)
-
-    import shutil
     shutil.rmtree(tmp_dir)
 
     print("train_log extracted successfully!")
@@ -65,38 +63,23 @@ def get_fps(video_path):
         else:
             fps = float(result)
 
-        return int(fps)
+        return int(fps)  # làm tròn xuống
 
     except:
         return 0
 
 
-# -------- RIFE RUN WITH REAL PROGRESS --------
+# -------- RUN RIFE + STREAM TERMINAL LOG --------
 def run_rife(video, multi):
-    progress = Progress(track_tqdm=True)
-
     input_path = video
+
     if not os.path.isfile(input_path):
-        return "Error: invalid input video", None
+        yield "Error: invalid input video", None
+        return
 
     ext = os.path.splitext(input_path)[1].lower()
     output_path = f"output{ext}"
 
-    # ---- Lấy tổng frame để chạy % ----
-    total_frames = 0
-    try:
-        cmd_count = [
-            "ffprobe", "-v", "error",
-            "-count_frames", "-select_streams", "v:0",
-            "-show_entries", "stream=nb_read_frames",
-            "-of", "default=nokey=1:noprint_wrappers=1",
-            input_path
-        ]
-        total_frames = int(subprocess.check_output(cmd_count).decode().strip())
-    except:
-        total_frames = 0
-
-    # ---- Chạy RIFE và đọc stdout để cập nhật % ----
     cmd = [
         "python3", "inference_video.py",
         "--video", input_path,
@@ -104,45 +87,27 @@ def run_rife(video, multi):
         "--multi", str(multi)
     ]
 
+    # Run process + stream output line by line
     process = subprocess.Popen(
         cmd,
         stdout=subprocess.PIPE,
         stderr=subprocess.STDOUT,
-        text=True
+        text=True,
+        bufsize=1
     )
 
-    current = 0
-
-    # --- Đọc log realtime ---
+    log_buffer = ""
     for line in process.stdout:
-        line = line.strip()
-
-        # RIFE chuẩn hay in "XX/YYY"
-        if "/" in line:
-            try:
-                parts = line.split("/")
-                cur = int(parts[0])
-                total = int(parts[1])
-
-                percent = cur / total
-                progress(percent)
-
-            except:
-                pass
-
-        # fallback nếu ffprobe lấy được số frame
-        elif total_frames > 0 and "frame" in line:
-            current += 1
-            progress(current / total_frames)
+        log_buffer += line
+        yield log_buffer, None
 
     process.wait()
 
     if process.returncode != 0:
-        return "Error running RIFE", None
-
-    progress(1.0)  # hoàn tất 100%
+        yield log_buffer + "\nError while running RIFE!", None
+        return
 
-    return "Done", output_path
+    yield log_buffer + "\nDone!", output_path
 
 
 # -------- AUTO UPDATE FPS WHEN VIDEO CHANGES ----------
@@ -164,10 +129,10 @@ with gr.Blocks() as demo:
     gr.Markdown("# **Practical RIFE**")
     gr.Markdown("### https://github.com/hzwer/Practical-RIFE")
 
-    video_input = gr.Video(label="Upload video (MP4 / MKV / MOV)", format=None)
+    video_input = gr.Video(label="Upload video (MP4 / MKV / MOV)")
 
     fps_display = gr.Number(label="Video FPS (auto-read)", value=0, interactive=False)
-    multi_input = gr.Slider(1, 10, value=4, step=1, label="multi (1-10)")
+    multi_input = gr.Slider(1, 10, value=4, step=1, label="multi (1–10)")
 
     fps_output = gr.Number(label="Output FPS (display only)", value=0, interactive=False)
 
@@ -176,11 +141,14 @@ with gr.Blocks() as demo:
 
     btn = gr.Button("Run RIFE")
 
-    log = gr.Textbox(label="Log")
+    log = gr.Textbox(label="Terminal Log (realtime)", lines=20)
     output_video = gr.Video(label="Output video")
 
-    btn.click(run_rife, inputs=[video_input, multi_input], outputs=[log, output_video])
-
+    btn.click(
+        run_rife,
+        inputs=[video_input, multi_input],
+        outputs=[log, output_video]
+    )
 
 if __name__ == "__main__":
     demo.launch()