Update main.py

main.py

@@ -1,7 +1,38 @@
-from fastapi import FastAPI
+from flask import Flask, render_template, request, redirect, url_for
+from utility import load_image,interpolate_batch,interpolate_single
+from io import BytesIO
+from PIL import Image
+import base64
 
-app = FastAPI()
+app = Flask(__name__)
 
-@app.get("/")
-def read_root():
-    return {"Hello": "World!"}
+
+
+def convert_frames_to_base64(frames):
+    """Converts numpy image arrays to base64 strings for displaying in HTML."""
+    base64_frames = []
+    for frame in frames:
+        img = Image.fromarray((frame * 255).astype(np.uint8))
+        buffered = BytesIO()
+        img.save(buffered, format="PNG")
+        base64_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+        base64_frames.append(base64_str)
+    return base64_frames
+
+@app.route("/", methods=["GET", "POST"])
+def index():
+    if request.method == "POST":
+        img1_url = request.form.get("img1_url")
+        img2_url = request.form.get("img2_url")
+
+        if not img1_url or not img2_url:
+            return redirect(url_for("index"))
+
+        frames = interpolate_single(img1_url, img2_url)
+        if frames is not None:
+            base64_frames = convert_frames_to_base64(frames)
+            return render_template("result.html", frames=base64_frames)
+        else:
+            return "Error processing images.", 500
+
+    return render_template("index.html")