AItool's picture
clear output
b2946a9 verified
# app.py
import os
import uuid
import subprocess
from pathlib import Path
import gradio as gr
FRAME1 = Path("demo/frame1.png")
FRAME2 = Path("demo/frame2.png")
TARGET_DIR = "/home/user/app/output/"
PALETTE_PNG = Path(TARGET_DIR) / "palette.png"
OUTPUT_GIF = Path(TARGET_DIR) / "output.gif"
os.makedirs(TARGET_DIR, exist_ok=True)
def interpolate_image(img_a_path: str, img_b_path: str) -> str:
# --- clear any previous output ---
if OUTPUT_GIF.exists():
OUTPUT_GIF.unlink() # delete old GIF
if PALETTE_PNG.exists():
PALETTE_PNG.unlink() # delete old palette
# optional: clear any old frame PNGs
for f in TARGET_DIR.glob("img*.png"):
f.unlink()
# 1) Run inference to generate frames into TARGET_DIR/img%d.png
subprocess.run([
"python3", "inference_img.py",
"--img", str(img_a_path), str(img_b_path),
"--exp", "4"
], check=True)
# 2) Generate palette from frames
subprocess.run([
"ffmpeg", "-y", "-r", "14", "-f", "image2",
"-i", f"{TARGET_DIR}img%d.png",
"-vf", "palettegen=stats_mode=single",
"-frames:v", "1",
str(PALETTE_PNG)
], check=True)
# 3) Generate final GIF using palette
subprocess.run([
"ffmpeg", "-y", "-r", "14", "-f", "image2",
"-i", f"{TARGET_DIR}img%d.png",
"-i", str(PALETTE_PNG),
"-lavfi", "paletteuse",
str(OUTPUT_GIF)
], check=True)
return str(OUTPUT_GIF)
# helper to read the Markdown file
def load_description(path):
with open(path, "r", encoding="utf-8") as f:
return f.read()
description_text = load_description("TITLE.md")
with gr.Blocks(title="RIFE Image Interpolation") as demo:
# render the markdown text at the top of the UI
gr.Markdown(description_text)
with gr.Tab("Demo"):
gr.Markdown("### Demo: Preloaded images")
input_imageA = gr.Image(type="filepath", value=str(FRAME1), label="Image A")
input_imageB = gr.Image(type="filepath", value=str(FRAME2), label="Image B")
run_btn = gr.Button("Interpolate")
result_img = gr.Image(type="filepath", label="Interpolated GIF")
#result_path = gr.Textbox(label="Output path", interactive=False)
run_btn.click(interpolate_image, [input_imageA, input_imageB], [result_img])
with gr.Tab("Upload your images"):
gr.Markdown("### Upload any two images")
user_A = gr.Image(type="filepath", label="Image A")
user_B = gr.Image(type="filepath", label="Image B")
run_btn2 = gr.Button("Interpolate")
user_img = gr.Image(type="filepath", label="Interpolated GIF")
#user_path = gr.Textbox(label="Output path", interactive=False)
run_btn2.click(interpolate_image, [user_A, user_B], [user_img])
gr.HTML("""<div style="margin: 0.75em 0;"><a href="https://www.buymeacoffee.com/Artgen" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/default-orange.png" alt="Buy Me A Coffee" height="41" width="174"></a></div>
<div style="margin: 0.75em 0;">But what would really help me is a <strong>PRO subscription</strong> to Google Colab, Kaggle or Hugging Face. Many thanks.</div>""")
demo.launch()