Update app.py

app.py

@@ -2,20 +2,23 @@ import gradio as gr
 from pathlib import Path
 import subprocess
 from inference_img import *
+import spaces
 
+# Constants
 OUTPUT_GIF = Path("demo/output.gif")
+FRAME1_PATH = Path("demo/frame1.jpg")
+FRAME2_PATH = Path("demo/frame2.jpg")
 
+# Function to generate GIF
 def generate_demo_gif(img1, img2):
     # Save uploaded images
-    frame1_path = Path("demo/frame1.jpg")
-    frame2_path = Path("demo/frame2.jpg")
-    img1.save(frame1_path)
-    img2.save(frame2_path)
+    img1.save(FRAME1_PATH)
+    img2.save(FRAME2_PATH)
 
     try:
         result = subprocess.run([
             "python", "inference_img.py",
-            "--img", str(frame1_path), str(frame2_path),
+            "--img", str(FRAME1_PATH), str(FRAME2_PATH),
             "--exp=4",
             "--model", "train_log/"
         ], capture_output=True, text=True)
@@ -24,13 +27,15 @@ def generate_demo_gif(img1, img2):
         print("STDERR:", result.stderr)
 
         if result.returncode == 0 and OUTPUT_GIF.exists():
-            return OUTPUT_GIF
+            return str(OUTPUT_GIF), "✅ GIF generated successfully!"
         else:
-            return "GIF generation failed."
+            return None, "❌ GIF generation failed. Check logs or file paths."
+
     except Exception as e:
         print(f"Error: {e}")
-        return "No subprocss. Error during GIF generation."
+        return None, f"❌ Subprocess error: {e}"
 
+# Gradio UI
 with gr.Blocks() as demo_ui:
     with gr.Tab("Demo"):
         gr.Markdown("### Upload two frames to generate a GIF")
@@ -39,11 +44,12 @@ with gr.Blocks() as demo_ui:
             img2_input = gr.Image(type="pil", label="Frame 2")
         generate_btn = gr.Button("Generate GIF")
         output_gif = gr.Image(label="Generated GIF")
+        status_msg = gr.Markdown()
 
         generate_btn.click(
             fn=generate_demo_gif,
             inputs=[img1_input, img2_input],
-            outputs=output_gif
+            outputs=[output_gif, status_msg]
         )
 
 demo_ui.launch()